dev add vio

2026-03-12 00:21:19 +07:00
parent 88bfe51297
commit ffa977f06e
7 changed files with 193 additions and 7 deletions
@@ -0,0 +1,77 @@
+package ru.soune.nocopy.handler;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
+import ru.soune.nocopy.dto.BaseRequest;
+import ru.soune.nocopy.dto.BaseResponse;
+import ru.soune.nocopy.dto.violation.ViolationStatisticsRequest;
+import ru.soune.nocopy.dto.violation.ViolationStatisticsResponse;
+import ru.soune.nocopy.service.violation.ViolationService;
+
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
+import java.time.format.DateTimeParseException;
+
+@Slf4j
+@Component
+@RequiredArgsConstructor
+public class ViolationStatisticsHandler implements RequestHandler {
+    private final ObjectMapper objectMapper;
+
+    private final ViolationService violationService;
+
+    private static final DateTimeFormatter DATE_FORMATTER = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
+
+    @Override
+    public BaseResponse handle(BaseRequest request) throws Exception {
+        ViolationStatisticsRequest statRequest = objectMapper.convertValue(
+                request.getMessageBody(), ViolationStatisticsRequest.class);
+
+        validateRequest(statRequest);
+
+        LocalDateTime startDate = null;
+        LocalDateTime endDate = null;
+
+        if (statRequest.getStartDate() != null && statRequest.getEndDate() != null) {
+            startDate = parseDate(statRequest.getStartDate());
+            endDate = parseDate(statRequest.getEndDate());
+
+            if (endDate != null && !statRequest.getEndDate().contains(" ")) {
+                endDate = endDate.withHour(23).withMinute(59).withSecond(59);
+            }
+        }
+
+        ViolationStatisticsResponse statistics = violationService.getViolationStatistics(
+                statRequest.getUserId(),
+                statRequest.getFileId(),
+                startDate,
+                endDate
+        );
+
+        return BaseResponse.builder()
+                .msgId(request.getMsgId())
+                .messageBody(statistics)
+                .build();
+    }
+
+    private void validateRequest(ViolationStatisticsRequest request) {
+        if (request.getUserId() == null) {
+            throw new IllegalArgumentException("User ID is required");
+        }
+    }
+
+    private LocalDateTime parseDate(String dateStr) {
+        try {
+            return LocalDateTime.parse(dateStr, DATE_FORMATTER);
+        } catch (DateTimeParseException e) {
+            try {
+                return LocalDateTime.parse(dateStr + " 00:00:00", DATE_FORMATTER);
+            } catch (DateTimeParseException ex) {
+                log.error("Error parsing date: {}", dateStr, ex);
+                return null;
+            }
+        }
+    }
+}