jhynsoo/worlabel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix: Ai 리턴 오류 해결

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backend/src/main/java/com/worlabel/domain/alarm/service/AlarmService.java
View File

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

2
backend/src/main/java/com/worlabel/domain/model/service/AiModelService.java
View File

@ -133,7 +133,7 @@ public class AiModelService {
String endPoint = "/models/download"; String endPoint = "/models/download";
endPoint += "?modelKey=" + modelKey; 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) { public TrainRequest getTrainRequest(final ModelTrainRequest trainRequest, final Project project, final AiModel model) {

2
backend/src/main/java/com/worlabel/global/service/FcmService.java
View File

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