package com.ycl.calculate;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.ycl.platform.domain.entity.CheckIndexVideo;
|
import com.ycl.platform.domain.entity.TMonitor;
|
import com.ycl.platform.domain.result.HK.SnapshotDelayMonitorResult;
|
import com.ycl.platform.domain.result.UY.QueryVqdResult;
|
import com.ycl.platform.mapper.CheckIndexVideoMapper;
|
import com.ycl.platform.service.ICheckIndexFaceService;
|
import com.ycl.platform.service.ICheckIndexVideoService;
|
import com.ycl.platform.service.ITMonitorService;
|
import constant.ApiConstants;
|
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.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
|
* 循环map计算数据上传及时性
|
* 更新或新增
|
*/
|
@Component
|
//TODO：重点指挥图像在线率
|
public class VideoOnlineCalculation extends IndexCalculationServe implements CalculationStrategy<QueryVqdResult> {
|
@Autowired
|
private CheckIndexVideoMapper checkIndexVideoMapper;
|
@Autowired
|
private ITMonitorService monitorService;
|
@Autowired
|
private ICheckIndexVideoService checkIndexVideoService;
|
|
//区域视频在线率的内部类
|
private static class AreaStats {
|
int totalSites = 0;
|
int onlineSites = 0;
|
int importantSites = 0;
|
int importantOnlineSites = 0;
|
}
|
|
@Override
|
public void calculate(List<QueryVqdResult> list) {
|
if (CollectionUtils.isEmpty(list)) {
|
return;
|
}
|
|
//获得国标码为key的设备map
|
Map<String, TMonitor> monitorMap = monitorService.list(new QueryWrapper<TMonitor>()
|
.in("serial_number", list.stream().map(QueryVqdResult::getDeviceId).collect(Collectors.toList())))
|
.stream().collect(Collectors.toMap(TMonitor::getSerialNumber, Function.identity()));
|
//获取省厅国标码集合
|
List<String> provinceIds = getProvince();
|
//重点点位集合
|
List<String> important = getImportant();
|
|
Map<String, AreaStats> areaStatsMap = new HashMap<>();
|
for (QueryVqdResult result : list) {
|
TMonitor monitor = monitorMap.get(result.getDeviceId());
|
if (monitor == null) continue;
|
|
String deptId = monitor.getDeptId().toString();
|
updateAreaStats(areaStatsMap, deptId, result, important);
|
|
// 处理省厅数据
|
if (!CollectionUtils.isEmpty(provinceIds) && provinceIds.contains(monitor.getSerialNumber())) {
|
String provinceKey = "Province_" + deptId;
|
updateAreaStats(areaStatsMap, provinceKey, result, important);
|
}
|
}
|
|
// 查询是否index表已经存在今日数据
|
List<CheckIndexVideo> checkIndexVideoList = checkIndexVideoMapper.selectToday(DateUtils.getDate());
|
List<CheckIndexVideo> checkIndexVideos = new ArrayList<>();
|
areaStatsMap.forEach((deptId, stats) -> {
|
if (stats.totalSites > 0) {
|
CheckIndexVideo CheckIndexVideo = createOrUpdateCheckIndexFace(deptId, stats, checkIndexVideoList);
|
if (CheckIndexVideo != null) {
|
checkIndexVideos.add(CheckIndexVideo);
|
}
|
}
|
});
|
|
checkIndexVideoService.saveOrUpdateBatch(checkIndexVideos);
|
}
|
|
/**
|
* 累计总点位数、在线点位数、重点点位数、重点点位在线数
|
*/
|
private void updateAreaStats(Map<String, AreaStats> areaStatsMap, String key, QueryVqdResult result, List<String> important) {
|
//返回对象的引用，如果不存在会放入新的key,value
|
AreaStats stats = areaStatsMap.computeIfAbsent(key, k -> new AreaStats());
|
stats.totalSites++;
|
if(important.contains(result.getDeviceId())){
|
stats.importantSites++;
|
}
|
if (ApiConstants.UY_OnlineSite_Online.equals(result.getOnlineStatus())){
|
stats.onlineSites++;
|
if(important.contains(result.getDeviceId())){
|
stats.importantOnlineSites++;
|
}
|
}
|
}
|
|
/**
|
* 视频点位在线率
|
*/
|
private CheckIndexVideo createOrUpdateCheckIndexFace(String key, AreaStats stats, List<CheckIndexVideo> checkIndexVideoList) {
|
CheckIndexVideo CheckIndexVideo = getCheckIndex(key, checkIndexVideoList, CheckIndexVideo.class);
|
if (CheckIndexVideo == null) {
|
return null;
|
}
|
//调用点位在线率计算方法
|
Map<String, Object> param = new HashMap<>();
|
param.put("totalSites", stats.totalSites);
|
param.put("onlineSites", stats.onlineSites);
|
BigDecimal siteOnline = siteOnline(param);
|
CheckIndexVideo.setSiteOnline(siteOnline);
|
//调用点位在线率计算方法 计算重点点位在线率
|
Map<String, Object> importantParam = new HashMap<>();
|
param.put("totalSites", stats.importantSites);
|
param.put("onlineSites", stats.importantOnlineSites);
|
BigDecimal importantSiteOnline = siteOnline(importantParam);
|
CheckIndexVideo.setKeySiteOnline(importantSiteOnline);
|
return CheckIndexVideo;
|
}
|
}
|
