diff --git a/backend/src/main/java/com/worlabel/domain/progress/repository/ProgressCacheRepository.java b/backend/src/main/java/com/worlabel/domain/progress/repository/ProgressCacheRepository.java index ddebcd9..1d0fb37 100644 --- a/backend/src/main/java/com/worlabel/domain/progress/repository/ProgressCacheRepository.java +++ b/backend/src/main/java/com/worlabel/domain/progress/repository/ProgressCacheRepository.java @@ -29,4 +29,9 @@ public class ProgressCacheRepository { String key = CacheKey.autoLabelingProgressKey(); redisTemplate.opsForSet().add(key, projectId); } + + public void removePredictProgress(final int projectId) { + String key = CacheKey.autoLabelingProgressKey(); + redisTemplate.opsForSet().remove(key, projectId); + } } diff --git a/backend/src/main/java/com/worlabel/domain/progress/service/ProgressService.java b/backend/src/main/java/com/worlabel/domain/progress/service/ProgressService.java index aab44ef..a5b13a8 100644 --- a/backend/src/main/java/com/worlabel/domain/progress/service/ProgressService.java +++ b/backend/src/main/java/com/worlabel/domain/progress/service/ProgressService.java @@ -16,7 +16,8 @@ public class ProgressService { public void predictCheck(final int projectId){ if(progressCacheRepository.predictCheck(projectId)){ - throw new CustomException(ErrorCode.AI_IN_PROGRESS); +// throw new CustomException(ErrorCode.AI_IN_PROGRESS); + progressCacheRepository.removePredictProgress(projectId); } } diff --git a/backend/src/main/java/com/worlabel/domain/project/service/ProjectService.java b/backend/src/main/java/com/worlabel/domain/project/service/ProjectService.java index 7d5690b..5295c2c 100644 --- a/backend/src/main/java/com/worlabel/domain/project/service/ProjectService.java +++ b/backend/src/main/java/com/worlabel/domain/project/service/ProjectService.java @@ -179,7 +179,7 @@ public class ProjectService { // 응답없음 log.debug("요청"); - progressService.registerPredictProgress(projectId); +// progressService.registerPredictProgress(projectId); List list = aiService.postRequest(endPoint, autoLabelingRequest, List.class, this::converter); saveAutoLabelList(list); }