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}