@@ -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()) {
|
||||||
|
|||||||
Reference in New Issue
Block a user