dev add check limits logic

2026-01-28 23:25:16 +07:00
parent 31fac0ab93
commit ec20d68261
12 changed files with 283 additions and 16 deletions
@@ -0,0 +1,79 @@
+package ru.soune.nocopy.service.file;
+
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.orm.ObjectOptimisticLockingFailureException;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import ru.soune.nocopy.dto.file.CheckIncrementResult;
+import ru.soune.nocopy.dto.file.CheckStatus;
+import ru.soune.nocopy.entity.user.ProtectedFileCheck;
+import ru.soune.nocopy.entity.user.User;
+import ru.soune.nocopy.exception.RetryableException;
+import ru.soune.nocopy.repository.ProtectedFileCheckRepository;
+
+@Service
+@RequiredArgsConstructor
+@Slf4j
+public class CheckCounterService {
+
+    private final ProtectedFileCheckRepository protectedFileCheckRepository;
+
+    @Transactional
+    public void incrementCheckCount(Long userId) {
+        try {
+            ProtectedFileCheck check = protectedFileCheckRepository.findByUser_Id(userId)
+                    .orElseThrow(NullPointerException::new);
+
+            boolean success = check.incrementCheck();
+
+            if (success) {
+                protectedFileCheckRepository.save(check);
+                log.info("Check count incremented for user {}: count={}, limit={}",
+                        userId, check.getCountChecked(), check.getLimitCheck());
+                CheckIncrementResult.success(check.getCountChecked(), check.getLimitCheck());
+            } else {
+                log.warn("Check limit exceeded for user {}", userId);
+                CheckIncrementResult.limitExceeded(check.getLimitCheck());
+            }
+
+        } catch (ObjectOptimisticLockingFailureException e) {
+            log.warn("Optimistic lock failure for user {}, retrying...", userId);
+            throw new RetryableException("Please try again");
+        }
+    }
+
+    @Transactional(readOnly = true)
+    public CheckStatus getCurrentStatus(Long userId) {
+        ProtectedFileCheck check = protectedFileCheckRepository.findByUser_Id(userId)
+                .orElseThrow(NullPointerException::new);
+
+        return CheckStatus.builder()
+                .userId(userId)
+                .countChecked(check.getCountChecked())
+                .remainingLimit(check.getLimitCheck())
+                .lastCheckAt(check.getLastCheckAt())
+                .build();
+    }
+
+
+    public void initProtectCheckLimit(User user, Integer checked, Integer limit) {
+        ProtectedFileCheck check = ProtectedFileCheck.builder()
+                .user(user)
+                .limitCheck(limit)
+                .countChecked(checked)
+                .build();
+
+        protectedFileCheckRepository.save(check);
+    }
+
+    @Transactional
+    public ProtectedFileCheck updateLimit(User user, Integer limit) {
+        ProtectedFileCheck check = ProtectedFileCheck.builder()
+                .user(user)
+                .limitCheck(limit)
+                .build();
+
+        return protectedFileCheckRepository.save(check);
+    }
+}