71 lines
2.9 KiB
Java
71 lines
2.9 KiB
Java
package ru.soune.nocopy.handler;
|
|
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
import lombok.RequiredArgsConstructor;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.springframework.stereotype.Component;
|
|
import ru.soune.nocopy.dto.BaseRequest;
|
|
import ru.soune.nocopy.dto.BaseResponse;
|
|
import ru.soune.nocopy.dto.MessageCode;
|
|
import ru.soune.nocopy.dto.file.ImageSearchRequest;
|
|
import ru.soune.nocopy.entity.file.FileEntity;
|
|
import ru.soune.nocopy.exception.NotValidFieldException;
|
|
import ru.soune.nocopy.repository.FileEntityRepository;
|
|
import ru.soune.nocopy.service.file.CheckCounterService;
|
|
import ru.soune.nocopy.service.search.SearchApiToYandexResponseMapper;
|
|
import ru.soune.nocopy.service.search.YandexSearchService;
|
|
import ru.soune.nocopy.service.search.GoogleVisionSearchService;
|
|
import ru.soune.nocopy.service.tariff.TariffConstants;
|
|
import ru.soune.nocopy.service.tariff.TariffInfoService;
|
|
|
|
import java.util.Map;
|
|
|
|
@Slf4j
|
|
@Component
|
|
@RequiredArgsConstructor
|
|
public class ImageFoundRequestHandler implements RequestHandler {
|
|
|
|
private final ObjectMapper objectMapper;
|
|
|
|
private final YandexSearchService yandexSearchService;
|
|
|
|
private final GoogleVisionSearchService googleVisionSearchService;
|
|
|
|
private final CheckCounterService checkCounterService;
|
|
|
|
private final FileEntityRepository fileEntityRepository;
|
|
|
|
private final TariffInfoService tariffInfoService;
|
|
|
|
private final SearchApiToYandexResponseMapper mapper = new SearchApiToYandexResponseMapper();
|
|
|
|
@Override
|
|
public BaseResponse handle(BaseRequest request) throws Exception {
|
|
ImageSearchRequest imageSearchRequest = objectMapper.convertValue(request.getMessageBody(),
|
|
ImageSearchRequest.class);
|
|
|
|
String fileId = imageSearchRequest.getFileId();
|
|
|
|
FileEntity fileEntity = fileEntityRepository.findById(fileId)
|
|
.orElseThrow(() -> {
|
|
throw new NotValidFieldException("File not found", new BaseResponse(20007,
|
|
MessageCode.FILE_NOT_FOUND.getCode(), MessageCode.FILE_NOT_FOUND.getDescription(),
|
|
Map.of("fileId",fileId)));
|
|
});
|
|
|
|
// YandexSearchResponse response = yandexSearchService.searchByFileEntity(fileEntity);
|
|
|
|
//TODO uncommited when add billing
|
|
// GoogleVisionSearchResponse googleVisionSearchResponse = googleVisionSearchService.searchByFileEntity(fileId);
|
|
|
|
String yandexReverseSearchResponse = yandexSearchService.searchReverseByPublicUrl(fileEntity);
|
|
|
|
tariffInfoService.writeOffTokens(fileEntity.getUserId(), TariffConstants.TOKEN_VALUE_FOR_SEARCH);
|
|
|
|
checkCounterService.incrementCheckCount(fileEntity.getUserId(), fileEntity.getMimeType());
|
|
|
|
return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
|
|
MessageCode.SUCCESS.getDescription(),mapper.mapToYandexResponse(yandexReverseSearchResponse));
|
|
}
|
|
}
|