From c386b102a2397d569131e3a459465ba84e2376be Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Mon, 21 Aug 2023 14:27:51 +0800
Subject: [PATCH] Merge branch 'dev' of http://blit.7drlb.com:8888/r/moral into wb

---
 screen-job/src/main/java/com/moral/api/service/impl/CityAqiYearlyServiceImpl.java |   49 ++++++++++++++++++++++++++++++++++---------------
 1 files changed, 34 insertions(+), 15 deletions(-)

diff --git a/screen-job/src/main/java/com/moral/api/service/impl/CityAqiYearlyServiceImpl.java b/screen-job/src/main/java/com/moral/api/service/impl/CityAqiYearlyServiceImpl.java
index fbea9a0..92736ee 100644
--- a/screen-job/src/main/java/com/moral/api/service/impl/CityAqiYearlyServiceImpl.java
+++ b/screen-job/src/main/java/com/moral/api/service/impl/CityAqiYearlyServiceImpl.java
@@ -2,6 +2,7 @@
 
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.moral.api.entity.CityAqiDaily;
 import com.moral.api.entity.CityAqiYearly;
 import com.moral.api.mapper.CityAqiYearlyMapper;
@@ -17,7 +18,7 @@
 import org.springframework.util.ObjectUtils;
 
 import java.text.DecimalFormat;
-import java.util.Arrays;
+import java.util.ArrayList;
 import java.util.Date;
 import java.util.HashMap;
 import java.util.List;
@@ -50,7 +51,6 @@
          * SO2,NO2,���������98���������
          * O3���������������90���������
          * */
-        List<String> sensors = Arrays.asList("PM2_5", "PM10", "SO2", "NO2");
 
         //���������������������1���
         Date start = DateUtils.getFirstDayOfLastYear();
@@ -59,12 +59,26 @@
         //���������������������1���
         Date end = DateUtils.getDate(DateUtils.getDateAddYear(DateUtils.dateToDateString(start, DateUtils.yyyy), 1), DateUtils.yyyy);
 
+        //���������1���1���������������������������
+        //������1���1���������������������������������������������������������
+        String monthAndDay = DateUtils.dateToDateString(new Date(), DateUtils.MM_dd_EN);
+        UpdateWrapper<CityAqiYearly> cityAqiYearlyUpdateWrapper = new UpdateWrapper<>();
+        if ("01-01".equals(monthAndDay)) {
+            cityAqiYearlyUpdateWrapper.eq("time", start);
+        } else {
+            start = end;
+            end = DateUtils.getDate(DateUtils.getDateAddYear(DateUtils.dateToDateString(start, DateUtils.yyyy), 1), DateUtils.yyyy);
+            lastLastYear = DateUtils.getDate(DateUtils.getDateAddYear(DateUtils.dateToDateString(start, DateUtils.yyyy), -1), DateUtils.yyyy);
+            cityAqiYearlyUpdateWrapper.eq("time", start);
+        }
+        cityAqiYearlyMapper.delete(cityAqiYearlyUpdateWrapper);
+
 
         //������������������aqi���������
         QueryWrapper<CityAqiDaily> wrapper = new QueryWrapper<>();
         wrapper.select("city_code", "time", "value")
-                .ge("time", DateUtils.dateToDateString(start))
-                .lt("time", DateUtils.dateToDateString(end));
+                .ge("time", start)
+                .lt("time", end);
         List<Map<String, Object>> monthlyData = cityAqiDailyService.listMaps(wrapper);
 
         if (monthlyData.size() == 0) {
@@ -77,18 +91,23 @@
         //������������
         QueryWrapper<CityAqiYearly> queryWrapper = new QueryWrapper<>();
         queryWrapper.select("city_code", "value")
-                .eq("time", DateUtils.dateToDateString(lastLastYear));
+                .eq("time", lastLastYear);
         //������������������
-        List<CityAqiYearly> lastCityAqiYearly = cityAqiYearlyMapper.selectList(queryWrapper);
-        Map<Integer, List<CityAqiYearly>> lastYearData = lastCityAqiYearly.stream()
-                .collect(Collectors.groupingBy(CityAqiYearly::getCityCode));
+        List<CityAqiYearly> lastCityAqiYearlyList = cityAqiYearlyMapper.selectList(queryWrapper);
+        Map<Integer, CityAqiYearly> lastYearData = new HashMap<>();
+        for (CityAqiYearly cityAqiYearly : lastCityAqiYearlyList) {
+            lastYearData.put(cityAqiYearly.getCityCode(), cityAqiYearly);
+        }
 
 
-        CityAqiYearly cityAqiYearly = new CityAqiYearly();
+        List<CityAqiYearly> cityAqiYearlyList = new ArrayList<>();
+
+        Date finalStart = start;
         data.forEach((cityCode, value) -> {
+            CityAqiYearly cityAqiYearly = new CityAqiYearly();
             Map<String, Object> jsonMap = new HashMap<>();
             cityAqiYearly.setCityCode(Integer.parseInt(cityCode));
-            cityAqiYearly.setTime(start);
+            cityAqiYearly.setTime(finalStart);
 
             //PM2.5
             Double pm25Avg = AmendUtils.getAvgOfYear(value, "PM2_5");
@@ -119,17 +138,17 @@
             jsonMap.put("compositeIndex", compositeIndex);
 
             //���������������������������
-            List<CityAqiYearly> cityAqiYearlies = lastYearData.get(Integer.parseInt(cityCode));
-            if (!ObjectUtils.isEmpty(cityAqiYearlies)) {
-                CityAqiYearly yearly = cityAqiYearlies.get(0);
-                Map<String, Object> map = JSONObject.parseObject(yearly.getValue(), Map.class);
+            CityAqiYearly lastCityAqiYearly = lastYearData.get(Integer.parseInt(cityCode));
+            if (lastCityAqiYearly != null) {
+                Map<String, Object> map = JSONObject.parseObject(lastCityAqiYearly.getValue(), Map.class);
                 double lastCompositeIndex = Double.parseDouble(map.get("compositeIndex").toString());
                 DecimalFormat decimalFormat = new DecimalFormat("0.00%");
                 String format = decimalFormat.format((compositeIndex - lastCompositeIndex) / lastCompositeIndex);
                 jsonMap.put("yearContrast", format);
             }
             cityAqiYearly.setValue(JSONObject.toJSONString(jsonMap));
-            cityAqiYearlyMapper.insert(cityAqiYearly);
+            cityAqiYearlyList.add(cityAqiYearly);
         });
+        cityAqiYearlyMapper.insertCityAqiYearly(cityAqiYearlyList);
     }
 }

--
Gitblit v1.8.0