From ec95e52c6f84c0a6cbdcbaecc2465b00ede6696d Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Thu, 24 Jun 2021 14:03:59 +0800
Subject: [PATCH] 数据校准工具类

---
 screen-manage/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java |  277 ++++++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 243 insertions(+), 34 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 c5573f4..fcf29d3 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
@@ -1,34 +1,32 @@
 package com.moral.api.service.impl;
 
+import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
-import com.moral.api.entity.Device;
-import com.moral.api.entity.ManageAccount;
-import com.moral.api.entity.MonitorPoint;
-import com.moral.api.entity.Organization;
-import com.moral.api.entity.SysDictData;
-import com.moral.api.entity.SysDictType;
-import com.moral.api.entity.Version;
-import com.moral.api.mapper.DeviceMapper;
-import com.moral.api.mapper.ManageAccountMapper;
-import com.moral.api.mapper.MonitorPointMapper;
-import com.moral.api.mapper.OrganizationMapper;
-import com.moral.api.mapper.SysDictDataMapper;
-import com.moral.api.mapper.SysDictTypeMapper;
+import com.googlecode.aviator.AviatorEvaluator;
+import com.googlecode.aviator.Expression;
+import com.moral.api.entity.*;
+import com.moral.api.mapper.*;
 import com.moral.api.pojo.vo.device.DeviceVO;
 import com.moral.api.service.DeviceService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 
+import com.moral.api.util.CacheUtils;
+import com.moral.api.util.AdjustDataUtils;
 import com.moral.api.util.LogUtils;
 import com.moral.constant.Constants;
-import com.moral.redis.RedisUtil;
+import com.moral.constant.RedisConstants;
 import com.moral.util.ConvertUtils;
 import com.moral.util.DateUtils;
 
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
+import org.springframework.util.ObjectUtils;
+import org.springframework.util.StringUtils;
 import org.springframework.web.context.request.RequestContextHolder;
 import org.springframework.web.context.request.ServletRequestAttributes;
 
@@ -52,6 +50,7 @@
  * @since 2021-05-11
  */
 @Service
+@Slf4j
 public class DeviceServiceImpl extends ServiceImpl<DeviceMapper, Device> implements DeviceService {
 
     @Autowired
@@ -75,6 +74,39 @@
     @Autowired
     private LogUtils logUtils;
 
+    @Autowired
+    private RedisTemplate redisTemplate;
+
+    @Autowired
+    private OrganizationUnitAlarmMapper organizationUnitAlarmMapper;
+
+    @Autowired
+    private VersionSensorUnitMapper versionSensorUnitMapper;
+
+    @Autowired
+    private AdjustDataUtils adjustDataUtils;
+
+    /*
+     * ���redis������������������
+     * */
+    private Map<String, Object> getDeviceInfoFromRedis(String mac) {
+        return (Map<String, Object>) redisTemplate.opsForValue().get(RedisConstants.DEVICE + mac);
+    }
+
+    /*
+     * ������������������redis
+     */
+    private void setDeviceInfoToRedis(String mac, Map<String, Object> deviceInfo) {
+        redisTemplate.opsForValue().set(RedisConstants.DEVICE + mac, deviceInfo);
+    }
+
+    /*
+     * ���redis������������������
+     */
+    private void delDeviceInfoFromRedis(String mac) {
+        redisTemplate.delete(RedisConstants.DEVICE + mac);
+    }
+
     @Override
     @Transactional
     public void insert(Device device) {
@@ -82,25 +114,39 @@
         device.setOrganizationId(orgId);
         deviceMapper.insert(device);
         Map<String, Object> deviceInfo = selectDeviceInfoById(device.getId());
+        //���������������������������
+        insertOrganizationUnitAlarm(orgId, device.getDeviceVersionId());
         //������������������������redis
-        RedisUtil.set("device_" + device.getMac(), deviceInfo);
+        String mac = device.getMac();
+        //���redis���������������������
+        delDeviceInfoFromRedis(mac);
+        //������������������redis
+        setDeviceInfoToRedis(mac, deviceInfo);
+        //������deviceInfo������
+        CacheUtils.flushDeviceAlarmInfo();
         //������������������
         HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();
         StringBuilder content = new StringBuilder();
-        content.append("������������������").append(device.getName()).append("���").append("mac���").append(device.getMac());
+        content.append("������������������").append(device.getName()).append("���").append("mac���").append(mac);
         logUtils.saveOperationForManage(request, content.toString(), Constants.INSERT_OPERATE_TYPE);
     }
 
     @Override
     @Transactional
     public void delete(Integer deviceId) {
+        Device device = deviceMapper.selectById(deviceId);
         UpdateWrapper<Device> updateWrapper = new UpdateWrapper<>();
         updateWrapper.eq("id", deviceId).set("is_delete", Constants.DELETE);
         deviceMapper.update(null, updateWrapper);
-        Device device = deviceMapper.selectById(deviceId);
         String mac = device.getMac();
-        //������redis
-        RedisUtil.del("device_" + mac);
+        //���redis���������������������
+        delDeviceInfoFromRedis(mac);
+        //���������������������������
+        Integer versionId = device.getDeviceVersionId();
+        Integer orgId = device.getOrganizationId();
+        deleteOrganizationUnitAlarm(orgId, versionId);
+        //������deviceInfo������
+        CacheUtils.flushDeviceAlarmInfo();
         //������������������
         HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();
         StringBuilder content = new StringBuilder();
@@ -113,12 +159,30 @@
     public void update(Device device) {
         Integer deviceId = device.getId();
         Device oldDevice = deviceMapper.selectById(deviceId);
+        //���������������������������������������������������������������������������������id������������
+        if (!ObjectUtils.isEmpty(device.getMonitorPointId())) {
+            MonitorPoint monitorPoint = monitorPointMapper.selectById(device.getMonitorPointId());
+            device.setOrganizationId(monitorPoint.getOrganizationId());
+        }
         deviceMapper.updateById(device);
-        String mac = deviceMapper.selectById(deviceId).getMac();
-        //������redis
-        RedisUtil.del("device_" + mac);
+        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);
+        }
+        //���redis���������������������
+        delDeviceInfoFromRedis(mac);
         Map<String, Object> deviceInfo = selectDeviceInfoById(deviceId);
-        RedisUtil.set("device_" + mac, deviceInfo);
+        //������������������redis
+        setDeviceInfoToRedis(mac, deviceInfo);
+        //������deviceInfo������
+        CacheUtils.flushDeviceAlarmInfo();
         //������������������
         HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();
         StringBuilder content = new StringBuilder();
@@ -177,17 +241,13 @@
     }
 
     @Override
