diff --git a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/dataset/AnnotationTaskServiceImpl.java b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/dataset/AnnotationTaskServiceImpl.java index 449cbf9..2a91269 100644 --- a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/dataset/AnnotationTaskServiceImpl.java +++ b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/dataset/AnnotationTaskServiceImpl.java @@ -239,9 +239,15 @@ public class AnnotationTaskServiceImpl implements AnnotationTaskService { AnnotationTaskEntity task = new AnnotationTaskEntity(); task.setTaskId(taskId); task.setReviewerId(String.valueOf(userId)); - Long intAnnotationStatus = Long.valueOf(annotationStatus); - if (intAnnotationStatus <4 && intAnnotationStatus >= 0) { - task.setFileAnnotationStatus(annotationStatus); + if ("1".equals(annotationStatus)) { + //已审核文件列表 + task.setFileAnnotationStatus("2"); + }else if ("2".equals(annotationStatus)){ + //未审核(已标注)文件列表 + task.setFileAnnotationStatus("1"); + }else if ("3".equals(annotationStatus)){ + //审核驳回 + task.setFileAnnotationStatus("3"); } return annotationTaskMapper.getTaskBasicFile(task); } diff --git a/bonus-modules/bonus-ai/src/main/resources/mapper/AnnotationTaskMapper.xml b/bonus-modules/bonus-ai/src/main/resources/mapper/AnnotationTaskMapper.xml index fe6f9ae..a48a30f 100644 --- a/bonus-modules/bonus-ai/src/main/resources/mapper/AnnotationTaskMapper.xml +++ b/bonus-modules/bonus-ai/src/main/resources/mapper/AnnotationTaskMapper.xml @@ -265,7 +265,7 @@ AND atap.reviewer_id = #{reviewerId} - + AND atap.annotation_status = #{fileAnnotationStatus}