zgyw.git - Gitblit

New file
			@@ -0,0 +1,146 @@
			package com.ycl.calculate;

			import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
			import com.ycl.platform.domain.entity.CheckIndexCar;
			import com.ycl.platform.domain.entity.TMonitor;
			import com.ycl.platform.domain.result.HK.SnapshotDataMonitorResult;
			import com.ycl.platform.mapper.CheckIndexCarMapper;
			import com.ycl.platform.service.ICheckIndexCarService;
			import com.ycl.platform.service.ITMonitorService;
			import com.ycl.system.mapper.SysConfigMapper;
			import constant.ApiConstants;
			import constant.CheckConstants;
			import constant.CheckSnapCountConstants;
			import org.springframework.beans.factory.annotation.Autowired;
			import org.springframework.stereotype.Component;
			import org.springframework.util.CollectionUtils;
			import utils.DateUtils;

			import java.math.BigDecimal;
			import java.math.RoundingMode;
			import java.util.*;
			import java.util.function.Function;
			import java.util.stream.Collectors;

			/**
			* 计算车辆、人脸点位在线率、视图库对接稳定性
			* 获取分省厅、区域的map<k,v> k为deptId或者Province_deptId
			* 循环map计算点位在线率
			* 更新或新增
			*/
			@Component
			public class CarSiteOnlineCalculation implements CalculationStrategy<SnapshotDataMonitorResult> {
			@Autowired
			private CheckIndexCarMapper checkIndexCarMapper;
			@Autowired
			private SysConfigMapper sysConfigMapper;
			@Autowired
			private ITMonitorService monitorService;
			@Autowired
			private ICheckIndexCarService checkIndexCarService;

			//区域车辆点位在线指标的内部类
			private static class AreaStats {
			int totalSites = 0;
			int offlineSites = 0;
			int totalDataSum = 0;
			}

			@Override
			public void calculate(List<SnapshotDataMonitorResult> list) {
			if (CollectionUtils.isEmpty(list)) {
			return;
			}
			Map<String, AreaStats> areaStatsMap = new HashMap<>();
			Map<String, TMonitor> monitorMap = monitorService.list(new QueryWrapper<TMonitor>()
			.in("serial_number", list.stream().map(SnapshotDataMonitorResult::getExternalIndexCode).collect(Collectors.toList())))
			.stream().collect(Collectors.toMap(TMonitor::getSerialNumber, Function.identity()));

			// TODO: 分省厅市局需要补充集合数据
			List<String> provinceIds = new ArrayList<>(); // 这里需要根据点位补充

			for (SnapshotDataMonitorResult result : list) {
			TMonitor monitor = monitorMap.get(result.getExternalIndexCode());
			if (monitor == null) continue;

			String deptId = monitor.getDeptId().toString();
			updateAreaStats(areaStatsMap, deptId, result);

			// 处理省厅数据
			if (!CollectionUtils.isEmpty(provinceIds) && provinceIds.contains(monitor.getSerialNumber())) {
			String provinceKey = "Province_" + deptId;
			updateAreaStats(areaStatsMap, provinceKey, result);
			}
			}

			//获取2022同期抓拍平均值省厅、市局
			BigDecimal cityCountAvg = getAverageCount(CheckSnapCountConstants.City);
			BigDecimal countyCountAvg = getAverageCount(CheckSnapCountConstants.County);

			// 查询今日数据
			List<CheckIndexCar> checkIndexCarList = checkIndexCarMapper.selectToday(DateUtils.getDate());
			List<CheckIndexCar> checkIndexCars = new ArrayList<>();
			areaStatsMap.forEach((deptId, stats) -> {
			if (stats.totalSites > 0) {
			CheckIndexCar checkIndexCar = createOrUpdateCheckIndexCar(deptId, stats, cityCountAvg, countyCountAvg, checkIndexCarList);
			checkIndexCars.add(checkIndexCar);
			}
			});

			checkIndexCarService.saveOrUpdateBatch(checkIndexCars);
			}

			//累计总点位数、离线数、总抓拍量
			private void updateAreaStats(Map<String, AreaStats> areaStatsMap, String key, SnapshotDataMonitorResult result) {
			//返回对象的引用，如果不存在会放入新的key,value
			AreaStats stats = areaStatsMap.computeIfAbsent(key, k -> new AreaStats());
			stats.totalSites++;
			if (ApiConstants.HK_SnapCount_ResultType_Null == result.getResultType()) {
			stats.offlineSites++;
			} else {
			stats.totalDataSum += result.getDataCount();
			}
			}

			//获取2022同期抓拍平均值
			private BigDecimal getAverageCount(String configKey) {
			String count = sysConfigMapper.checkConfigKeyUnique(configKey).getConfigValue();
			return new BigDecimal(count).multiply(new BigDecimal(CheckSnapCountConstants.Multiply))
			.divide(new BigDecimal(CheckSnapCountConstants.CountyNum), 0, RoundingMode.HALF_UP);
			}

			//车辆点位在线率和视图库对接稳定性
			private CheckIndexCar createOrUpdateCheckIndexCar(String key, AreaStats stats, BigDecimal cityCountAvg, BigDecimal countyCountAvg, List<CheckIndexCar> checkIndexCarList) {
			CheckIndexCar checkIndexCar;

			// 检查是否已存在今日数据
			Optional<CheckIndexCar> existingCar = checkIndexCarList.stream()
			.filter(car -> key.equals(car.getDeptId().toString()) &&
			(key.startsWith("Province_") ? CheckConstants.Examine_Tag_City.equals(car.getExamineTag())
			: CheckConstants.Examine_Tag_County.equals(car.getExamineTag())))
			.findFirst();

			if (existingCar.isPresent()) {
			checkIndexCar = existingCar.get();
			} else {
			checkIndexCar = new CheckIndexCar();
			checkIndexCar.setDeptId(key.startsWith("Province_") ? Long.parseLong(key.split("_")[1]) : Long.parseLong(key));
			checkIndexCar.setExamineTag(key.startsWith("Province_") ? CheckConstants.Examine_Tag_City : CheckConstants.Examine_Tag_County);
			checkIndexCar.setCreateTime(new Date());
			}
			//调用点位在线计算方法
			Map<String, Object> siteOnlineParam = new HashMap<>();
			siteOnlineParam.put("totalSites", stats.totalSites);
			siteOnlineParam.put("offlineSites", stats.offlineSites);
			BigDecimal siteOnline = IndexCalculationUtils.siteOnline(siteOnlineParam);
			checkIndexCar.setSiteOnline(siteOnline);
			//视图库对接稳定性
			BigDecimal avgCount = key.startsWith("Province_") ? cityCountAvg : countyCountAvg;
			Map<String, Object> viewConnectParam = new HashMap<>();
			viewConnectParam.put("totalDataSum", stats.totalDataSum);
			viewConnectParam.put("avgCount", avgCount);
			BigDecimal viewConnectStability = IndexCalculationUtils.viewConnectStability(viewConnectParam);
			checkIndexCar.setViewConnectStability(viewConnectStability);
			return checkIndexCar;
			}
			}