From bd9a7f2d97af968d43ccd5516644f6933b42f5b2 Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Thu, 30 Dec 2021 11:47:59 +0800
Subject: [PATCH] screen-api            增加城市优良天气对比接口

---
 screen-api/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java |  153 ++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 108 insertions(+), 45 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java b/screen-api/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java
index 93b9cf5..44e67e4 100644
--- a/screen-api/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java
+++ b/screen-api/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java
@@ -4,13 +4,17 @@
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.moral.api.entity.CityAqiDaily;
+import com.moral.api.entity.SysArea;
 import com.moral.api.mapper.CityAqiDailyMapper;
+import com.moral.api.pojo.dto.cityAQI.CityPollutionLevel;
 import com.moral.api.pojo.dto.cityAQI.MonthlyPollutionLevel;
 import com.moral.api.pojo.dto.cityAQI.PollutionDaysAndProportion;
+import com.moral.api.pojo.form.aqi.AreaPollutionLevelForm;
 import com.moral.api.pojo.vo.cityAQI.PieChartOfPollutionLevelVO;
-import com.moral.api.pojo.form.aqi.QueryChartOfPollutionLevelForm;
+import com.moral.api.pojo.form.aqi.ChartOfPollutionLevelForm;
 import com.moral.api.service.CityAqiDailyService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.moral.api.service.SysAreaService;
 import com.moral.constant.Constants;
 import com.moral.util.AQIUtils;
 import com.moral.util.DateUtils;
@@ -35,9 +39,11 @@
 
     @Autowired
     CityAqiDailyMapper cityAqiDailyMapper;
+    @Autowired
+    SysAreaService sysAreaService;
 
     @Override
