From 58fdbd07beae496c4c3b0744efe86a7d82e382b7 Mon Sep 17 00:00:00 2001
From: cjl <276999030@qq.com>
Date: Tue, 22 Aug 2023 16:59:34 +0800
Subject: [PATCH] fix:部分功能优化

---
 screen-api/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java |   86 +++++++++++++++++++++++++++---------------
 1 files changed, 55 insertions(+), 31 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java b/screen-api/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java
index 10d5f6d..ccfb212 100644
--- a/screen-api/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java
+++ b/screen-api/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java
@@ -6,11 +6,9 @@
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.moral.api.config.properties.SpecialCitiesProperties;
+import com.moral.api.dto.CityAqiDailyListDTO;
 import com.moral.api.entity.*;
-import com.moral.api.mapper.CityAqiMapper;
-import com.moral.api.mapper.DeviceMapper;
-import com.moral.api.mapper.ForecastMapper;
-import com.moral.api.mapper.SysAreaMapper;
+import com.moral.api.mapper.*;
 import com.moral.api.pojo.dto.cityAQI.CityPollutionLevel;
 import com.moral.api.pojo.dto.cityAQI.ConcentrationAndPercent;
 import com.moral.api.pojo.form.aqi.AirQualityComparisonForm;
@@ -28,6 +26,7 @@
 
 import com.moral.util.MathUtils;
 
+import org.apache.commons.collections4.CollectionUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Service;
@@ -52,6 +51,8 @@
 
     @Autowired
     private CityAqiMapper cityAqiMapper;
+    @Autowired
+    private CityAqiDailyMapper cityAqiDailyMapper;
 
     @Autowired
     private ForecastMapper forecastMapper;
