backdev/no-copy
1
0
Fork 0
Code Issues Pull Requests 5 Actions Packages Projects Releases Wiki Activity

dev add config for nginx
Test Workflow / test (push) Successful in 2s
Details

Browse Source
This commit is contained in:
vladp
2026-03-10 22:47:43 +07:00
parent 590d42690b
commit c591ab40df
17 changed files with 495 additions and 306 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/ru/soune/nocopy/service/monitoring/MonitoringSearchService.java
+13 -53
View File
@@ -3,32 +3,28 @@ package ru.soune.nocopy.service.monitoring;
import jakarta.mail.MessagingException;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import ru.soune.nocopy.dto.file.YandexSearchResponse;
import ru.soune.nocopy.dto.tarriff.TariffDTO;
import ru.soune.nocopy.entity.file.*;
import ru.soune.nocopy.entity.monitoring.FileMonitoringEntity;
import ru.soune.nocopy.entity.monitoring.FoundViolationEntity;
import ru.soune.nocopy.entity.monitoring.MonitoringType;
import ru.soune.nocopy.entity.monitoring.SearchEngine;
import ru.soune.nocopy.entity.search.GlobalSearchResult;
import ru.soune.nocopy.entity.tarif.TariffInfo;
import ru.soune.nocopy.entity.tarif.TariffType;
import ru.soune.nocopy.entity.user.User;
import ru.soune.nocopy.exception.TariffNotFoundException;
import ru.soune.nocopy.repository.FileMonitoringRepository;
import ru.soune.nocopy.repository.FoundViolationRepository;
import ru.soune.nocopy.repository.UserRepository;
import ru.soune.nocopy.service.mail.EmailService;
import ru.soune.nocopy.service.search.SearchImageService;
import ru.soune.nocopy.service.tariff.TariffInfoService;
import ru.soune.nocopy.service.tariff.TariffService;
import ru.soune.nocopy.service.violation.ViolationService;
import java.io.IOException;
import java.time.LocalDateTime;
import java.util.List;
import java.util.UUID;
import java.util.concurrent.TimeoutException;
@Slf4j
@@ -37,8 +33,6 @@ import java.util.concurrent.TimeoutException;
public class MonitoringSearchService {
private final SearchImageService searchImageService;
private final FoundViolationRepository violationRepository;
private final FileMonitoringRepository monitoringRepository;
private final TariffInfoService tariffInfoService;
@@ -49,18 +43,16 @@ public class MonitoringSearchService {
private final UserRepository userRepository;
private final ViolationService violationService;
@Transactional
public void processFileSearch(FileMonitoringEntity monitoring) throws MessagingException, IOException {
String searchSessionId = UUID.randomUUID().toString();
log.info("Starting search session: {} for file: {}", searchSessionId, monitoring.getFile().getId());
monitoring.setLastRun(LocalDateTime.now());
monitoring.setLastRunStatus("IN_PROGRESS");
monitoringRepository.save(monitoring);
MonitoringType monitoringType = monitoring.getMonitoringType();
TariffDTO tariffMonitoring = tariffService.getTariffByType(TariffType.valueOf(monitoringType.name()));
try {
tariffInfoService.writeOffTokens(monitoring.getUserId(), tariffMonitoring.getTokens());
@@ -70,7 +62,9 @@ public class MonitoringSearchService {
List<YandexSearchResponse.ImageResult> yandexImages =
searchImageService.getAllImagesWithoutPagination(yandexResponse, "visual_matches");
int savedCount = saveFoundImages(monitoring, yandexImages, SearchEngine.YANDEX, searchSessionId);
for (YandexSearchResponse.ImageResult imageResult : yandexImages) {
violationService.processViolation(imageResult, monitoring.getFile(), null);
}
try {
String googleResponse = searchImageService.searchReverseByPublicUrl(
@@ -79,13 +73,16 @@ public class MonitoringSearchService {
List<YandexSearchResponse.ImageResult> googleImages =
searchImageService.getAllImagesWithoutPagination(googleResponse, "exact_matches");
savedCount += saveFoundImages(monitoring, googleImages, SearchEngine.GOOGLE, searchSessionId);
for (YandexSearchResponse.ImageResult imageResult : googleImages) {
violationService.processViolation(imageResult, monitoring.getFile(), null);
}
} catch (TimeoutException | IOException e) {
log.warn("Google search failed for session: {}", searchSessionId, e);
log.warn("Google search failed");
}
monitoring.setLastRunStatus("SUCCESS - Found " + savedCount + " violations");
monitoring.setLastRunStatus("SUCCESS");
} catch (TariffNotFoundException e) {
User user = userRepository.findById(monitoring.getUserId()).orElseThrow();
TariffInfo activeTariffInfo = user.getActiveTariffInfo();
@@ -97,7 +94,6 @@ public class MonitoringSearchService {
monitoringRepository.save(monitoring);
} catch (Exception e) {
log.error("Search failed for session: {}", searchSessionId, e);
monitoring.setLastRunStatus("ERROR: " + e.getMessage());
} finally {
updateNextRun(monitoring);
@@ -105,42 +101,6 @@ public class MonitoringSearchService {
}
}
private int saveFoundImages(FileMonitoringEntity monitoring,
List<YandexSearchResponse.ImageResult> images,
SearchEngine engine,
String sessionId) {
int savedCount = 0;
for (YandexSearchResponse.ImageResult image : images) {
List<FoundViolationEntity> existing = violationRepository.findDuplicate(
monitoring.getFile().getId(), image.getUrl());
if (existing.isEmpty()) {
FoundViolationEntity violation = FoundViolationEntity.builder()
.file(monitoring.getFile())
.fileId(monitoring.getFile().getId())
.fileName(monitoring.getFile().getOriginalFileName())
.userId(monitoring.getUserId())
.foundUrl(image.getUrl())
.pageUrl(image.getPageUrl())
.pageTitle(image.getPageTitle())
.host(image.getHost())
.imageWidth(image.getWidth())
.imageHeight(image.getHeight())
.searchEngine(engine)
.searchSessionId(sessionId)
.isNew(true)
.isNotified(false)
.build();
violationRepository.save(violation);
savedCount++;
}
}
return savedCount;
}
private void updateNextRun(FileMonitoringEntity monitoring) {
LocalDateTime nextRun = calculateNextRun(monitoring.getMonitoringType());
monitoring.setNextRun(nextRun);