From 24d45b49ad478538b2c0ba4eda1f2725ac3ca049 Mon Sep 17 00:00:00 2001 From: fuliqi <fuliqi@qq.com> Date: 星期一, 12 八月 2024 17:06:12 +0800 Subject: [PATCH] 数据中心增加查询总条数 --- ycl-server/src/main/java/com/ycl/task/MonitorTask.java | 89 +++++++++++++++++++++++++++++++++++--------- 1 files changed, 71 insertions(+), 18 deletions(-) diff --git a/ycl-server/src/main/java/com/ycl/task/MonitorTask.java b/ycl-server/src/main/java/com/ycl/task/MonitorTask.java index ae6b11e..017bcbc 100644 --- a/ycl-server/src/main/java/com/ycl/task/MonitorTask.java +++ b/ycl-server/src/main/java/com/ycl/task/MonitorTask.java @@ -3,14 +3,19 @@ import com.alibaba.druid.support.json.JSONUtils; import com.alibaba.fastjson2.JSONArray; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.ycl.platform.domain.entity.TMonitor; +import com.ycl.platform.domain.entity.YwPoint; import com.ycl.platform.domain.result.UY.OneMachineFileResult; import com.ycl.platform.domain.vo.TMonitorVO; import com.ycl.platform.service.ITMonitorService; +import com.ycl.platform.service.YwPointService; import com.ycl.system.entity.SysDictData; import com.ycl.system.service.ISysDictDataService; import com.ycl.utils.DateUtils; import constant.RedisConstant; +import enumeration.general.AreaDeptEnum; +import enumeration.general.PointStatus; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.data.mongodb.core.MongoTemplate; @@ -37,40 +42,88 @@ @Autowired private ISysDictDataService dictDataService; @Autowired + private YwPointService ywPointService; + @Autowired private RedisTemplate redisTemplate; + //鍚屾mongodb涓�鏈轰竴妗e埌鏁版嵁搴� public void synchronize() { Query query = new Query(Criteria.where("mongoCreateTime").gte(DateUtils.getDayStart(new Date())).lt(DateUtils.getDayEnd(new Date()))); List<OneMachineFileResult> oneMachineFileResults = mongoTemplate.find(query, OneMachineFileResult.class); - List<String> serialNumberInBase = monitorService.selectTMonitorList(null).stream().map(TMonitorVO::getSerialNumber).collect(Collectors.toList()); - //鍑嗗鎻掑叆鏁版嵁搴撶殑鏁版嵁 + //鏁版嵁搴搈onitor琛ㄦ暟鎹� + Map<String, TMonitorVO> monitorVOMap = monitorService.selectTMonitorList(null).stream().collect(Collectors.toMap(TMonitorVO::getSerialNumber, Function.identity())); + //鍑嗗鎻掑叆璁惧琛ㄧ殑鏁版嵁 List<TMonitor> monitorList = new ArrayList<>(); + //鍑嗗鎻掑叆鐐逛綅琛ㄧ殑鏁版嵁 + List<YwPoint> ywPointList = new ArrayList<>(); //鏂扮殑鏁版嵁锛屽師鏁版嵁搴撲腑涓嶅瓨鍦ㄧ殑鏁版嵁 Set<TMonitor> newMonitorList = new HashSet<>(); - for (OneMachineFileResult result : oneMachineFileResults) { - TMonitor monitor = setMonitor(result); - monitorList.add(monitor); - //姣斿绛涢�夊嚭鏂扮殑鏁版嵁 - if(!CollectionUtils.isEmpty(serialNumberInBase) && !serialNumberInBase.contains(result.getSBBM())){ - newMonitorList.add(monitor); - } - } - //鏂扮殑鏁版嵁鏀惧叆Redis涓瓑寰呰�冩牳鎸囨爣浠诲姟浣跨敤 - redisTemplate.opsForValue().set(RedisConstant.New_Monitor_Set, JSONArray.toJSONString(newMonitorList)); - //TODO:瑙f瀽鍖哄煙 - - //TODO:瑙f瀽閲嶇偣鐐逛綅 - //閲嶇偣鐐逛綅闆嗗悎瀛楀吀 + //鐐逛綅鏁版嵁 + Map<String, YwPoint> pointMap = ywPointService.list(new QueryWrapper<YwPoint>()).stream().collect(Collectors.toMap(YwPoint::getSerialNumber, Function.identity())); + //閲嶇偣鐐逛綅闆嗗悎瀛楀吀(瑙f瀽SXJCJQY瀛楁) SysDictData sysDictData = new SysDictData(); sysDictData.setDictType("platform_important_site"); List<SysDictData> DictDataList = dictDataService.selectDictDataList(sysDictData); List<String> importantSite = DictDataList.stream().map(SysDictData::getDictValue).collect(Collectors.toList()); - + //TODO:鍏ㄥ勾鐣欏瓨杩樻槸鍗曟鐣欏瓨 + for (OneMachineFileResult result : oneMachineFileResults) { + TMonitor monitor = getMonitor(result,monitorVOMap); + YwPoint point = getPoint(result, pointMap, importantSite); + monitorList.add(monitor); + ywPointList.add(point); + //姣斿绛涢�夊嚭鏂扮殑鏁版嵁 + if (!CollectionUtils.isEmpty(monitorVOMap) && !monitorVOMap.containsKey(result.getSBBM())) { + newMonitorList.add(monitor); + } + } + //鎻掑叆鏁版嵁搴� + if (!CollectionUtils.isEmpty(monitorList)) { + monitorService.saveOrUpdateBatch(monitorList); + } + if(!CollectionUtils.isEmpty(ywPointList)){ + ywPointService.saveOrUpdateBatch(ywPointList); + } + //鏂扮殑鏁版嵁鏀惧叆Redis涓瓑寰呰�冩牳鎸囨爣浠诲姟浣跨敤 + redisTemplate.opsForValue().set(RedisConstant.New_Monitor_Set, JSONArray.toJSONString(newMonitorList)); } - private TMonitor setMonitor(OneMachineFileResult result) { + private YwPoint getPoint(OneMachineFileResult result, Map<String, YwPoint> pointMap, List<String> importantSite) { + YwPoint ywPoint = new YwPoint(); + if (pointMap.containsKey(result.getSBBM())) { + ywPoint = pointMap.get(result.getSBBM()); + }else { + ywPoint.setPointName(result.getSBMC()); + ywPoint.setStatus(PointStatus.WAIT.getDesc()); + ywPoint.setSerialNumber(result.getSBBM()); + ywPoint.setImportantTag(Boolean.FALSE); + ywPoint.setProvinceTag(Boolean.FALSE); + ywPoint.setCreateTime(new Date()); + ywPoint.setUpdateTime(new Date()); + } + //姣斿鏄惁鏄噸鐐圭偣浣� + if (importantSite.contains(result.getSBBM())) { + ywPoint.setImportantTag(Boolean.TRUE); + } + //瑙f瀽deptId + //鍖哄煙琛屾斂缂栫爜 + String areaCode = result.getSBBM().substring(0, 6); + AreaDeptEnum areaDeptEnum = AreaDeptEnum.fromCode(areaCode); + //濡傛灉瑙f瀽涓嶅嚭鍖哄煙deptId涓�-1 + Integer deptId = -1; + if(areaDeptEnum!=null){ + deptId = areaDeptEnum.getDeptId(); + } + ywPoint.setDeptId(Long.valueOf(deptId + "")); + + return ywPoint; + } + + private TMonitor getMonitor(OneMachineFileResult result, Map<String, TMonitorVO> monitorVOMap) { TMonitor monitor = new TMonitor(); + if (monitorVOMap.containsKey(result.getSBBM())) { + monitor.setId(monitorVOMap.get(result.getSBBM()).getId()); + } monitor.setSerialNumber(result.getSBBM()); monitor.setName(result.getSBMC()); monitor.setSiteType(Long.valueOf(result.getJKDWLX())); -- Gitblit v1.8.0