-    public List<Map<String, Object>> selectDevicesByOrgId(Integer orgId) {
-        QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
-        queryWrapper.eq("is_delete", Constants.NOT_DELETE).eq("organization_id", orgId);
-        return deviceMapper.selectMaps(queryWrapper);
+    public Map<String, Object> selectDevicesByOrgId(Map<String, Object> parameters) {
+        return selectDevices(parameters);
     }
 
     @Override
-    public List<Map<String, Object>> selectDevicesByMpId(Integer mpId) {
-        QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
-        queryWrapper.eq("is_delete", Constants.NOT_DELETE).eq("monitor_point_id", mpId);
-        return deviceMapper.selectMaps(queryWrapper);
+    public Map<String, Object> selectDevicesByMpId(Map<String, Object> parameters) {
+        return selectDevices(parameters);
     }
 
     @Override
@@ -199,9 +259,25 @@
         Object orderType = parameters.get("orderType");
         Object name = parameters.get("name");
         Object mac = parameters.get("mac");
+        Object orgId = parameters.get("organizationId");
+        Object mpId = parameters.get("monitorPointId");
+
+        //������������������������������
+        if (orgId != null) {
+            queryWrapper.eq("organization_id", orgId);
+        }
+
+        //������������������������������
+        if (mpId != null) {
+            queryWrapper.eq("monitor_point_id", mpId);
+        }
+
+        //���������������������
         if (name != null) {
             queryWrapper.like("name", name);
         }
+
+        //mac������������
         if (mac != null) {
             queryWrapper.like("mac", mac);
         }
@@ -234,11 +310,10 @@
         return result;
     }
 
-
     @Override
     public Map<String, Object> selectDeviceInfoById(Integer deviceId) {
         String mac = deviceMapper.selectById(deviceId).getMac();
-        Map<String, Object> deviceInfo = (Map<String, Object>) RedisUtil.get("device_" + mac);
+        Map<String, Object> deviceInfo = getDeviceInfoFromRedis(mac);
         //������redis������
         if (deviceInfo != null) {
             return deviceInfo;
@@ -254,6 +329,9 @@
         deviceInfo.put("latitude", device.getLatitude());
         deviceInfo.put("createTime", DateUtils.dateToDateString(device.getCreateTime()));
         deviceInfo.put("installTime", device.getInstallTime() == null ? null : DateUtils.dateToDateString(device.getInstallTime()));
+
+        //������������
+        deviceInfo.put("extend", device.getExtend());
 
         //������
         deviceInfo.put("profession", device.getProfession());
@@ -301,8 +379,12 @@
         MonitorPoint monitorPoint = device.getMonitorPoint();
         mpInfo.put("id", monitorPoint.getId());
         mpInfo.put("name", monitorPoint.getName());
+        mpInfo.put("areaCode", monitorPoint.getAreaCode());
+        mpInfo.put("cityCode", monitorPoint.getCityCode());
+        mpInfo.put("provinceCode", monitorPoint.getProvinceCode());
         deviceInfo.put("monitorPoint", mpInfo);
-        RedisUtil.set("device_" + mac, deviceInfo);
+
+        setDeviceInfoToRedis(mac, deviceInfo);
         return deviceInfo;
     }
 
@@ -320,4 +402,131 @@
         return monitorPointMapper.selectMaps(queryWrapper);
     }
 
+    @Override
+    public Map<String, Object> getDeviceByMac(String mac) {
+        Map<String, Object> deviceInfo = getDeviceInfoFromRedis(mac);
+        if (deviceInfo == null) {
+            QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
+            queryWrapper.eq("mac", mac).eq("is_delete", Constants.NOT_DELETE);
+            Device device = deviceMapper.selectOne(queryWrapper);
+            if (device != null) {
+                deviceInfo = selectDeviceInfoById(device.getId());
+                setDeviceInfoToRedis(mac, deviceInfo);
+            }
+        }
+        return deviceInfo;
+    }
+
+    @Override
+    public Map<String, Object> adjustDeviceData(Map<String, Object> deviceData) {
+        String mac = deviceData.get("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> monitorPoint = (Map<String, Object>) deviceInfo.get("monitorPoint");
+            Object areaCode = monitorPoint.get("areaCode");
+            Object cityCode = monitorPoint.get("cityCode");
+
+            Map<String, Object> aqiMap = redisTemplate.opsForHash().entries(RedisConstants.AQI_DATA + areaCode);
+            if (ObjectUtils.isEmpty(aqiMap)) {
+                aqiMap = redisTemplate.opsForHash().entries(RedisConstants.AQI_DATA + cityCode);
+            }
+            return adjustDataUtils.adjust(deviceData, adjustFormula, ObjectUtils.isEmpty(aqiMap) ? null : aqiMap);
+        }
+        return deviceData;
+    }
+
+    @Override
+    public void judgeDeviceState(Map<String, Object> deviceData) {
+        String mac = deviceData.remove("mac").toString();
+        Device device = (Device) redisTemplate.opsForHash().get(RedisConstants.DEVICE_INFO, mac);
+        Version version = device.getVersion();
+        List<Sensor> sensors = version.getSensors();
+        Expression expression;
+        int state = 1;
+        for (Sensor sensor : sensors) {
+            //������������������
+            String alarmLevel = sensor.getAlarmLevel();
+            if (StringUtils.isEmpty(alarmLevel)) {
+                continue;
+            }
+            List<Double> 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));
+            //������������
+            if (formula != null) {
+                //������������������
+                sensorValue = formula.replace("{0}", sensorValue);
+                expression = AviatorEvaluator.compile(sensorValue);
+                value = Double.parseDouble(String.format("%.3f", expression.execute()));
+            }
+            int sensorState = judgeState(list, value);
+            if (sensorState > state) {
+                state = sensorState;
+            }
+        }
+        //������������������
+        UpdateWrapper<Device> updateWrapper = new UpdateWrapper<>();
+        updateWrapper.eq("id", device.getId()).set("state", state);
+        deviceMapper.update(null, updateWrapper);
+    }
+
+    //���������������������������
+    private int judgeState(List<Double> levels, Double data) {
+        int state = 1;
+        for (int i = levels.size() - 1; i >= 0; i--) {
+            Double level = levels.get(i);
+            if (data >= level) {
+                state = i + 2;
+                break;
+            }
+        }
+        return state;
+    }
+
+    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);
+        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);
+            List<VersionSensorUnit> versionSensorUnits = versionSensorUnitMapper.selectList(queryVersionSensorUnitWrapper);
+            if (!ObjectUtils.isEmpty(versionSensorUnits)) {
+                for (VersionSensorUnit versionSensorUnit : versionSensorUnits) {
+                    OrganizationUnitAlarm organizationUnitAlarm = new OrganizationUnitAlarm();
+                    organizationUnitAlarm.setOrganizationId(orgId);
+                    organizationUnitAlarm.setVersionId(versionId);
+                    organizationUnitAlarm.setSensorCode(versionSensorUnit.getSensorCode());
+                    organizationUnitAlarm.setUnitKey(versionSensorUnit.getUnitKey());
+                    organizationUnitAlarm.setShowUnitKey(versionSensorUnit.getUnitKey());
+                    organizationUnitAlarmMapper.insert(organizationUnitAlarm);
+                }
+            }
+        }
+    }
+
+    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);
+        List<Device> devices = deviceMapper.selectList(queryOrganizationVersionWrapper);
+        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);
+        }
+    }
 }

--
Gitblit v1.8.0