From 93fe0b66e85e95e3a7e77a28504d6d289b6906b9 Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Fri, 08 Oct 2021 13:11:22 +0800
Subject: [PATCH] screen-manage screen-api      更改拦截器逻辑,记录token无效的请求IP以及URL。      manage记录到数据库日志中      api记录在日志文件中

---
 screen-manage/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java |  217 +++++++++++++++++++++++++++++++++++------------------
 1 files changed, 143 insertions(+), 74 deletions(-)

diff --git a/screen-manage/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java b/screen-manage/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java
index 4af7aef..973e952 100644
--- a/screen-manage/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java
+++ b/screen-manage/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java
@@ -12,6 +12,8 @@
 import com.moral.api.service.DeviceService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 
+import com.moral.api.service.OrganizationService;
+import com.moral.api.service.SysDictDataService;
 import com.moral.api.util.CacheUtils;
 import com.moral.api.util.AdjustDataUtils;
 import com.moral.api.util.LogUtils;
@@ -34,7 +36,6 @@
 import java.lang.reflect.Field;
 import java.lang.reflect.Method;
 import java.util.ArrayList;
-import java.util.HashMap;
 import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
@@ -74,7 +75,8 @@
     private SysDictDataMapper sysDictDataMapper;
 
     @Autowired
-    private LogUtils logUtils;
+    private SysDictDataService sysDictDataService;
+
 
     @Autowired
     private RedisTemplate redisTemplate;
@@ -86,44 +88,33 @@
     private VersionSensorUnitMapper versionSensorUnitMapper;
 
     @Autowired
+    private SpecialDeviceMapper specialDeviceMapper;
+
+    @Autowired
     private AdjustDataUtils adjustDataUtils;
+
+    @Autowired
+    private OrganizationService organizationService;
 
     /*
      * ���redis������������������
      * */
     private Map<String, Object> getDeviceInfoFromRedis(String mac) {
-        return (Map<String, Object>) redisTemplate.opsForValue().get(getDeviceKey(mac));
+        return (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.DEVICE, mac);
     }
 
     /*
      * ������������������redis
      */
     private void setDeviceInfoToRedis(String mac, Map<String, Object> deviceInfo) {
-        redisTemplate.opsForValue().set(getDeviceKey(mac), deviceInfo);
+        redisTemplate.opsForHash().put(RedisConstants.DEVICE, mac, deviceInfo);
     }
 
     /*
      * ���redis������������������
      */
     private void delDeviceInfoFromRedis(String mac) {
-        redisTemplate.delete(getDeviceKey(mac));
-    }
-
-    /*
-     * ���������������������redis������key
-     */
-    private String getDeviceKey(String mac) {
-        return keysConnect(RedisConstants.DEVICE, mac);
-    }
-
-    //redis key������
-    private String keysConnect(String... keys) {
-        StringBuilder key = new StringBuilder(keys[0]);
-        for (int i = 1; i < keys.length; i++) {
-            key.append("_");
-            key.append(keys[i]);
-        }
-        return key.toString().toLowerCase();
+        redisTemplate.opsForHash().delete(RedisConstants.DEVICE, mac);
     }
 
     @Override
@@ -134,7 +125,7 @@
         deviceMapper.insert(device);
         Map<String, Object> deviceInfo = selectDeviceInfoById(device.getId());
         //���������������������������
-        insertOrganizationUnitAlarm(orgId,device.getDeviceVersionId());
+        insertOrganizationUnitAlarm(orgId, device.getDeviceVersionId());
         //������������������������redis
         String mac = device.getMac();
         //���redis���������������������
@@ -142,12 +133,12 @@
         //������������������redis
         setDeviceInfoToRedis(mac, deviceInfo);
         //������deviceInfo������
-        CacheUtils.flushDeviceAlarmInfo();
+        CacheUtils.refreshDeviceAlarmInfo();
         //������������������
         HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();
         StringBuilder content = new StringBuilder();
-        content.append("������������������").append(device.getName()).append("���").append("mac���").append(mac);
-        logUtils.saveOperationForManage(request, content.toString(), Constants.INSERT_OPERATE_TYPE);
+        content.append("������������������").append(device.getName()).append(":").append("mac���").append(mac);
+        LogUtils.saveOperationForManage(request, content.toString(), Constants.INSERT_OPERATE_TYPE);
     }
 
     @Override
