weiweiw
/
Bonus-Cloud-AI-V2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

增加接口

This commit is contained in:
weiweiw 2024-11-28 11:24:44 +08:00
parent 72118386d0
commit 0791329e2e
4 changed files with 22 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
bonus-modules/bonus-ai/src/main/java/com/bonus/ai/domain/dataset/AnnotationTaskEntity.java
View File

@ -55,12 +55,13 @@ public class AnnotationTaskEntity extends BaseEntity {
private List<AnnotationTaskAnnotatorEntity> subAnnotationTasks; private List<AnnotationTaskAnnotatorEntity> subAnnotationTasks;
/**用于查询目的*/ /**用于查询目的*/
/**根据任务id，标注人id或审核人id 和文件标注状态返回满足条件的文件列表*/ // /**根据任务id，标注人id或审核人id 和文件标注状态返回满足条件的文件列表*/
private String annotatorId= ""; private String annotatorId= "";
private String reviewerId = ""; private String reviewerId = "";
private String fileAnnotationStatus; // private String fileAnnotationStatus;
/**根据任务id 和文件id 返回审核驳回的原因*/ // private String fileName;
private Long fileId = 0L; // /**根据任务id 和文件id 返回审核驳回的原因*/
// private Long fileId = 0L;
/**任务查询数据*/ /**任务查询数据*/
private String datasetName; private String datasetName;
private String ownerName; private String ownerName;

4
bonus-modules/bonus-ai/src/main/java/com/bonus/ai/mapper/AnnotationTaskMapper.java
View File

@ -64,9 +64,9 @@ public interface AnnotationTaskMapper
int updateAnnotationInfo(AnnotationTaskAnnotatorEntity entity); int updateAnnotationInfo(AnnotationTaskAnnotatorEntity entity);
/**根据文件标注状态，标注人或审核人等信息 返回任务关联的文件详情*/ /**根据文件标注状态，标注人或审核人等信息 返回任务关联的文件详情*/
List<DataSetBasicFileEntity>getTaskBasicFile(AnnotationTaskEntity annotationTask); List<DataSetBasicFileEntity>getTaskBasicFile(AnnotationTaskAnnotatorEntity annotationTask);
/**根据任务Id，文件id返回审核驳回的原因*/ /**根据任务Id，文件id返回审核驳回的原因*/
String getAuditFailReasonByFileId(Long taskId, Long fileId); String getAuditFailReasonByFileId(AnnotationTaskAnnotatorEntity annotationTask);
//这个接口不需要，可以扩展selectAnnotationTaskList //这个接口不需要，可以扩展selectAnnotationTaskList
// AnnotationTaskEntity selectAnnotationTaskListUUID(Long taskId); // AnnotationTaskEntity selectAnnotationTaskListUUID(Long taskId);

22
bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/dataset/AnnotationTaskServiceImpl.java
View File

@ -211,17 +211,16 @@ public class AnnotationTaskServiceImpl implements AnnotationTaskService {
@Override @Override
public List<DataSetBasicFileEntity> getMyAnnotaionTaskBasicFile(Long taskId, String annotationStatus){ public List<DataSetBasicFileEntity> getMyAnnotaionTaskBasicFile(Long taskId, String annotationStatus){
try { try {
String userId = SecurityUtils.getUserId().toString(); Long userId = SecurityUtils.getUserId();
AnnotationTaskEntity task = new AnnotationTaskEntity(); AnnotationTaskAnnotatorEntity task = new AnnotationTaskAnnotatorEntity();
task.setTaskId(taskId); task.setTaskId(taskId);
task.setAnnotatorId(userId); task.setAnnotatorId(userId);
Long intAnnotationStatus = Long.valueOf(annotationStatus); Long intAnnotationStatus = Long.valueOf(annotationStatus);
if (intAnnotationStatus < 4 && intAnnotationStatus >= 0) { if (intAnnotationStatus < 4 && intAnnotationStatus >= 0) {
task.setFileAnnotationStatus(annotationStatus); task.setAnnotationStatus(annotationStatus);
} else { } else {
task.setFileAnnotationStatus("5"); task.setAnnotationStatus("5");
} }
task.setFileAnnotationStatus(annotationStatus);
return annotationTaskMapper.getTaskBasicFile(task); return annotationTaskMapper.getTaskBasicFile(task);
}catch(Exception e){ }catch(Exception e){
e.printStackTrace(); e.printStackTrace();
@ -231,16 +230,16 @@ public class AnnotationTaskServiceImpl implements AnnotationTaskService {
@Override @Override
public List<DataSetBasicFileEntity> getMyAuditTaskBasicFile(Long taskId, String annotationStatus){ public List<DataSetBasicFileEntity> getMyAuditTaskBasicFile(Long taskId, String annotationStatus){
String userId = SecurityUtils.getUserId().toString(); Long userId = SecurityUtils.getUserId();
AnnotationTaskEntity task = new AnnotationTaskEntity(); AnnotationTaskAnnotatorEntity task = new AnnotationTaskAnnotatorEntity();
task.setTaskId(taskId); task.setTaskId(taskId);
task.setReviewerId(userId); task.setReviewerId(userId);
Long intAnnotationStatus = Long.valueOf(annotationStatus); Long intAnnotationStatus = Long.valueOf(annotationStatus);
if (intAnnotationStatus <4 && intAnnotationStatus >= 0) { if (intAnnotationStatus <4 && intAnnotationStatus >= 0) {
task.setFileAnnotationStatus(annotationStatus); task.setAnnotationStatus(annotationStatus);
} }
else { else {
task.setFileAnnotationStatus("5"); task.setAnnotationStatus("5");
} }
return annotationTaskMapper.getTaskBasicFile(task); return annotationTaskMapper.getTaskBasicFile(task);
} }
@ -249,7 +248,10 @@ public class AnnotationTaskServiceImpl implements AnnotationTaskService {
@Override @Override
public String getAuditFailReasonByFileId(Long taskId, Long fileId){ public String getAuditFailReasonByFileId(Long taskId, Long fileId){
try { try {
return annotationTaskMapper.getAuditFailReasonByFileId(taskId, fileId); AnnotationTaskAnnotatorEntity task = new AnnotationTaskAnnotatorEntity();
task.setTaskId(taskId);
task.setFileId(fileId);
return annotationTaskMapper.getAuditFailReasonByFileId(task);
}catch(Exception e){ }catch(Exception e){
e.printStackTrace(); e.printStackTrace();
return ""; return "";

6
bonus-modules/bonus-ai/src/main/resources/mapper/AnnotationTaskMapper.xml
View File

@ -241,7 +241,7 @@
</update> </update>
<!-- 根据taskid 返回由某人标注 或审核的的文件列表和文件详情--> <!-- 根据taskid 返回由某人标注 或审核的的文件列表和文件详情-->
<select id="getTaskBasicFile" parameterType="com.bonus.ai.domain.dataset.AnnotationTaskEntity" resultType="com.bonus.ai.domain.DataSetBasicFileEntity"> <select id="getTaskBasicFile" parameterType="com.bonus.ai.domain.dataset.AnnotationTaskAnnotatorEntity" resultType="com.bonus.ai.domain.DataSetBasicFileEntity">
SELECT SELECT
distinct adfm.file_id AS fileId, distinct adfm.file_id AS fileId,
abf.file_name AS fileName, abf.file_name AS fileName,
@ -264,12 +264,12 @@
AND atap.reviewer_id = #{reviewerId} AND atap.reviewer_id = #{reviewerId}
</if> </if>
<if test="fileAnnotationStatus != null and fileAnnotationStatus != '5'"> <if test="fileAnnotationStatus != null and fileAnnotationStatus != '5'">
AND atap.annotation_status = #{fileAnnotationStatus} AND atap.annotation_status = #{annotationStatus}
</if> </if>
</select> </select>
<!-- 根据taskid 和fileid 返回审核驳回的原因--> <!-- 根据taskid 和fileid 返回审核驳回的原因-->
<select id="getAuditFailReasonByFileId" parameterType="com.bonus.ai.domain.dataset.AnnotationTaskEntity" resultType="String"> <select id="getAuditFailReasonByFileId" parameterType="com.bonus.ai.domain.dataset.AnnotationTaskAnnotatorEntity" resultType="String">
SELECT SELECT
atap.audit_failed_reason atap.audit_failed_reason
FROM ai_basic_file abf FROM ai_basic_file abf