dev add vio
Test Workflow / test (push) Successful in 3s

This commit is contained in:
vladp
2026-03-12 02:52:38 +07:00
parent 37b1f370e5
commit 2ca73e4417
4 changed files with 19 additions and 23 deletions
@@ -61,7 +61,10 @@ public class GlobalSearchStatisticsResponse {
@JsonProperty("duration_seconds") @JsonProperty("duration_seconds")
private Long durationSeconds; private Long durationSeconds;
public static SearchStatDto fromEntity(GlobalSearchTask task, long violationsCount) { @JsonProperty("file_ids")
private List<String> fileIds;
public static SearchStatDto fromEntity(GlobalSearchTask task, long violationsCount, List<String> fileIds ) {
DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"); DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
Long duration = null; Long duration = null;
@@ -75,6 +78,7 @@ public class GlobalSearchStatisticsResponse {
.status(task.getStatus()) .status(task.getStatus())
.totalFiles(task.getTotalFiles()) .totalFiles(task.getTotalFiles())
.violationsFound(violationsCount) .violationsFound(violationsCount)
.fileIds(fileIds)
.createdAt(task.getCreatedAt() != null ? task.getCreatedAt().format(formatter) : null) .createdAt(task.getCreatedAt() != null ? task.getCreatedAt().format(formatter) : null)
.durationSeconds(duration) .durationSeconds(duration)
.build(); .build();
@@ -133,7 +133,7 @@ public class FileEntityService {
} }
@Transactional(readOnly = true) @Transactional(readOnly = true)
public List<FileEntity> getAllUserFiles(Long userId, int version) { public List<FileEntity> getAllUserFiles(Long userId) {
User user = userRepository.findById(userId).orElseThrow(); User user = userRepository.findById(userId).orElseThrow();
List<FileEntity> allFiles = new ArrayList<>(); List<FileEntity> allFiles = new ArrayList<>();
@@ -6,16 +6,13 @@ import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional; import org.springframework.transaction.annotation.Transactional;
import ru.soune.nocopy.dto.search.GlobalSearchStatisticsResponse; import ru.soune.nocopy.dto.search.GlobalSearchStatisticsResponse;
import ru.soune.nocopy.entity.company.Company; import ru.soune.nocopy.entity.company.Company;
import ru.soune.nocopy.entity.search.GlobalSearchResult;
import ru.soune.nocopy.entity.search.GlobalSearchTask; import ru.soune.nocopy.entity.search.GlobalSearchTask;
import ru.soune.nocopy.entity.user.User; import ru.soune.nocopy.entity.user.User;
import ru.soune.nocopy.repository.GlobalSearchResultRepository; import ru.soune.nocopy.repository.*;
import ru.soune.nocopy.repository.GlobalSearchTaskRepository;
import ru.soune.nocopy.repository.UserRepository;
import ru.soune.nocopy.repository.ViolationRepository;
import java.time.LocalDateTime; import java.time.LocalDateTime;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Arrays;
import java.util.List; import java.util.List;
@Slf4j @Slf4j
@@ -35,7 +32,7 @@ public class GlobalSearchStatisticsService {
public GlobalSearchStatisticsResponse getStatistics(Long userId, Integer limit, Integer days) { public GlobalSearchStatisticsResponse getStatistics(Long userId, Integer limit, Integer days) {
User user = userRepository.findById(userId).orElseThrow(); User user = userRepository.findById(userId).orElseThrow();
Company company = user.getCompany(); Company company = user.getCompany();
List<Long> userIds = company == null ? Arrays.asList(userId): List<Long> userIds = company == null ? List.of(userId) :
user.getCompany() user.getCompany()
.getUsers() .getUsers()
.stream() .stream()
@@ -83,9 +80,16 @@ public class GlobalSearchStatisticsService {
List<GlobalSearchStatisticsResponse.SearchStatDto> recentSearchDtos = new ArrayList<>(); List<GlobalSearchStatisticsResponse.SearchStatDto> recentSearchDtos = new ArrayList<>();
for (GlobalSearchTask task : recentTasks) { for (GlobalSearchTask task : recentTasks) {
long violationsCount = getViolationsCountForTask(task.getTaskId()); List<GlobalSearchResult> globalSearchResults = resultRepository.findByTaskId(task.getTaskId());
List<String> fileIds = globalSearchResults.stream()
.map(GlobalSearchResult::getFileId)
.toList();
recentSearchDtos.add(GlobalSearchStatisticsResponse.SearchStatDto.fromEntity(task, violationsCount)); GlobalSearchStatisticsResponse.SearchStatDto searchStatDto =
GlobalSearchStatisticsResponse.SearchStatDto.fromEntity(task,
violationRepository.countByFileEntityIdIn(fileIds), fileIds);
recentSearchDtos.add(searchStatDto);
} }
return GlobalSearchStatisticsResponse.builder() return GlobalSearchStatisticsResponse.builder()
@@ -94,16 +98,4 @@ public class GlobalSearchStatisticsService {
.recentSearches(recentSearchDtos) .recentSearches(recentSearchDtos)
.build(); .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);
}
} }
@@ -136,7 +136,7 @@ public class ViolationService {
userFiles = List.of(file); userFiles = List.of(file);
} else { } else {
userFiles = fileEntityService.getAllUserFiles(userId, 1); userFiles = fileEntityService.getAllUserFiles(userId);
} }
if (userFiles.isEmpty()) { if (userFiles.isEmpty()) {