@@ -163,14 +154,14 @@
         //���������������������������
         Integer versionId = device.getDeviceVersionId();
         Integer orgId = device.getOrganizationId();
-        deleteOrganizationUnitAlarm(orgId,versionId);
+        deleteOrganizationUnitAlarm(orgId, versionId);
         //������deviceInfo������
-        CacheUtils.flushDeviceAlarmInfo();
+        CacheUtils.refreshDeviceAlarmInfo();
         //������������������
         HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();
         StringBuilder content = new StringBuilder();
         content.append("������������������").append(device.getName()).append(";").append("mac���").append(mac);
-        logUtils.saveOperationForManage(request, content.toString(), Constants.DELETE_OPERATE_TYPE);
+        LogUtils.saveOperationForManage(request, content.toString(), Constants.DELETE_OPERATE_TYPE);
     }
 
     @Override
@@ -179,7 +170,7 @@
         Integer deviceId = device.getId();
         Device oldDevice = deviceMapper.selectById(deviceId);
         //���������������������������������������������������������������������������������id������������
-        if(!ObjectUtils.isEmpty(device.getMonitorPointId())){
+        if (!ObjectUtils.isEmpty(device.getMonitorPointId())) {
             MonitorPoint monitorPoint = monitorPointMapper.selectById(device.getMonitorPointId());
             device.setOrganizationId(monitorPoint.getOrganizationId());
         }
@@ -187,25 +178,25 @@
         Device updateDevice = deviceMapper.selectById(deviceId);
         String mac = updateDevice.getMac();
         //���������������������������
-       Integer oldOrgId = oldDevice.getOrganizationId();
-       Integer newOrgId = updateDevice.getOrganizationId();
-       Integer oldVersionId = oldDevice.getDeviceVersionId();
-       Integer newVersionId = updateDevice.getDeviceVersionId();
-       if(!oldOrgId.equals(newOrgId)||!oldVersionId.equals(newVersionId)){
-           deleteOrganizationUnitAlarm(oldOrgId,oldVersionId);
-           insertOrganizationUnitAlarm(newOrgId,newVersionId);
-       }
+        Integer oldOrgId = oldDevice.getOrganizationId();
+        Integer newOrgId = updateDevice.getOrganizationId();
+        Integer oldVersionId = oldDevice.getDeviceVersionId();
+        Integer newVersionId = updateDevice.getDeviceVersionId();
+        if (!oldOrgId.equals(newOrgId) || !oldVersionId.equals(newVersionId)) {
+            deleteOrganizationUnitAlarm(oldOrgId, oldVersionId);
+            insertOrganizationUnitAlarm(newOrgId, newVersionId);
+        }
         //���redis���������������������
-        delDeviceInfoFromRedis(mac);
+        delDeviceInfoFromRedis(oldDevice.getMac());
         Map<String, Object> deviceInfo = selectDeviceInfoById(deviceId);
         //������������������redis
         setDeviceInfoToRedis(mac, deviceInfo);
         //������deviceInfo������
-        CacheUtils.flushDeviceAlarmInfo();
+        CacheUtils.refreshDeviceAlarmInfo();
         //������������������
         HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();
         StringBuilder content = new StringBuilder();
-        content.append("������������������").append(mac).append("���");
+        content.append("������������������" + oldDevice.getMac()).append("==>").append(mac).append("���");
         Field[] fields = Device.class.getDeclaredFields();
         for (Field field : fields) {
             if (field.getName().equals("id")) {
@@ -222,7 +213,7 @@
                 Object o1 = method.invoke(oldDevice);
                 Object o2 = method.invoke(device);
                 if (o2 != null) {
-                    content.append(fieldName).append("���").append(o1).append("-->").append(o2).append("���");
+                    content.append(fieldName).append("���").append(o1).append("-->").append(o2).append(":");
                 }
 
             } catch (Exception e) {
@@ -230,7 +221,7 @@
             }
 
         }
-        logUtils.saveOperationForManage(request, content.toString(), Constants.UPDATE_OPERATE_TYPE);
+        LogUtils.saveOperationForManage(request, content.toString(), Constants.UPDATE_OPERATE_TYPE);
     }
 
     @Override
