AI任务查询修改

security-protection-service/src/main/java/com/ai/bss/security/protection/controller/SysConfigAiTaskController.java

@@ -18,6 +18,7 @@ import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@@ -53,13 +54,16 @@ public class SysConfigAiTaskController {
		int pageNumber = aiQuery.getPageNumber() < 1 ? 1 : aiQuery.getPageNumber();
		// 每页条数
		int pageSize = aiQuery.getPageSize() < 1 ? EbcConstant.DEFAULT_PAGE_SIZE : aiQuery.getPageSize();
		Map<String, Object> params = new HashMap<>();
		if(null!=aiQuery.getAiTaskStatus()&&!"".equals(aiQuery.getAiTaskStatus())){
			List<String> statusList = Arrays.asList(aiQuery.getAiTaskStatus());
			params.put("statusList",statusList);
		}
		params.put("workOrgId",aiQuery.getWorkOrgId());
		params.put("name",aiQuery.getName());
		params.put("nameAsLike",aiQuery.getNameAsLike());
		params.put("aiIdenModel",aiQuery.getAiIdenModel());
		params.put("statusList",aiQuery.getStatusList());
		CommonResponse<PageBean<Map<String, Object>>> commonResponse = sysConfigAiTaskService.queryPageAiTask(params, pageNumber, pageSize);

security-protection-service/src/main/java/com/ai/bss/security/protection/model/AiQuery.java

@@ -57,5 +57,5 @@ public class AiQuery {
    private String relateEmployeeNameAsLike; // 员工名称（模糊匹配）
    private String aiTaskStatus; // AI任务执行状态
}

security-protection-service/src/main/java/com/ai/bss/security/protection/service/impl/SysConfigAiTaskServiceImpl.java

@@ -36,7 +36,19 @@ public class SysConfigAiTaskServiceImpl implements SysConfigAiTaskService {
    @Override
    public CommonResponse<PageBean<Map<String, Object>>> queryPageAiTask(Map<String, Object> params, int pageNumber, int pageSize) {
        CommonResponse<List<AiModel>> aiModelList = aiTaskQuery.findAiModelList(new CommonRequest<Void>(null));
        CommonResponse<PageBean<Map<String, Object>>> queryResult = aiTaskQuery.queryAiTask(new CommonRequest<>(params, pageNumber, pageSize));
        List<Map<String, Object>> aiTaskList = queryResult.getData().getData();
        if(null!=aiTaskList&&aiTaskList.size()>0){
            aiTaskList.forEach((aiTask)->{
                aiModelList.getData().forEach((aiModel)->{
                    if(null!=aiTask.get("aiIdenModel")&& aiTask.get("aiIdenModel").equals(aiModel.getAiIdenModel() )){
                        aiTask.put("aiIdenModel", aiModel.getName());
                    }
                });
            });
        }
        return queryResult;
    }