dev add check limits logic

src/main/java/ru/soune/nocopy/handler/ImageFoundRequestHandler.java

@@ -9,9 +9,15 @@ import ru.soune.nocopy.dto.BaseResponse;
 import ru.soune.nocopy.dto.MessageCode;
 import ru.soune.nocopy.dto.file.ImageSearchRequest;
 import ru.soune.nocopy.dto.file.YandexSearchResponse;
+import ru.soune.nocopy.entity.file.FileEntity;
+import ru.soune.nocopy.exception.NotValidFieldException;
+import ru.soune.nocopy.repository.FileEntityRepository;
+import ru.soune.nocopy.service.file.CheckCounterService;
 import ru.soune.nocopy.service.search.YandexSearchService;
 import ru.soune.nocopy.service.search.GoogleVisionSearchService;
 
+import java.util.Map;
+
 @Slf4j
 @Component
 @RequiredArgsConstructor
@@ -23,6 +29,10 @@ public class ImageFoundRequestHandler implements RequestHandler {
 
     private final GoogleVisionSearchService googleVisionSearchService;
 
+    private final CheckCounterService checkCounterService;
+
+    private final FileEntityRepository fileEntityRepository;
+
     @Override
     public BaseResponse handle(BaseRequest request) throws Exception {
         ImageSearchRequest imageSearchRequest = objectMapper.convertValue(request.getMessageBody(),
@@ -30,11 +40,20 @@ public class ImageFoundRequestHandler implements RequestHandler {
 
         String fileId = imageSearchRequest.getFileId();
 
-        YandexSearchResponse response = yandexSearchService.searchByFileEntity(fileId);
+        FileEntity fileEntity = fileEntityRepository.findById(fileId)
+                .orElseThrow(() -> {
+                    throw new NotValidFieldException("File not found", new BaseResponse(20007,
+                            MessageCode.FILE_NOT_FOUND.getCode(), MessageCode.FILE_NOT_FOUND.getDescription(),
+                            Map.of("fileId",fileId)));
+                });
+
+        YandexSearchResponse response = yandexSearchService.searchByFileEntity(fileEntity);
 
         //TODO uncommited when add billing
 //        GoogleVisionSearchResponse googleVisionSearchResponse = googleVisionSearchService.searchByFileEntity(fileId);
 
+        checkCounterService.incrementCheckCount(fileEntity.getUserId());
+
         return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
                 MessageCode.SUCCESS.getDescription(), response);
     }