-    public PieChartOfPollutionLevelVO queryPieChartOfPollutionLevels(QueryChartOfPollutionLevelForm form) {
+    public PieChartOfPollutionLevelVO queryPieChartOfPollutionLevels(ChartOfPollutionLevelForm form) {
         //������
         Integer regionCode = form.getRegionCode();
         Date startDate = form.getStartDate();
@@ -106,7 +112,7 @@
     }
 
     @Override
-    public List<MonthlyPollutionLevel> queryBarChartOfPollutionLevels(QueryChartOfPollutionLevelForm form) {
+    public List<MonthlyPollutionLevel> queryBarChartOfPollutionLevels(ChartOfPollutionLevelForm form) {
         //������
         Integer regionCode = form.getRegionCode();
         Date startDate = form.getStartDate();
@@ -138,7 +144,7 @@
         List<MonthlyPollutionLevel> monthlyPollutionLevels = new ArrayList<>();
         Set<Map.Entry<String, List<CityAqiDaily>>> monthlyEntries = monthlyDataMap.entrySet();
         Iterator<Map.Entry<String, List<CityAqiDaily>>> monthlyIterator = monthlyEntries.iterator();
-        while(monthlyIterator.hasNext()){
+        while (monthlyIterator.hasNext()) {
             Map.Entry<String, List<CityAqiDaily>> monthlyEntry = monthlyIterator.next();
             List<CityAqiDaily> monthlyValues = monthlyEntry.getValue();
             //���������������������������������������
@@ -187,44 +193,44 @@
     }
 
     @Override
-    public Map<String, Map<String,Object>> oneYearsData(Map map) {
+    public Map<String, Map<String, Object>> oneYearsData(Map map) {
         int city_code = Integer.parseInt(map.get("cityCode").toString());
         String year = map.get("year").toString();
-        String startTime = year+"-01-01 00:00:00";
-        String endTime = year+"-12-31 23:59:59";
+        String startTime = year + "-01-01 00:00:00";
+        String endTime = year + "-12-31 23:59:59";
         QueryWrapper<CityAqiDaily> cityAqiDailyQueryWrapper = new QueryWrapper<>();
         cityAqiDailyQueryWrapper.eq("city_code", city_code);
-        cityAqiDailyQueryWrapper.between("time",startTime,endTime);
+        cityAqiDailyQueryWrapper.between("time", startTime, endTime);
         List<CityAqiDaily> cityAqiDailies = cityAqiDailyMapper.selectList(cityAqiDailyQueryWrapper);
-        Map<String,Map<String,Object>> resultMap = new HashMap<>();
-        for (int i=1;i<13;i++){
+        Map<String, Map<String, Object>> resultMap = new HashMap<>();
+        for (int i = 1; i < 13; i++) {
             Map<String, Object> monthMap = new HashMap<>();
             String month = null;
-            if (i<10){
-                month = "0"+i;
-            }else {
+            if (i < 10) {
+                month = "0" + i;
+            } else {
                 month = String.valueOf(i);
             }
-            String monthDate = year+"-"+month;
-            String monthFirstDay = year+"-"+month+"-01";
+            String monthDate = year + "-" + month;
+            String monthFirstDay = year + "-" + month + "-01";
             int monthDay = DateUtils.getMonthDay(monthFirstDay);
-            for (int j=1;j<monthDay+1;j++){
+            for (int j = 1; j < monthDay + 1; j++) {
                 String day = null;
-                if (j<10){
-                    day = "0"+j;
-                }else {
+                if (j < 10) {
+                    day = "0" + j;
+                } else {
                     day = String.valueOf(j);
                 }
-                String dayDate = year+"-"+month+"-"+day;
-                monthMap.put(dayDate,null);
+                String dayDate = year + "-" + month + "-" + day;
+                monthMap.put(dayDate, null);
             }
-            resultMap.put(monthDate,monthMap);
+            resultMap.put(monthDate, monthMap);
         }
         SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
-        for (CityAqiDaily cityAqiDaily:cityAqiDailies) {
+        for (CityAqiDaily cityAqiDaily : cityAqiDailies) {
             String time = sdf.format(cityAqiDaily.getTime());
-            String monthDate = time.substring(0,7);
-            String dayDate = time.substring(0,10);
+            String monthDate = time.substring(0, 7);
+            String dayDate = time.substring(0, 10);
             Double PM2_5 = null;
             Double PM10 = null;
             Double SO2 = null;
@@ -236,45 +242,102 @@
             String value = cityAqiDaily.getValue();
             JSONObject jsonObject = new JSONObject();
             jsonObject = JSONObject.parseObject(value);
-            if (!ObjectUtils.isEmpty(jsonObject.get("PM2_5"))){
+            if (!ObjectUtils.isEmpty(jsonObject.get("PM2_5"))) {
                 PM2_5 = (Double.parseDouble(jsonObject.get("PM2_5").toString()));
             }
-            if (!ObjectUtils.isEmpty(jsonObject.get("PM10"))){
+            if (!ObjectUtils.isEmpty(jsonObject.get("PM10"))) {
                 PM10 = (Double.parseDouble(jsonObject.get("PM10").toString()));
             }
-            if (!ObjectUtils.isEmpty(jsonObject.get("SO2"))){
+            if (!ObjectUtils.isEmpty(jsonObject.get("SO2"))) {
                 SO2 = (Double.parseDouble(jsonObject.get("SO2").toString()));
             }
-            if (!ObjectUtils.isEmpty(jsonObject.get("NO2"))){
+            if (!ObjectUtils.isEmpty(jsonObject.get("NO2"))) {
                 NO2 = (Double.parseDouble(jsonObject.get("NO2").toString()));
             }
-            if (!ObjectUtils.isEmpty(jsonObject.get("CO"))){
+            if (!ObjectUtils.isEmpty(jsonObject.get("CO"))) {
                 CO = (Double.parseDouble(jsonObject.get("CO").toString()));
             }
-            if (!ObjectUtils.isEmpty(jsonObject.get("O3"))){
+            if (!ObjectUtils.isEmpty(jsonObject.get("O3"))) {
                 O3 = (Double.parseDouble(jsonObject.get("O3").toString()));
             }
-            if (!ObjectUtils.isEmpty(jsonObject.get("primaryPollutant"))){
+            if (!ObjectUtils.isEmpty(jsonObject.get("primaryPollutant"))) {
                 primaryPollutant = jsonObject.get("primaryPollutant").toString();
             }
-            if (!ObjectUtils.isEmpty(jsonObject.get("AQI"))){
+            if (!ObjectUtils.isEmpty(jsonObject.get("AQI"))) {
                 AQI = jsonObject.get("AQI").toString();
             }
-            Map<String,Object> valueMap = new HashMap<>();
-            valueMap.put("PM2_5",PM2_5);
-            valueMap.put("PM10",PM10);
-            valueMap.put("SO2",SO2);
-            valueMap.put("NO2",NO2);
-            valueMap.put("CO",CO);
-            valueMap.put("O3",O3);
-            valueMap.put("AQI",AQI);
-            valueMap.put("primaryPollutant",primaryPollutant);
-            Map<String,Object> monthMap = new HashMap<>();
+            Map<String, Object> valueMap = new HashMap<>();
+            valueMap.put("PM2_5", PM2_5);
+            valueMap.put("PM10", PM10);
+            valueMap.put("SO2", SO2);
+            valueMap.put("NO2", NO2);
+            valueMap.put("CO", CO);
+            valueMap.put("O3", O3);
+            valueMap.put("AQI", AQI);
+            valueMap.put("primaryPollutant", primaryPollutant);
+            Map<String, Object> monthMap = new HashMap<>();
             monthMap = resultMap.get(monthDate);
-            monthMap.put(dayDate,valueMap);
-            resultMap.put(monthDate,monthMap);
+            monthMap.put(dayDate, valueMap);
+            resultMap.put(monthDate, monthMap);
         }
         return resultMap;
     }
 
+    @Override
+    public List<CityPollutionLevel> queryAreaPollutionLevel(AreaPollutionLevelForm form) {
+        //������
+        Integer regionCode = form.getRegionCode();
+        Date startDate = form.getStartDate();
+        Date endDate = form.getEndDate();
+        //������������������������������
+        List<SysArea> allAreas = sysAreaService.getChildren(regionCode);
+        //������������������������
+        List<CityPollutionLevel> datas = new ArrayList<>();
+        //��������������������� ������������
+        for (SysArea area : allAreas) {
+            //������������
+            QueryWrapper<CityAqiDaily> wrapper = new QueryWrapper<>();
+            wrapper.eq("city_code", area.getAreaCode());
+            wrapper.between("time", startDate, endDate);
+            wrapper.orderByAsc("time");
+            List<CityAqiDaily> cityAqiDailies = cityAqiDailyMapper.selectList(wrapper);
+            //���������������������������������������
+            Map<String, CityAqiDaily> tmpMap = new LinkedHashMap<>();
+            for (CityAqiDaily cityAqiDaily : cityAqiDailies) {
+                String tmpTime = DateUtils.dateToDateString(cityAqiDaily.getTime(), "yyyy-MM-dd");
+                tmpMap.put(tmpTime, cityAqiDaily);
+            }
+            cityAqiDailies = new ArrayList<CityAqiDaily>(tmpMap.values());
+            //���������������������������������������
+            Map<String, Integer> pollutionDaysMap = new LinkedHashMap<>();
+            pollutionDaysMap.put(Constants.EXCELLENT_WEATHER_CODE, 0);
+            pollutionDaysMap.put(Constants.GOOD_WEATHER_CODE, 0);
+            pollutionDaysMap.put(Constants.MILD_WEATHER_CODE, 0);
+            pollutionDaysMap.put(Constants.MIDDLE_WEATHER_CODE, 0);
+            pollutionDaysMap.put(Constants.SERIOUS_WEATHER_CODE, 0);
+            pollutionDaysMap.put(Constants.SERVER_WEATHER_CODE, 0);
+            for (CityAqiDaily cityAqiDaily : cityAqiDailies) {
+                Map<String, Object> valueMap = JSON.parseObject(cityAqiDaily.getValue(), Map.class);
+                if (valueMap.get("AQI") == null)
+                    continue;
+                Integer aqi = Integer.valueOf(valueMap.get("AQI").toString());
+                //������AQI������������
+                String pollutionClass = AQIUtils.classCodeOfPollutionByAqi(aqi);
+                Integer days = pollutionDaysMap.get(pollutionClass);
+                pollutionDaysMap.put(pollutionClass, days + 1);
+            }
+            //������������
+            CityPollutionLevel cityPollutionLevel = new CityPollutionLevel();
+            cityPollutionLevel.setRegionName(area.getAreaName());
+            cityPollutionLevel.setExcellentWeatherDays(pollutionDaysMap.get(Constants.EXCELLENT_WEATHER_CODE));
+            cityPollutionLevel.setGoodWeatherDays(pollutionDaysMap.get(Constants.GOOD_WEATHER_CODE));
+            cityPollutionLevel.setMildWeatherDays(pollutionDaysMap.get(Constants.MILD_WEATHER_CODE));
+            cityPollutionLevel.setMiddleWeatherDays(pollutionDaysMap.get(Constants.MIDDLE_WEATHER_CODE));
+            cityPollutionLevel.setSeriousWeatherDays(pollutionDaysMap.get(Constants.SERIOUS_WEATHER_CODE));
+            cityPollutionLevel.setServerWeatherDays(pollutionDaysMap.get(Constants.SERVER_WEATHER_CODE));
+            datas.add(cityPollutionLevel);
+        }
+        return datas;
+    }
+
 }

--
Gitblit v1.8.0