Fix: Ai 리턴 오류 해결

This commit is contained in:
김용수 2024-09-30 00:34:37 +09:00
parent 81e9f81d8b
commit 6a1409dbe5
3 changed files with 4 additions and 1 deletions

View File

@ -19,6 +19,7 @@ public class AlarmService {
private final FcmService fcmService;
public void save(int memberId, AlarmType type) {
log.debug("알람 전송 memberId {}, type {}", memberId, type);
alarmCacheRepository.save(memberId, type);
fcmService.send(memberId, type.toString());
}

View File

@ -133,7 +133,7 @@ public class AiModelService {
String endPoint = "/models/download";
endPoint += "?modelKey=" + modelKey;
ResponseEntity<Resource> fileRequest = aiRequestService.getFileRequest(endPoint);
return aiRequestService.getFileRequest(endPoint);
}
public TrainRequest getTrainRequest(final ModelTrainRequest trainRequest, final Project project, final AiModel model) {

View File

@ -26,6 +26,8 @@ public class FcmService {
public void send(Integer memberId, String data) {
String token = fcmCacheRepository.getToken(memberId);
log.debug("토큰 {}", token);
if(Objects.nonNull(token)){
sendNotification(token, data);
}