dev add check count for all type
Test Workflow / test (push) Successful in 4s

This commit is contained in:
vladp
2026-03-16 15:03:02 +07:00
parent 3a478f9195
commit 0f7079d118
2 changed files with 26 additions and 25 deletions
@@ -50,9 +50,9 @@ public class ImageFoundRequestHandler implements RequestHandler {
Map.of("fileId",fileId))); Map.of("fileId",fileId)));
}); });
// String searchResponseYandex; // String searchResponseYandex;
String searchResponseGoogle; String searchResponseGoogle;
// List<YandexSearchResponse.ImageResult> allYandexImages = new ArrayList<>(); // List<YandexSearchResponse.ImageResult> allYandexImages = new ArrayList<>();
List<YandexSearchResponse.ImageResult> allGoogleImages = new ArrayList<>(); List<YandexSearchResponse.ImageResult> allGoogleImages = new ArrayList<>();
List<YandexSearchResponse.ImageResult> allUniqueImages; List<YandexSearchResponse.ImageResult> allUniqueImages;
boolean hasTimeout = false; boolean hasTimeout = false;
@@ -70,21 +70,24 @@ public class ImageFoundRequestHandler implements RequestHandler {
log.error("Google search failed for file {}", fileId, e); log.error("Google search failed for file {}", fileId, e);
} }
// try { /*
// searchResponseYandex = searchImageService.searchReverseByPublicUrl(fileEntity, "yandex_reverse_image", try {
// "visual_matches"); searchResponseYandex = searchImageService.searchReverseByPublicUrl(fileEntity, "yandex_reverse_image",
// log.info("searchResponseYandex OK"); "visual_matches");
// allYandexImages = searchImageService.getAllImagesWithoutPagination( log.info("searchResponseYandex OK");
// searchResponseYandex, "visual_matches"); allYandexImages = searchImageService.getAllImagesWithoutPagination(
// } catch (TimeoutException e) { searchResponseYandex, "visual_matches");
// log.warn("Yandex search timeout for file {}", fileId); } catch (TimeoutException e) {
// hasTimeout = true; log.warn("Yandex search timeout for file {}", fileId);
// } catch (IOException e) { hasTimeout = true;
// log.error("Yandex search failed for file {}", fileId, e); } catch (IOException e) {
// } log.error("Yandex search failed for file {}", fileId, e);
}
*/
// allUniqueImages = searchImageService.removeDuplicateUrls(allYandexImages, allGoogleImages); // allUniqueImages = searchImageService.removeDuplicateUrls(allYandexImages, allGoogleImages);
// log.info("allUniqueImages OK: {} images", allUniqueImages.size()); allUniqueImages = allGoogleImages;
log.info("allUniqueImages OK: {} images", allUniqueImages.size());
tariffInfoService.writeOffTokens(fileEntity.getUserId(), TariffConstants.TOKEN_VALUE_FOR_SEARCH); tariffInfoService.writeOffTokens(fileEntity.getUserId(), TariffConstants.TOKEN_VALUE_FOR_SEARCH);
checkCounterService.incrementCheckCount(fileEntity.getUserId(), fileEntity.getMimeType()); checkCounterService.incrementCheckCount(fileEntity.getUserId(), fileEntity.getMimeType());
@@ -92,17 +95,15 @@ public class ImageFoundRequestHandler implements RequestHandler {
int page = imageSearchRequest.getPage() != null ? imageSearchRequest.getPage() : 1; int page = imageSearchRequest.getPage() != null ? imageSearchRequest.getPage() : 1;
int pageSize = 5; int pageSize = 5;
// List<YandexSearchResponse.ImageResult> pagedResults = searchImageService.paginateResults(allUniqueImages, page, List<YandexSearchResponse.ImageResult> pagedResults = searchImageService.paginateResults(allUniqueImages, page,
// pageSize * 2);
List<YandexSearchResponse.ImageResult> pagedResults = searchImageService.paginateResults(allGoogleImages, page,
pageSize * 2); pageSize * 2);
YandexSearchResponse finalResponse = new YandexSearchResponse(); YandexSearchResponse finalResponse = new YandexSearchResponse();
finalResponse.setImages(pagedResults); finalResponse.setImages(pagedResults);
finalResponse.setPage(page); finalResponse.setPage(page);
finalResponse.setPageSize(pageSize * 2); finalResponse.setPageSize(pageSize * 2);
finalResponse.setTotalResults(allGoogleImages.size()); finalResponse.setTotalResults(allUniqueImages.size());
finalResponse.setTotalPages((int) Math.ceil((double) allGoogleImages.size() / (pageSize * 2))); finalResponse.setTotalPages((int) Math.ceil((double) allUniqueImages.size() / (pageSize * 2)));
String message = hasTimeout ? "Search completed (partial)" : MessageCode.SUCCESS.getDescription(); String message = hasTimeout ? "Search completed (partial)" : MessageCode.SUCCESS.getDescription();
@@ -202,10 +202,10 @@ public class SearchImageService {
uniqueByUrl.putIfAbsent(normalizedUrl, image); uniqueByUrl.putIfAbsent(normalizedUrl, image);
} }
// for (YandexSearchResponse.ImageResult image : yandexImages) { for (YandexSearchResponse.ImageResult image : yandexImages) {
// String normalizedUrl = normalizeUrl(image.getUrl()); String normalizedUrl = normalizeUrl(image.getUrl());
// uniqueByUrl.putIfAbsent(normalizedUrl, image); uniqueByUrl.putIfAbsent(normalizedUrl, image);
// } }
return new ArrayList<>(uniqueByUrl.values()); return new ArrayList<>(uniqueByUrl.values());
} }