@@ -254,8 +245,14 @@
 
     @Override
     public List<Map<String, Object>> selectMonitorsByOrgId(Integer orgId) {
+        //���������������������
+        List<Organization> organizations = organizationService.getAllChildrenOrganization(orgId);
+        List<Integer> orgIds = organizations.stream().map(Organization::getId).collect(Collectors.toList());
+        orgIds.add(orgId);
         QueryWrapper<MonitorPoint> queryWrapper = new QueryWrapper<>();
-        queryWrapper.select("id", "name").eq("organization_id", orgId).eq("is_delete", Constants.NOT_DELETE);
+        queryWrapper.select("id", "name")
+                .eq("is_delete", Constants.NOT_DELETE)
+                .in("organization_id", orgIds);
         return monitorPointMapper.selectMaps(queryWrapper);
     }
 
@@ -281,9 +278,13 @@
         Object orgId = parameters.get("organizationId");
         Object mpId = parameters.get("monitorPointId");
 
-        //������������������������������
+        //������������������������������,���������������
         if (orgId != null) {
-            queryWrapper.eq("organization_id", orgId);
+            //���������������������
+            List<Organization> organizations = organizationService.getAllChildrenOrganization(Integer.parseInt(orgId.toString()));
+            List<Integer> orgIds = organizations.stream().map(Organization::getId).collect(Collectors.toList());
+            orgIds.add(Integer.parseInt(orgId.toString()));
+            queryWrapper.in("organization_id", orgIds);
         }
 
         //������������������������������
@@ -403,6 +404,16 @@
         mpInfo.put("provinceCode", monitorPoint.getProvinceCode());
         deviceInfo.put("monitorPoint", mpInfo);
 
+        //������������
+        Map<String, Object> govMpInfo = new LinkedHashMap<>();
+
+        govMpInfo.put("guid", null);
+        govMpInfo.put("name", null);
+        if (device.getGovMonitorPoint() != null) {
+            govMpInfo.put("guid", device.getGovMonitorPoint().getGuid());
+            govMpInfo.put("name", device.getGovMonitorPoint().getName());
+        }
+        deviceInfo.put("govMonitorPoint", govMpInfo);
         setDeviceInfoToRedis(mac, deviceInfo);
         return deviceInfo;
     }
