From 133982999dc61cd4e8acbf8a8d72aeb0251db47f Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Tue, 22 Jun 2021 14:07:02 +0800
Subject: [PATCH] screen-manage                增加报警以及因子显示单位缓存

---
 screen-manage/src/main/java/com/moral/api/util/CacheUtils.java |   83 +++++++++++++++++++++++++++++++++++++++--
 1 files changed, 78 insertions(+), 5 deletions(-)

diff --git a/screen-manage/src/main/java/com/moral/api/util/CacheUtils.java b/screen-manage/src/main/java/com/moral/api/util/CacheUtils.java
index 958845a..8fa6e54 100644
--- a/screen-manage/src/main/java/com/moral/api/util/CacheUtils.java
+++ b/screen-manage/src/main/java/com/moral/api/util/CacheUtils.java
@@ -1,15 +1,24 @@
 package com.moral.api.util;
 
-import com.moral.api.entity.Device;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.moral.api.entity.*;
 import com.moral.api.mapper.DeviceMapper;
+import com.moral.api.mapper.SensorMapper;
+import com.moral.api.mapper.UnitConversionMapper;
+import com.moral.api.service.SensorService;
+import com.moral.api.service.SysDictDataService;
+import com.moral.constant.Constants;
 import com.moral.constant.RedisConstants;
 import com.moral.util.TokenUtils;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Component;
+import org.springframework.util.ObjectUtils;
 
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 /**
  * @ClassName CacheUtils
@@ -26,21 +35,85 @@
 
     private static DeviceMapper deviceMapper;
 
+    private static SysDictDataService sysDictDataService;
+
+    private static UnitConversionMapper unitConversionMapper;
+
+    private static SensorService sensorService;
+
+    @Autowired
+    public void setSensorService(SensorService sensorService) {
+        CacheUtils.sensorService = sensorService;
+    }
+
     @Autowired
     public void setRedisTemplate(RedisTemplate redisTemplate) {
         CacheUtils.redisTemplate = redisTemplate;
     }
 
     @Autowired
-    public void setDeviceMapper(DeviceMapper deviceMapper) { CacheUtils.deviceMapper = deviceMapper; }
+    public void setDeviceMapper(DeviceMapper deviceMapper) {
+        CacheUtils.deviceMapper = deviceMapper;
+    }
 
-    public static void clearDictionariesCache(){
+    @Autowired
+    public void setSysDictDataService(SysDictDataService sysDictDataService) {
+        CacheUtils.sysDictDataService = sysDictDataService;
+    }
+
+    @Autowired
+    public void setUnitConversionMapper(UnitConversionMapper unitConversionMapper) {
+        CacheUtils.unitConversionMapper = unitConversionMapper;
+    }
+
+    public static void clearDictionariesCache() {
         redisTemplate.delete(RedisConstants.DICT_DATA_KEY);
         redisTemplate.delete(RedisConstants.DICT_TYPE_KEY);
     }
 
-    public static void flushDeviceAlarmUnit(){
+    public static void flushDeviceAlarmUnit() {
         List<Device> devices = deviceMapper.queryDeviceUnitAlarmInfo();
-        System.out.println(devices);
+        Map<String, SysDictData> unitMap = sysDictDataService.getDictDatasByType("unit");
+        //���������������������������������������������
+        Map<String,Device> result = new HashMap<>();
+        for (Device device : devices) {
+            Version version = device.getVersion();
+            List<Sensor> sensors = version.getSensors();
+            for (Sensor sensor : sensors) {
+                //������������������
+                String unitKey = sensor.getUnitKey();
+                String showUnitKey = sensor.getShowUnitKey();
+                SysDictData unitData = unitMap.get(unitKey);
+                SysDictData showUnitData = unitMap.get(showUnitKey);
+                sensor.setUnit(unitData.getDataValue());
+                sensor.setShowUnit(showUnitData.getDataValue());
+                //������������������
+                Map<String, Sensor> allSensors = sensorService.getAllSensorFromCache();
+                String sensorName = allSensors.get(sensor.getCode()).getName();
+                sensor.setName(sensorName);
+                //������������������
+                if (!unitKey.equals(showUnitKey)) {
+                    QueryWrapper<UnitConversion> queryWrapper = new QueryWrapper<>();
+                    queryWrapper.eq("is_delete", Constants.NOT_DELETE);
+                    queryWrapper.eq("original_unit_key", unitKey);
+                    queryWrapper.eq("target_unit_key", showUnitKey);
+                    List<UnitConversion> unitConversions = unitConversionMapper.selectList(queryWrapper);
+                    if (unitConversions.size() == 1) {
+                        UnitConversion unitConversion = unitConversions.get(0);
+                        if (ObjectUtils.isEmpty(unitConversion.getSensorCode()))
+                            sensor.setFormula(unitConversion.getFormula());
+
+                    } else {
+                        for (UnitConversion unitConversion : unitConversions) {
+                            if (sensor.getCode().equals(unitConversion.getSensorCode()))
+                                sensor.setFormula(unitConversion.getFormula());
+                        }
+                    }
+                }
+            }
+            result.put(device.getMac(),device);
+        }
+        //������redis
+        redisTemplate.opsForHash().putAll(RedisConstants.DEVICE_INFO,result);
     }
 }

--
Gitblit v1.8.0