From 21a44d6cb9a372bce5c7418d2a82c88bb0485e60 Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Mon, 17 Jan 2022 14:44:56 +0800
Subject: [PATCH] screen-api            完成城市空气质量同期对比接口

---
 screen-api/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java |   98 ++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 92 insertions(+), 6 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java b/screen-api/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java
index 561c8b6..3a68890 100644
--- a/screen-api/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java
+++ b/screen-api/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java
@@ -1,5 +1,6 @@
 package com.moral.api.service.impl;
 
+import com.alibaba.fastjson.JSON;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.moral.api.entity.Device;
 import com.moral.api.entity.Sensor;
@@ -13,11 +14,14 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Service;
+import org.springframework.util.ObjectUtils;
 
 import java.util.ArrayList;
 
+import java.util.Arrays;
 import java.util.HashMap;
 
+import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 
@@ -48,6 +52,20 @@
 
     @Override
     public Map<String, Object> getSensorsByMac(Map<String, Object> params) {
+        List<String> sensorCodes = Arrays.asList(Constants.SENSOR_CODE_PM25
+                , Constants.SENSOR_CODE_PM10
+                , Constants.SENSOR_CODE_SO2
+                , Constants.SENSOR_CODE_NO2
+                , Constants.SENSOR_CODE_CO
+                , Constants.SENSOR_CODE_O3
+                , Constants.SENSOR_CODE_TEMP
+                , "a01002"
+                , Constants.SENSOR_CODE_WIND_SPEED
+                , Constants.SENSOR_CODE_WIND_DIR
+                , "a01006"
+                , "a00e12"
+        );
+
         //������mac
         List<String> macs = (List<String>) params.remove("macs");
         List<Map<String, Object>> elementLists = new ArrayList<>();
@@ -65,13 +83,21 @@
             elementLists.add(map);
         }
 
-        Map<String, Object> map = elementLists.parallelStream()
+        Map<String, Object> map = elementLists.stream()
                 .filter(elementList -> elementList.size() != 0)
                 .reduce((a, b) -> {
                     a.keySet().retainAll(b.keySet());
                     return a;
                 }).orElse(new HashMap<>());
-        return map;
+        Map<String, Object> result = new LinkedHashMap<>();
+        sensorCodes.forEach(sensorCode -> {
+            Object o = map.remove(sensorCode);
+            if (o != null) {
+                result.put(sensorCode, o);
+            }
+        });
+        result.putAll(map);
+        return result;
     }
 
     @Override
@@ -84,33 +110,36 @@
         queryWrapper.select("mac", "name").in("mac", macs);
         List<Device> devices = deviceMapper.selectList(queryWrapper);
 
-
         //������������
         List<String> times = (List<String>) params.remove("times");
         //������code
         String sensorCode = params.get("sensorCode").toString();
         String end;
         String timeUnits;
-
+        String dateFormat;
         //���������������,time=data
         List<Map<String, Object>> result = new ArrayList<>();
 
         for (String start : times) {
             if ("hour".equals(type)) {
                 end = DateUtils.getDateAddDay(start, 1);
-                timeUnits = "hourly";
+                String yearAndMonth = DateUtils.dateToDateString(DateUtils.getDate(start, DateUtils.yyyy_MM_dd_EN), DateUtils.yyyyMM_EN);
+                timeUnits = "hourly_" + yearAndMonth;
+                dateFormat = "%Y-%m-%d %H";
             } else if ("day".equals(type)) {
                 end = DateUtils.getDateAddMonth(start, 1);
                 timeUnits = "daily";
+                dateFormat = "%Y-%m-%d";
             } else {
                 end = DateUtils.getDateAddYear(start, 1);
                 timeUnits = "monthly";
+                dateFormat = "%Y-%m";
             }
             params.put("timeUnits", timeUnits);
             params.put("start", start);
             params.put("end", end);
             params.put("macs", macs);
-            params.put("dateFormat", "%Y-%m-%d %H:%i:%s");
+            params.put("dateFormat", dateFormat);
             //���������������������������������
             List<Map<String, Object>> list = deviceMapper.getTrendChartData(params);
 
@@ -139,5 +168,62 @@
         return result;
     }
 
+    @Override
+    public Device getDeviceByMac(String mac) {
+        Map<String, Object> deviceMap = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.DEVICE, mac);
+        Device device = JSON.parseObject(JSON.toJSONString(deviceMap), Device.class);
+        //���map������organizationId���monitorPointId������versionId
+        Map<String, Object> organizationMap = (Map<String, Object>) deviceMap.get("organization");
+        Map<String, Object> monitorPointMap = (Map<String, Object>) deviceMap.get("monitorPoint");
+        Map<String, Object> versionMap = (Map<String, Object>) deviceMap.get("version");
+        device.setDeviceVersionId((Integer) versionMap.get("id"));
+        device.setOrganizationId((Integer) organizationMap.get("id"));
+        device.setMonitorPointId((Integer) monitorPointMap.get("id"));
+        //������������������������������������
+        if (ObjectUtils.isEmpty(device)) {
+            return getDeviceByMacFromDB(mac);
+        }
+        return device;
+    }
+
+    @Override
+    public List<Map<String, Object>> getDevicesByOrganizationId(Integer orgId) {
+        //������������������mac
+        List macs = getMacsByOrganizationId(orgId);
+        //���redis������������������������
+        List<Map<String, Object>> result = new ArrayList<>();
+        for (Object mac : macs) {
+            Map<String, Object> map = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.DEVICE, mac.toString());
+            result.add(map);
+        }
+        return result;
+    }
+
+    @Override
+    public List getMacsByOrganizationId(Integer organizationId) {
+        QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
+        queryWrapper.select("mac")
+                .eq("organization_id", organizationId)
+                .eq("is_delete", Constants.NOT_DELETE);
+        return deviceMapper.selectObjs(queryWrapper);
+    }
+
+    @Override
+    public List getMacsByOrgIdAndRegionCode(Integer organizationId, Integer regionCode) {
+        QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
+        queryWrapper.select("mac")
+                .eq("organization_id", organizationId)
+                .eq("is_delete", Constants.NOT_DELETE)
+                .eq("town_code", regionCode);
+        return deviceMapper.selectObjs(queryWrapper);
+    }
+
+    private Device getDeviceByMacFromDB(String mac) {
+        QueryWrapper<Device> wrapper = new QueryWrapper<>();
+        wrapper.eq("mac", mac);
+        wrapper.eq("is_delete", Constants.NOT_DELETE);
+        return deviceMapper.selectOne(wrapper);
+    }
+
 
 }

--
Gitblit v1.8.0