修改AI报警的问题

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

@@ -3,6 +3,7 @@ package com.ai.bss.security.protection.controller;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.commons.lang.StringUtils;
@@ -56,8 +57,6 @@ public class AiAlarmManageController {
		// 每页条数
		int pageSize = aiQuery.getPageSize() < 1 ? EbcConstant.DEFAULT_PAGE_SIZE : aiQuery.getPageSize();
		ArrayList<Object> statusList = new ArrayList<>();
		statusList.add(aiQuery.getStatusList());
		Map<String, Object> params = new HashMap<>();
		params.put("workOrgRoleId", aiQuery.getWorkOrgRoleId());
		params.put("alarmTypeCode", aiQuery.getAlarmTypeCode());
@@ -65,7 +64,12 @@ public class AiAlarmManageController {
		params.put("workEmployeeRoleId", aiQuery.getWorkEmployeeRoleId());
		params.put("beginTime", aiQuery.getBeginTime());
		params.put("endTime", aiQuery.getEndTime());
		params.put("statusList",statusList);
		
		if (StringUtils.isNotBlank(aiQuery.getStatusList())) {
			ArrayList<Object> statusList = new ArrayList<>();
			statusList.add(aiQuery.getStatusList());
			params.put("statusList",statusList);
		}
		CommonResponse<PageBean<Map<String, Object>>> commonResponse = aiAlarmManageService.queryPageAiAlarm(params,
				pageNumber, pageSize);
@@ -120,4 +124,18 @@ public class AiAlarmManageController {
		return aiAlarmManageService.queryOneAiAlarmInfo(workTaskId);
	}
	/**
	 * 查询AI告警信息数量
	 * @param workTaskId
	 * @return
	 * @throws Exception
	 */
	@ResponseBody
	@RequestMapping("/queryAiAlarmCount")
	public CommonResponse<List<Map<String, Object>>> queryAiAlarmCount() throws Exception {
		Map<String, Object> params=new HashMap<String, Object>();
		List<Map<String, Object>> resultList=aiAlarmManageService.queryAiAlarmCount(params);
		return CommonResponse.ok(resultList);
	}
	
}

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

@@ -1,5 +1,6 @@
package com.ai.bss.security.protection.service.impl;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@@ -176,4 +177,36 @@ public class AiAlarmManageServiceImpl implements AiAlarmManageService {
		resultMap.put("videoInfo", ebcMonitorVideoLog);
		return CommonResponse.ok(resultMap);
	}
	@Override
	public List<Map<String,Object>> queryAiAlarmCount(Map<String, Object> params) throws Exception {
		CommonRequest<Map<String, Object>> request=new CommonRequest<Map<String,Object>>(params);
        CommonResponse<List<Map<String,Object>>> response = aiTaskQuery.safetyAlarmStatusStatistics(request);
		
        List<Map<String,Object>> resultList=new ArrayList<Map<String,Object>>();
        long iniCount=0;
        long runCount=0;
        
        if (response!=null&&!CollectionUtils.isEmpty(response.getData())) {
	        for (Map<String, Object> map : response.getData()) {
	        	if (WorkTask.INIT_STATUS.equals(String.valueOf(map.get("status")))) {
	        		iniCount=(long)map.get("cn");
				}else if (WorkTask.RUNNING_STATUS.equals(String.valueOf(map.get("status")))){
					runCount=(long)map.get("cn");
				}
			}
        }
        
        Map<String, Object> map=new HashMap<String, Object>();
        map.put("name", "ini");
        map.put("value", iniCount);
        resultList.add(map);
        
        map=new HashMap<String, Object>();
        map.put("name", "run");
        map.put("value", iniCount+runCount);
        resultList.add(map);
        
        return resultList;
	}
}

security-protection-service/src/main/java/com/ai/bss/security/protection/service/interfaces/AiAlarmManageService.java

@@ -32,4 +32,12 @@ public interface AiAlarmManageService {
	 * @return
	 */
	CommonResponse<List<Map<String, Object>>> queryAiTaskByDevice(String resourceToolId, String aiIdenModel);
	
	/**
	 * 查询AI告警信息数量
	 * @param params
	 * @return
	 * @throws Exception
	 */
	List<Map<String, Object>> queryAiAlarmCount(Map<String, Object> params) throws Exception;
}