Merge branch 'master' of http://10.1.235.20:3000/asiainfo/ebc

monitor-manage-service/src/main/java/com/ai/bss/monitorManage/service/impl/KafkaProcessImpl.java

@@ -36,6 +36,8 @@ import com.alibaba.fastjson.JSONObject;
import lombok.extern.slf4j.Slf4j;
import javax.sound.sampled.Line;
@Slf4j
@Service
public class KafkaProcessImpl implements KafkaProcess {
@@ -129,7 +131,6 @@ public class KafkaProcessImpl implements KafkaProcess {
		}
		// 文件上传到 minio文件服务器
		multipartFile = new CommonsMultipartFile(FileUtils.createFileItem(new File(filePath), fileName));
		log.info("保存图片文件："+fileName);
		minioFileMap = fileManageService.uploadFile(multipartFile, bucketName, deviceCode, null);
@@ -291,13 +292,13 @@ public class KafkaProcessImpl implements KafkaProcess {
	 */
	private AiIdenLogDto aiHelmetDetect(AiIdenLogDto aiIdenLogDto, String aifilePath) {
		try {
			log.info("开始AI处理...");
			log.info("开始AI帽子识别处理...");
			// 安全帽识别
			String aiResponseData = ebcAiService.helmetDetect(ImageBase64Converter.convertFileToBase64(aifilePath),
					Strings.EMPTY, Strings.EMPTY);
//			String aiResponseData = ebcAiService.helmetDetect(ImageBase64Converter.convertInputStreamToBase64(multipartFile.getInputStream()), Strings.EMPTY, Strings.EMPTY);
			JSONObject jsonObject = JSONObject.parseObject(aiResponseData);// JSONUtil.formatJsonStr(aiResponseData);
			System.out.println("AI 识别结果： " + jsonObject);
			System.out.println("AI帽子识别结果： " + jsonObject);
			IdenResultDto idenResultDto = null;
			List<IdenResultDto> idenResultDtoList = null;
			if (jsonObject.getString("Status").equals("Ok")) {
@@ -342,7 +343,7 @@ public class KafkaProcessImpl implements KafkaProcess {
			// AI模型
//			aiIdenLogDto.setAiIdenModel(MonitorManageConsts.AI_MODEL_CLOTHING_CODE);
		} catch (Exception e) {
			log.error("AI处理异常");
			log.error("AI处理异常，帽子识别");
		}
		return aiIdenLogDto;
	}
@@ -375,7 +376,6 @@ public class KafkaProcessImpl implements KafkaProcess {
			log.info("开始AI人数识别处理...");
			String aiResponseData = ebcAiService.headDetect(ImageBase64Converter.convertFileToBase64(aifilePath), true,
					0.9f);
			JSONObject jsonObject = JSONObject.parseObject(aiResponseData);
			JSONArray jsonArray = jsonObject.getJSONArray("Result");
			if (!CollectionUtils.isEmpty(jsonArray)) {
@@ -384,9 +384,8 @@ public class KafkaProcessImpl implements KafkaProcess {
			} else {
				log.info("AI人数识别结果：没有人");
			}
		} catch (Exception e) {
			log.error("AI处理异常");
			log.error("AI处理异常，人数识别");
		}
		return isHuman;
	}
@@ -412,12 +411,10 @@ public class KafkaProcessImpl implements KafkaProcess {
				if (resultJsonObject.getIntValue("is_match") == 0) {
					aiIdenLogDto.setIdenResult(jsonObject.getString("result"));
					aiIdenLogDto.setIdenResultType(MonitorManageConsts.AI_IDENTIFY_RESULT);
				} else {
					aiIdenLogDto.setIdenResult(jsonObject.getString("result"));
					aiIdenLogDto.setRelateEmployeeRoleId(resultJsonObject.getString("user_id"));
					aiIdenLogDto.setIdenResultType(MonitorManageConsts.AI_IDENTIFY_RESULT_ATTENDANCE);
				}
				log.info("识别成功，推送KAFKA消息");
@@ -427,7 +424,7 @@ public class KafkaProcessImpl implements KafkaProcess {
				return aiIdenLogDto;
			}
		} catch (Exception e) {
			log.error("AI处理异常");
			log.error("AI处理异常，人脸识别");
		}
		return aiIdenLogDto;
	}
@@ -468,16 +465,17 @@ public class KafkaProcessImpl implements KafkaProcess {
                                    aiIdenLogDto.setIdenResultType(MonitorManageConsts.AI_IDENTIFY_RESULT);
                                }
                        }
                        log.info("识别成功，推送KAFKA消息");
						log.info("吸烟识别成功，推送KAFKA消息");
                        processKafka(aiIdenLogDto.getTopic(), JSONObject.toJSONString(aiIdenLogDto));
                    	log.info("消息推送完成");
                    }else{
                        log.info("没有识别到吸烟者，不推送KAFKA消息");
                    }
            }else{
            	log.error("识别异常返回");
            	log.error("吸烟识别识别异常");
			}
        }catch(Exception e){
        log.error("AI处理异常");
        log.error("AI处理异常，吸烟识别");
        }
        return aiIdenLogDto;
    }