@@ -208,7 +209,7 @@
     @Override
     public Map<String, Object> query24HoursAqiByRegionCode(Integer regionCode) {
         //������������������������������������������������������
-        QueryWrapper<CityAqi> lastDataWrapper = new QueryWrapper<>();
+        /*QueryWrapper<CityAqi> lastDataWrapper = new QueryWrapper<>();
         lastDataWrapper.eq("city_code", regionCode);
         lastDataWrapper.orderByDesc("time");
         lastDataWrapper.last(true, "limit 1");
@@ -221,39 +222,33 @@
         //������������
         QueryWrapper<CityAqi> wrapper = new QueryWrapper<>();
         wrapper.between("time", startDate, endDate);
-        wrapper.eq("city_code", regionCode);
-        List<CityAqi> cityAqis = cityAqiMapper.selectList(wrapper);
+        wrapper.eq("city_code", regionCode);*/
+        List<CityAqi> cityAqis = cityAqiMapper.listCity(regionCode);
         //������������������24���������������
-        if (cityAqis.size() != 24) {
-            Map<Date, CityAqi> dateCityAqiMap = new HashMap<>();
-            cityAqis.forEach(value -> dateCityAqiMap.put(value.getTime(), value));
+        if(CollectionUtils.isEmpty(cityAqis)){
+            return new HashMap<>();
+        }
+        Date startDate = cityAqis.get(0).getTime();
+        if(cityAqis.size()<24){
             for (int i = 0; i < 24; i++) {
-                Date date = DateUtils.addHours(startDate, i);
-                CityAqi cityAqi1 = dateCityAqiMap.get(date);
-                if (cityAqi1 == null) {
+                Date dateTime = DateUtils.addHours(startDate, i);
+                if(cityAqis.size()-1 <i){
                     CityAqi newCityAqi = new CityAqi();
-                    newCityAqi.setTime(date);
+                    newCityAqi.setTime(dateTime);
+                    newCityAqi.setValue("0");
                     cityAqis.add(newCityAqi);
+                    continue;
                 }
             }
-            //������������������������
-            cityAqis.sort(Comparator.comparing(CityAqi::getTime));
         }
+            //������������������������
+        cityAqis.sort(Comparator.comparing(CityAqi::getTime));
+
         //������������������,map���key���yyyy-MM-dd HH:mm������������������value���aqi���������
         Map<String, Object> result = new LinkedHashMap<>();
         for (CityAqi aqi : cityAqis) {
             String key = DateUtils.dateToDateString(aqi.getTime(), "yyyy-MM-dd HH:mm");
-            String allDataJson = aqi.getValue();
-            if (allDataJson == null) {
-                result.put(key, "");
-                continue;
-            }
-            Map<String, Object> allDataMap = JSON.parseObject(allDataJson, Map.class);
-            Object aqiData = allDataMap.get("AQI");
-            if (aqiData == null)
-                result.put(key, "");
-            else
-                result.put(key, aqiData);
+            result.put(key,aqi.getValue());
         }
         return result;
     }
@@ -691,14 +686,35 @@
 
         Date now = new Date();
         //������
-        Date yesterday = DateUtils.dataToTimeStampTime(DateUtils.getDateOfDay(now, -1), DateUtils.yyyy_MM_dd_EN);
-        String dateString = DateUtils.dateToDateString(yesterday, DateUtils.yyyy_MM_dd_HH_mm_ss_EN);
+        Date yesterday = DateUtils.dataToTimeStampTime(DateUtils.getDateOfDay(now, 0), DateUtils.yyyy_MM_dd_EN);
+        String dateString = DateUtils.dateToDateString(yesterday, DateUtils.yyyy_MM_dd_EN);
 
         String s = String.valueOf(regionCode);
         //���������,���code
         Integer provinceCode = Integer.parseInt(s.substring(0, 2) + "0000");
         Integer cityCode = Integer.parseInt(s.substring(0, 4) + "00");
-        //������������������city_code
+        Map<String,Object> params = new HashMap<>();
+        params.put("cityCode",provinceCode);
+        params.put("start",dateString);
+        params.put("end",dateString);
+        params.put("type",1);
+        List<CityAqiDailyListDTO> dayList = cityAqiDailyMapper.CityAqiDailyMap(params);
+        String start = DateUtils.dateToDateString(DateUtils.addMonths(DateUtils.getFirstDayOfLastMonth(), 1));
+        params.put("start",start);
+        params.put("type",2);
+        List<CityAqiDailyListDTO> monList = cityAqiDailyMapper.CityAqiDailyMap(params);
+        start =DateUtils.dateToDateString(now, DateUtils.yyyy);
+        params.put("start",start);
+        params.put("end",start);
+        params.put("type",3);
+        List<CityAqiDailyListDTO> yearList = cityAqiDailyMapper.CityAqiDailyMap(params);
+        int indexDay = dayList.stream().map(CityAqiDailyListDTO::getCityCode).collect(Collectors.toList()).indexOf(cityCode.toString());
+        int indexMon = monList.stream().map(CityAqiDailyListDTO::getCityCode).collect(Collectors.toList()).indexOf(cityCode.toString());
+        int indexYear = yearList.stream().map(CityAqiDailyListDTO::getCityCode).collect(Collectors.toList()).indexOf(cityCode.toString());
+        result.put("day",mapResult(dayList,indexDay));
+        result.put("month",mapResult(monList,indexMon));
+        result.put("year",mapResult(yearList,indexYear));
+       /* //������������������city_code
         QueryWrapper<SysArea> wrapper = new QueryWrapper<>();
         wrapper.select("area_code").eq("parent_code", provinceCode);
         List<Object> cityCodes = sysAreaService.listObjs(wrapper);
@@ -786,13 +802,21 @@
             yearMap.put("size", null);
         }
         yearMap.put("compositeIndex", yearMap.remove("value"));
-        result.put("year", yearMap);
+        result.put("year", yearMap);*/
 
         //���������������
         result.put("time", DateUtils.dateToDateString(yesterday, DateUtils.yyyy_MM_dd_EN));
         return result;
     }
 
+    private Map<String,Object> mapResult(List<CityAqiDailyListDTO> list, int length){
+        Map<String,Object> map = new HashMap<>();
+        map.put("AQI",CollectionUtils.isNotEmpty(list)&&length>0&&Objects.nonNull(list.get(length-1))?list.get(length-1).getAqi():0);
+        map.put("rank",CollectionUtils.isNotEmpty(list)&&length>0?length:0);
+        map.put("size",CollectionUtils.isNotEmpty(list)&&length>0?list.size():0);
+        return map;
+    }
+
     @Override
     public List<AirQualityComparisonVO> queryAirQualityComparison(AirQualityComparisonForm form) {
         //������

--
Gitblit v1.8.0