@@ -424,7 +435,7 @@
     @Override
     public Map<String, Object> getDeviceByMac(String mac) {
         Map<String, Object> deviceInfo = getDeviceInfoFromRedis(mac);
-        if (deviceInfo == null) {
+        if (ObjectUtils.isEmpty(deviceInfo)) {
             QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
             queryWrapper.eq("mac", mac).eq("is_delete", Constants.NOT_DELETE);
             Device device = deviceMapper.selectOne(queryWrapper);
@@ -438,10 +449,44 @@
 
     @Override
     public Map<String, Object> adjustDeviceData(Map<String, Object> deviceData) {
-        return adjustDataUtils.adjust(deviceData);
+        String mac = deviceData.remove("mac").toString();
+        //���redis������������������
+        Map<String, Object> adjustFormula = redisTemplate.opsForHash().entries(RedisConstants.ADJUST + "_" + mac);
+        if (!ObjectUtils.isEmpty(adjustFormula)) {
+            Map<String, Object> deviceInfo = getDeviceByMac(mac);
+            //������������������������������������
+            Map<String, Object> govMpInfo = (Map<String, Object>) deviceInfo.get("govMonitorPoint");
+            Map<String, Object> aqiMap = null;
+            if (govMpInfo.get("guid") != null) {
+                aqiMap = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.AQI_DATA, govMpInfo.get("guid").toString());
+            }
+            return adjustDataUtils.adjust(deviceData, adjustFormula, ObjectUtils.isEmpty(aqiMap) ? null : aqiMap);
+        }
+        deviceData.remove("DataTime");
+        return deviceData;
     }
 
     @Override
+    public Map<String, Object> adjustSpecialDeviceData(Map<String, Object> deviceData) {
+        String mac = deviceData.remove("mac").toString();
+        //���redis������������������
+        Map<String, Object> adjustFormula = redisTemplate.opsForHash().entries(RedisConstants.ADJUST + "_" + mac);
+        if (!ObjectUtils.isEmpty(adjustFormula)) {
+            Organization organization = specialDeviceMapper.selectOrgByMac(mac);
+            String areaCode = organization.getAreaCode().toString();
+            String cityCode = organization.getCityCode().toString();
+            Map<String, Object> aqiMap = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.AQI_DATA, areaCode);
+            if (ObjectUtils.isEmpty(aqiMap)) {
+                aqiMap = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.AQI_DATA, cityCode);
+            }
+            return adjustDataUtils.adjust(deviceData, adjustFormula, ObjectUtils.isEmpty(aqiMap) ? null : aqiMap);
+        }
+        deviceData.remove("DataTime");
+        return deviceData;
+    }
+
+    @Override
+    @Transactional
     public void judgeDeviceState(Map<String, Object> deviceData) {
         String mac = deviceData.remove("mac").toString();
         Device device = (Device) redisTemplate.opsForHash().get(RedisConstants.DEVICE_INFO, mac);
@@ -455,19 +500,19 @@
             if (StringUtils.isEmpty(alarmLevel)) {
                 continue;
             }
-            List<Double> list = JSONObject.parseObject(alarmLevel, List.class);
+            List<Object> list = JSONObject.parseObject(alarmLevel, List.class);
             String sensorCode = sensor.getCode();
             //������������
             String formula = sensor.getFormula();
             //������������������������
             String sensorValue = (String) deviceData.get(sensorCode);
-            double value = Double.parseDouble(String.format("%.3f", sensorValue));
+            double value = Double.parseDouble(sensorValue);
             //������������
             if (formula != null) {
                 //������������������
                 sensorValue = formula.replace("{0}", sensorValue);
                 expression = AviatorEvaluator.compile(sensorValue);
-                value = Double.parseDouble(String.format("%.3f", expression.execute()));
+                value = Double.parseDouble(expression.execute().toString());
             }
             int sensorState = judgeState(list, value);
             if (sensorState > state) {
@@ -476,37 +521,54 @@
         }
         //������������������
         UpdateWrapper<Device> updateWrapper = new UpdateWrapper<>();
-        updateWrapper.eq("id", device.getId()).set("state", state);
+        updateWrapper.eq("id", device.getId()).set("state", String.valueOf(state));
         deviceMapper.update(null, updateWrapper);
     }
 
     //���������������������������
