dev add vio

2026-03-12 01:32:11 +07:00
parent 869a25043a
commit 65bbb00abf
10 changed files with 321 additions and 6 deletions
@@ -0,0 +1,109 @@
+package ru.soune.nocopy.service.search;
+
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import ru.soune.nocopy.dto.search.GlobalSearchStatisticsResponse;
+import ru.soune.nocopy.entity.company.Company;
+import ru.soune.nocopy.entity.search.GlobalSearchTask;
+import ru.soune.nocopy.entity.user.User;
+import ru.soune.nocopy.repository.GlobalSearchResultRepository;
+import ru.soune.nocopy.repository.GlobalSearchTaskRepository;
+import ru.soune.nocopy.repository.UserRepository;
+import ru.soune.nocopy.repository.ViolationRepository;
+
+import java.time.LocalDateTime;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+@Slf4j
+@Service
+@RequiredArgsConstructor
+public class GlobalSearchStatisticsService {
+
+    private final GlobalSearchTaskRepository taskRepository;
+
+    private final GlobalSearchResultRepository resultRepository;
+
+    private final ViolationRepository violationRepository;
+
+    private final UserRepository userRepository;
+
+    @Transactional(readOnly = true)
+    public GlobalSearchStatisticsResponse getStatistics(Long userId, Integer limit, Integer days) {
+        User user = userRepository.findById(userId).orElseThrow();
+        Company company = user.getCompany();
+        List<Long> userIds = company == null ? Arrays.asList(userId):
+                user.getCompany()
+                        .getUsers()
+                        .stream()
+                        .map(User::getId).toList();
+
+        long totalSearches = taskRepository.countByUserId(userIds);
+        List<Object[]> statusCounts = taskRepository.countByStatusAndUserId(userIds);
+
+        GlobalSearchStatisticsResponse.StatusCountDto statusCountDto =
+                GlobalSearchStatisticsResponse.StatusCountDto.builder()
+                .processing(0)
+                .completed(0)
+                .failed(0)
+                .total(totalSearches)
+                .build();
+
+        for (Object[] row : statusCounts) {
+            String status = (String) row[0];
+            Long count = (Long) row[1];
+
+            if ("PROCESSING".equalsIgnoreCase(status)) {
+                statusCountDto.setProcessing(count);
+            } else if ("COMPLETED".equalsIgnoreCase(status) || "SUCCESS".equalsIgnoreCase(status)) {
+                statusCountDto.setCompleted(statusCountDto.getCompleted() + count);
+            } else if ("FAILED".equalsIgnoreCase(status) || "TIMEOUT".equalsIgnoreCase(status)) {
+                statusCountDto.setFailed(statusCountDto.getFailed() + count);
+            }
+        }
+
+        List<GlobalSearchTask> recentTasks;
+
+        if (days != null && days > 0) {
+            LocalDateTime fromDate = LocalDateTime.now().minusDays(days);
+            recentTasks = taskRepository.findByUserIdAndCreatedAtAfterOrderByCreatedAtDesc(userId, fromDate);
+
+            if (limit != null && limit > 0 && recentTasks.size() > limit) {
+                recentTasks = recentTasks.subList(0, limit);
+            }
+        } else {
+            org.springframework.data.domain.Pageable pageable =
+                    org.springframework.data.domain.PageRequest.of(0, limit != null ? limit : 10);
+            recentTasks = taskRepository.findTopByUserIdOrderByCreatedAtDesc(userId, pageable);
+        }
+
+        List<GlobalSearchStatisticsResponse.SearchStatDto> recentSearchDtos = new ArrayList<>();
+
+        for (GlobalSearchTask task : recentTasks) {
+            long violationsCount = getViolationsCountForTask(task.getTaskId());
+
+            recentSearchDtos.add(GlobalSearchStatisticsResponse.SearchStatDto.fromEntity(task, violationsCount));
+        }
+
+        return GlobalSearchStatisticsResponse.builder()
+                .totalSearches(totalSearches)
+                .searchesByStatus(statusCountDto)
+                .recentSearches(recentSearchDtos)
+                .build();
+    }
+
+    @Transactional(readOnly = true)
+    public long getViolationsCountForTask(String taskId) {
+        List<String> fileIds = resultRepository.findByTaskId(taskId).stream()
+                .map(result -> result.getFileId())
+                .toList();
+
+        if (fileIds.isEmpty()) {
+            return 0;
+        }
+        return violationRepository.countByFileEntityIdIn(fileIds);
+    }
+}