@@ -491,17 +489,16 @@ private AiIdenLogDto workClothesDetect(AiIdenLogDto aiIdenLogDto, String aifileP
		//工作服识别
		String aiResponseData=ebcAiService.workClothesDetect(ImageBase64Converter.convertFileToBase64(aifilePath),true,true,0.9f);
		JSONObject jsonObject = JSONObject.parseObject(aiResponseData);
		System.out.print("AI识别结果："+aiResponseData);
		System.out.print("AI工作服识别结果："+aiResponseData);
		IdenResultDto idenResultDto = null;
		List<IdenResultDto> idenResultDtoList = null;
		if(jsonObject.getString("Status").equals("Ok")){
			idenResultDtoList = new ArrayList<>();
			JSONArray jsonArray = JSONArray.parseArray(jsonObject.getString("Result"));
			if(jsonArray.size()>0){
				for (int i = 0; i < jsonArray.size(); i++) {
					JSONObject resultObj = JSONObject.parseObject(jsonArray.get(i).toString());
					JSONObject jsonArray1 = JSONObject.parseObject(resultObj.getString("clothes"));
					String aiClassName =jsonArray1.getString("class");
			List<Map> resultList = JSONArray.parseArray(jsonObject.getString("Result"),Map.class);
			if(resultList.size()>0){
				for (int i = 0; i < resultList.size(); i++) {
					JSONObject resultObject = JSONObject.parseObject(resultList.get(i).get("clothes").toString());
					String aiClassName =resultObject.getString("class");
					if(aiClassName.equals("qita")){
						idenResultDto = new IdenResultDto();
						idenResultDto.setAlarmEmployeeRoleId(MonitorManageConsts.alarmEmployeeRoleId);
@@ -524,17 +521,18 @@ private AiIdenLogDto workClothesDetect(AiIdenLogDto aiIdenLogDto, String aifileP
						log.info("识别到工作服,工作服类型为："+aiClassName);
					}
				}
				log.info("识别成功，推送KAFKA消息");
				log.info("工作服识别成功，推送KAFKA消息");
				processKafka(aiIdenLogDto.getTopic(), JSONObject.toJSONString(aiIdenLogDto));
				log.info("消息推送完成");
			}else {
				log.info("没有识别到人，不推送KAFKA消息");
			}
		}else if(jsonObject.getString("Status").equals("ERROR")){
			log.error("异常信息为："+jsonObject.getString("Des"));
			log.error("工作服识别异常："+jsonObject.getString("Des"));
		}
	}catch(Exception e){
	log.error("AI处理异常");
	log.error("AI处理异常，工作服识别");
	}
	return aiIdenLogDto;
}
@@ -548,17 +546,18 @@ private AiIdenLogDto instrumentDetect(AiIdenLogDto aiIdenLogDto, String aifilePa
		//仪表识别
		String aiResponseData=ebcAiService.instrumentDetect(ImageBase64Converter.convertFileToBase64(aifilePath),Strings.EMPTY,Strings.EMPTY,Strings.EMPTY);
		JSONObject jsonObject = JSONObject.parseObject(aiResponseData);
		System.out.print("AI识别结果："+aiResponseData);
		JSONObject jsonObjectpredictions=JSONObject.parseObject(jsonObject.getString("predictions"));
		System.out.print("AI仪表识别结果："+aiResponseData);
		JSONObject jsonObjectpredictions=jsonObject.getJSONObject("predictions");
		if(jsonObjectpredictions.getString("Errors").isEmpty()){
			if(jsonObjectpredictions.getInteger("Nums")>0){
				log.info("识别到的仪表的数量为："+jsonObjectpredictions.getInteger("Nums"));
				aiIdenLogDto.setIdenResultType(MonitorManageConsts.AI_IDENTIFY_RESULT);
				aiIdenLogDto.setRelateEmployeeRoleId(MonitorManageConsts.alarmEmployeeRoleId);
				log.info("返回的仪表信息为："+JSON.toJSONString(jsonObject));
				aiIdenLogDto.setIdenResult(JSON.toJSONString(jsonObject));
				log.info("识别成功，推送KAFKA消息");
				log.info("返回的仪表信息为："+jsonObjectpredictions.getString("Items"));
				aiIdenLogDto.setIdenResult(jsonObjectpredictions.getString("Items"));
				log.info("仪表识别成功，推送KAFKA消息");
				processKafka(aiIdenLogDto.getTopic(), JSONObject.toJSONString(aiIdenLogDto));
				log.info("消息推送完成");
			}else{
				log.info("仪表数量为0，不推送消息");
			}
@@ -566,13 +565,10 @@ private AiIdenLogDto instrumentDetect(AiIdenLogDto aiIdenLogDto, String aifilePa
			log.info("异常返回，异常原因为:"+jsonObjectpredictions.getString("Errors"));
		}
	}catch(Exception e){
		log.error("AI处理异常");
		log.error("AI处理异常，仪表识别");
	}
	return aiIdenLogDto;
}
	/**
	 * 根据设备code获取监控任务详情
	 * @param deviceCode
@@ -580,11 +576,9 @@ private AiIdenLogDto instrumentDetect(AiIdenLogDto aiIdenLogDto, String aifilePa
	 */
	private List<Map> getResourceToolAllInfo(String deviceCode) {
		List<Map> resourceToolInfoList = new ArrayList<Map>();
		try {
			String url = protectionResourceUrl + "?resourceToolCode=" + deviceCode;
			log.info("getResourceToolAllInfo method params: resourceToolCode=" + deviceCode + ", url=" + url);
			String resultJson = HttpServiceUtil.sendRequest(url);
			JSONObject resultData = JSON.parseObject(resultJson);
			if (StringUtils.isEmpty(resultData.getString("data"))) {