-    private int judgeState(List<Double> levels, Double data) {
+    private int judgeState(List<Object> levels, Double data) {
         int state = 1;
         for (int i = levels.size() - 1; i >= 0; i--) {
-            Double level = levels.get(i);
+            Double level = Double.parseDouble(levels.get(i).toString());
             if (data >= level) {
-                state = i + 2;
+                state = i + 1;
                 break;
             }
         }
         return state;
     }
 
-    private void insertOrganizationUnitAlarm(Integer orgId,Integer versionId){
+    /**
+     * @Description: ���������������������������������������������������������������������������������������������
+     * @Param: [orgId, versionId]
+     * @return: void
+     * @Author: ���������
+     * @Date: 2021/9/6
+     */
+    private void insertOrganizationUnitAlarm(Integer orgId, Integer versionId) {
+        //���������������������������
         QueryWrapper<OrganizationUnitAlarm> queryOrganizationVersionWrapper = new QueryWrapper<>();
-        queryOrganizationVersionWrapper.eq("organization_id",orgId);
-        queryOrganizationVersionWrapper.eq("version_id",versionId);
-        queryOrganizationVersionWrapper.eq("is_delete",Constants.NOT_DELETE);
+        queryOrganizationVersionWrapper.eq("organization_id", orgId);
+        queryOrganizationVersionWrapper.eq("version_id", versionId);
+        queryOrganizationVersionWrapper.eq("is_delete", Constants.NOT_DELETE);
         List<OrganizationUnitAlarm> organizationUnitAlarms = organizationUnitAlarmMapper.selectList(queryOrganizationVersionWrapper);
-        if(ObjectUtils.isEmpty(organizationUnitAlarms)){
-            QueryWrapper<VersionSensorUnit> queryVersionSensorUnitWrapper =new QueryWrapper<>();
-            queryVersionSensorUnitWrapper.eq("version_id",versionId);
-            queryVersionSensorUnitWrapper.eq("is_delete",Constants.NOT_DELETE);
+        //������������������������������������
+        if (ObjectUtils.isEmpty(organizationUnitAlarms)) {
+            //������������������������������
+            QueryWrapper<VersionSensorUnit> queryVersionSensorUnitWrapper = new QueryWrapper<>();
+            queryVersionSensorUnitWrapper.eq("version_id", versionId);
+            queryVersionSensorUnitWrapper.eq("is_delete", Constants.NOT_DELETE);
             List<VersionSensorUnit> versionSensorUnits = versionSensorUnitMapper.selectList(queryVersionSensorUnitWrapper);
-            if(!ObjectUtils.isEmpty(versionSensorUnits)){
+            //������������������������������������
+            Map<String, SysDictData> sensorAlarms = sysDictDataService.getDictDatasByType("defaultAlarm");
+            if (!ObjectUtils.isEmpty(versionSensorUnits)) {
                 for (VersionSensorUnit versionSensorUnit : versionSensorUnits) {
                     OrganizationUnitAlarm organizationUnitAlarm = new OrganizationUnitAlarm();
+                    String sensorCode = versionSensorUnit.getSensorCode();
+                    //���������������������
+                    SysDictData alarm = sensorAlarms.get(sensorCode);
+                    if (alarm != null)
+                        organizationUnitAlarm.setAlarmLevel(alarm.getDataValue());
                     organizationUnitAlarm.setOrganizationId(orgId);
                     organizationUnitAlarm.setVersionId(versionId);
                     organizationUnitAlarm.setSensorCode(versionSensorUnit.getSensorCode());
@@ -518,19 +580,26 @@
         }
     }
 
-    private void deleteOrganizationUnitAlarm(Integer orgId,Integer versionId){
+    /**
+     * @Description: ������������������������������������������������������������������������������������������������������������������������������������
+     * @Param: [orgId, versionId]
+     * @return: void
+     * @Author: ���������
+     * @Date: 2021/9/6
+     */
+    private void deleteOrganizationUnitAlarm(Integer orgId, Integer versionId) {
         QueryWrapper<Device> queryOrganizationVersionWrapper = new QueryWrapper<>();
-        queryOrganizationVersionWrapper.eq("organization_id",orgId);
-        queryOrganizationVersionWrapper.eq("device_version_id",versionId);
-        queryOrganizationVersionWrapper.eq("is_delete",Constants.NOT_DELETE);
+        queryOrganizationVersionWrapper.eq("organization_id", orgId);
+        queryOrganizationVersionWrapper.eq("device_version_id", versionId);
+        queryOrganizationVersionWrapper.eq("is_delete", Constants.NOT_DELETE);
         List<Device> devices = deviceMapper.selectList(queryOrganizationVersionWrapper);
-        if(ObjectUtils.isEmpty(devices)){//������������������������������������������������������
+        if (ObjectUtils.isEmpty(devices)) {//������������������������������������������������������
             UpdateWrapper deleteWrapper = new UpdateWrapper();
-            deleteWrapper.eq("organization_id",orgId);
-            deleteWrapper.eq("version_id",versionId);
-            deleteWrapper.eq("is_delete",Constants.NOT_DELETE);
-            deleteWrapper.set("is_delete",Constants.DELETE);
-            organizationUnitAlarmMapper.update(null,deleteWrapper);
+            deleteWrapper.eq("organization_id", orgId);
+            deleteWrapper.eq("version_id", versionId);
+            deleteWrapper.eq("is_delete", Constants.NOT_DELETE);
+            deleteWrapper.set("is_delete", Constants.DELETE);
+            organizationUnitAlarmMapper.update(null, deleteWrapper);
         }
     }
 }

--
Gitblit v1.8.0