package com.ycl.calculate;
|
|
import com.alibaba.fastjson2.JSONArray;
|
import com.ycl.platform.domain.entity.CheckIndexFace;
|
import com.ycl.platform.domain.entity.TMonitor;
|
import com.ycl.platform.domain.vo.TMonitorVO;
|
import com.ycl.platform.mapper.CheckIndexFaceMapper;
|
import com.ycl.platform.mapper.TMonitorMapper;
|
import com.ycl.platform.service.ICheckIndexFaceService;
|
import constant.ApiConstants;
|
import constant.CheckConstants;
|
import constant.RedisConstant;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.redis.core.RedisTemplate;
|
import org.springframework.stereotype.Component;
|
import org.springframework.util.CollectionUtils;
|
import utils.DateUtils;
|
import utils.StringUtils;
|
|
import java.math.BigDecimal;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.function.Function;
|
import java.util.stream.Collectors;
|
|
/**
|
* 计算车辆目录一致率
|
* 获取分省厅、区域的map<k,v> k为deptId或者Province_deptId
|
* 更新或新增
|
*/
|
@Component
|
public class FaceConsistentCalculation extends IndexCalculationServe implements CalculationStrategy<TMonitorVO> {
|
@Autowired
|
private CheckIndexFaceMapper checkIndexFaceMapper;
|
@Autowired
|
private TMonitorMapper monitorMapper;
|
@Autowired
|
private ICheckIndexFaceService checkIndexFaceService;
|
@Autowired
|
private RedisTemplate redisTemplate;
|
|
//区域车辆信息采集准确率的内部类
|
private static class AreaStats {
|
int totalSites = 0;
|
int newSites = 0;
|
}
|
|
@Override
|
public void calculate(List<TMonitorVO> list) {
|
if (CollectionUtils.isEmpty(list)) {
|
return;
|
}
|
//返回以国标码为key的设备map
|
Map<String, TMonitorVO> monitorMap = new HashMap<>();
|
if(!CollectionUtils.isEmpty(list)){
|
monitorMap = list.stream().collect(Collectors.toMap(TMonitorVO::getSerialNumber, Function.identity()));
|
}
|
//获取省厅国标码集合
|
List<String> provinceIds = getProvince();
|
//未注册车辆设备
|
Map<String,TMonitor> newMonitorMap = new HashMap<>();
|
//Mongo一机一档同步Mysql时放入Redis 筛选出车辆档案
|
String json = (String) redisTemplate.opsForValue().get(RedisConstant.New_Monitor_Set);
|
if (!StringUtils.isEmpty(json)) {
|
List<TMonitor> newMonitors = JSONArray.parseArray(json, TMonitor.class);
|
if(!CollectionUtils.isEmpty(newMonitors)){
|
newMonitorMap = newMonitors.stream().filter(tMonitor -> tMonitor.getCameraFunType().contains(CheckConstants.Rule_Category_Face+"")).collect(Collectors.toMap(TMonitor::getSerialNumber, Function.identity()));
|
}
|
}
|
Map<String, AreaStats> areaStatsMap = new HashMap<>();
|
for (TMonitorVO result : list) {
|
TMonitorVO monitor = monitorMap.get(result.getSerialNumber());
|
if (monitor == null) continue;
|
String deptId = monitor.getDeptId().toString();
|
updateAreaStats(areaStatsMap, deptId, result,newMonitorMap);
|
|
// 处理省厅数据
|
if (!CollectionUtils.isEmpty(provinceIds) && provinceIds.contains(monitor.getSerialNumber())) {
|
String provinceKey = ApiConstants.Province + deptId;
|
updateAreaStats(areaStatsMap, provinceKey, result,newMonitorMap);
|
}
|
}
|
|
// 查询是否index表已经存在今日数据
|
List<CheckIndexFace> checkIndexFaceList = checkIndexFaceMapper.selectToday(DateUtils.getDate());
|
List<CheckIndexFace> checkIndexFaces = new ArrayList<>();
|
areaStatsMap.forEach((deptId, stats) -> {
|
if (stats.totalSites > 0) {
|
CheckIndexFace checkIndexFace = createOrUpdateCheckIndexFace(deptId, stats, checkIndexFaceList);
|
if (checkIndexFace != null) {
|
checkIndexFaces.add(checkIndexFace);
|
}
|
}
|
});
|
|
checkIndexFaceService.saveOrUpdateBatch(checkIndexFaces);
|
}
|
|
/**
|
* 累计人脸目录一致率
|
*/
|
private void updateAreaStats(Map<String, AreaStats> areaStatsMap, String key, TMonitorVO result,Map<String,TMonitor> newMonitorMap) {
|
//返回对象的引用，如果不存在会放入新的key,value
|
AreaStats stats = areaStatsMap.computeIfAbsent(key, k -> new AreaStats());
|
stats.totalSites++;
|
if(newMonitorMap.containsKey(result.getSerialNumber())){
|
stats.newSites++;
|
}
|
}
|
|
/**
|
* 人脸信息采集正确率
|
*/
|
private CheckIndexFace createOrUpdateCheckIndexFace(String key, AreaStats stats, List<CheckIndexFace> checkIndexFaceList) {
|
CheckIndexFace checkIndexFace = getCheckIndex(key, checkIndexFaceList, CheckIndexFace.class);
|
if (checkIndexFace == null) {
|
return null;
|
}
|
//调用目录一致率计算方法
|
Map<String, Object> param = new HashMap<>();
|
param.put("totalSites", stats.totalSites);
|
param.put("newSites", stats.newSites);
|
BigDecimal directoryConstant = directoryConstant(param);
|
checkIndexFace.setDeviceDirectoryConsistent(directoryConstant);
|
return checkIndexFace;
|
}
|
|
|
}
|
