This commit is contained in:
@@ -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());
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user