dev add check count for all type

2026-03-16 14:12:34 +07:00
parent 168aa3812c
commit 68e102b6ce
4 changed files with 25 additions and 26 deletions
@@ -35,6 +35,7 @@ import ru.soune.nocopy.repository.AuthTokenRepository;
 import ru.soune.nocopy.repository.FileEntityRepository;
 import ru.soune.nocopy.repository.UserRepository;
 import ru.soune.nocopy.service.FileSimilarityService;
+import ru.soune.nocopy.service.file.CheckCounterService;
 import ru.soune.nocopy.service.file.ProtectionsLimitService;
 import ru.soune.nocopy.service.register.AuthService;
 import ru.soune.nocopy.service.file.FileEntityService;
@@ -74,6 +75,8 @@ public class ApiController {

    private final UserRepository userRepository;

+    private final CheckCounterService checkCounterService;
+
    @PostMapping("/v{version}/data")
    public ResponseEntity<?> handlePostRequest(@RequestBody BaseRequest request,
                                               @PathVariable("version") int version) {
@@ -174,6 +177,7 @@ public class ApiController {

        try {
            FileEntity fileEntity = fileEntityRepository.findByFileId(fileId);
+
            if (fileEntity == null) {
                return ResponseEntity.ok().body(new BaseResponse(0, MessageCode.INVALID_FIELD.getCode(),
                        MessageCode.INVALID_FIELD.getDescription(), null));
@@ -188,6 +192,7 @@ public class ApiController {
                List<SimilarFileDTO> results = processNonImageFile(fileEntity, authToken);
                log.info("results: {}", results);
                response = buildSuccessResponse(results, MessageCode.SIMILAR_FILES_FOUND);
+                checkCounterService.incrementCheckCount(fileEntity.getUserId(), fileEntity.getMimeType());
            }

            return ResponseEntity.ok(response);