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

Feat: 진행 테이블 제거 로직

Browse Source
This commit is contained in:
김용수 2024-09-24 22:28:56 +09:00
parent 4e3c6e195e
commit 3d1cf56261
3 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/src/main/java/com/worlabel/domain/progress/repository/ProgressCacheRepository.java
View File

@ -29,4 +29,9 @@ public class ProgressCacheRepository {
String key = CacheKey.autoLabelingProgressKey(); String key = CacheKey.autoLabelingProgressKey();
redisTemplate.opsForSet().add(key, projectId); redisTemplate.opsForSet().add(key, projectId);
} }
public void removePredictProgress(final int projectId) {
String key = CacheKey.autoLabelingProgressKey();
redisTemplate.opsForSet().remove(key, projectId);
}
} }

3
backend/src/main/java/com/worlabel/domain/progress/service/ProgressService.java
View File

@ -16,7 +16,8 @@ public class ProgressService {
public void predictCheck(final int projectId){ public void predictCheck(final int projectId){
if(progressCacheRepository.predictCheck(projectId)){ if(progressCacheRepository.predictCheck(projectId)){
throw new CustomException(ErrorCode.AI_IN_PROGRESS); // throw new CustomException(ErrorCode.AI_IN_PROGRESS);
progressCacheRepository.removePredictProgress(projectId);
} }
} }

2
backend/src/main/java/com/worlabel/domain/project/service/ProjectService.java
View File

@ -179,7 +179,7 @@ public class ProjectService {
// 응답없음 // 응답없음
log.debug("요청"); log.debug("요청");
progressService.registerPredictProgress(projectId); // progressService.registerPredictProgress(projectId);
List<AutoLabelingResult> list = aiService.postRequest(endPoint, autoLabelingRequest, List.class, this::converter); List<AutoLabelingResult> list = aiService.postRequest(endPoint, autoLabelingRequest, List.class, this::converter);
saveAutoLabelList(list); saveAutoLabelList(list);
} }