package com.ycl.calculate;
|
|
|
import com.ycl.platform.domain.result.BaseResult;
|
import com.ycl.platform.domain.vo.TMonitorVO;
|
import com.ycl.platform.mapper.TMonitorMapper;
|
import constant.ApiConstants;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
import org.springframework.util.CollectionUtils;
|
|
import java.util.*;
|
import java.util.function.Function;
|
import java.util.stream.Collectors;
|
|
/**
|
* 抽出各区域统计数量方法
|
*/
|
@Component
|
@Slf4j
|
public abstract class IndexCalculationServe <T extends BaseResult, S> extends IndexCalculationUtils {
|
@Autowired
|
private TMonitorMapper monitorMapper;
|
|
//抽象方法，由子类编写具体逻辑算法
|
protected abstract void updateAreaStats(Map<String, S> areaStatsMap, String key, T result);
|
|
protected Map<String, S> getAreaStatsMap(List<T> list){
|
if (CollectionUtils.isEmpty(list)) {
|
log.info("数据为空");
|
return null;
|
}
|
//返回以国标码为key的设备map
|
Map<String, TMonitorVO> monitorMap = monitorMapper.selectListByIds(list.stream().map(BaseResult::getNo).collect(Collectors.toList()))
|
.stream().collect(Collectors.toMap(TMonitorVO::getSerialNumber, Function.identity()));
|
Map<String, S> areaStatsMap = new HashMap<>();
|
|
for (T result : list) {
|
//获取设备所属部门
|
TMonitorVO monitor = monitorMap.get(result.getNo());
|
if (monitor == null) continue;
|
String deptId = monitor.getDeptId().toString();
|
updateAreaStats(areaStatsMap, deptId, result);
|
|
// 处理省厅数据
|
if (result.getProvinceTag()!=null && result.getProvinceTag()) {
|
String provinceKey = ApiConstants.Province + deptId;
|
updateAreaStats(areaStatsMap, provinceKey, result);
|
}
|
}
|
return areaStatsMap;
|
}
|
}
|
