@@ -11,9 +11,11 @@ import ru.soune.nocopy.dto.MessageCode;
|
||||
import ru.soune.nocopy.dto.file.ImageSearchRequest;
|
||||
import ru.soune.nocopy.dto.file.YandexSearchResponse;
|
||||
import ru.soune.nocopy.entity.file.FileEntity;
|
||||
import ru.soune.nocopy.entity.file.ImageHashEntity;
|
||||
import ru.soune.nocopy.entity.tokenoperation.OperationType;
|
||||
import ru.soune.nocopy.exception.NotValidFieldException;
|
||||
import ru.soune.nocopy.repository.FileEntityRepository;
|
||||
import ru.soune.nocopy.repository.ImageHashRepository;
|
||||
import ru.soune.nocopy.repository.SimilarImageProjection;
|
||||
import ru.soune.nocopy.service.FileSimilarityService;
|
||||
import ru.soune.nocopy.service.ImageHashService;
|
||||
@@ -51,6 +53,8 @@ public class ImageFoundRequestHandler implements RequestHandler {
|
||||
|
||||
private final ImageHashService imageHashService;
|
||||
|
||||
private final ImageHashRepository imageHashRepository;
|
||||
|
||||
private final FileSimilarityService fileSimilarityService;
|
||||
|
||||
@Override
|
||||
@@ -137,6 +141,7 @@ public class ImageFoundRequestHandler implements RequestHandler {
|
||||
|
||||
int page = imageSearchRequest.getPage() != null ? imageSearchRequest.getPage() : 1;
|
||||
int pageSize = 5;
|
||||
|
||||
boolean checkSimilar = imageSearchRequest.getCheckSimilarFirst() == null ? true:
|
||||
imageSearchRequest.getCheckSimilarFirst();
|
||||
|
||||
@@ -151,10 +156,10 @@ public class ImageFoundRequestHandler implements RequestHandler {
|
||||
|
||||
Map<String, Long> hashSearchApiFile = imageHashService.calculateHash(Paths.get(filePath));
|
||||
|
||||
Map<String, Long> hashCheckFile = imageHashService.calculateHash(Paths.get(fileEntity.getFilePath()));
|
||||
ImageHashEntity checkFileHash = imageHashRepository.findByFileId(fileId);
|
||||
|
||||
boolean hashesAreEqual = hashSearchApiFile.get("hi").equals(hashCheckFile.get("hi"))
|
||||
&& hashSearchApiFile.get("low").equals(hashCheckFile.get("low"));
|
||||
boolean hashesAreEqual = hashSearchApiFile.get("hi").equals(checkFileHash.getHash64Hi())
|
||||
&& hashSearchApiFile.get("low").equals(checkFileHash.getHash64Lo());
|
||||
|
||||
if (!hashesAreEqual) {
|
||||
return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
|
||||
|
||||
Reference in New Issue
Block a user