From c75b4210328b98913de7eeef507e6d88069bcecc Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Thu, 13 Aug 2020 11:19:48 +0800
Subject: [PATCH] 根据算法预测明天O3,插入数据库

---
 src/main/java/com/moral/service/impl/ForecastWeatherServiceImpl.java |  170 +++++++++++++++++++++++++++++++++++++++++++++-----------
 1 files changed, 137 insertions(+), 33 deletions(-)

diff --git a/src/main/java/com/moral/service/impl/ForecastWeatherServiceImpl.java b/src/main/java/com/moral/service/impl/ForecastWeatherServiceImpl.java
index 00accb9..949895d 100644
--- a/src/main/java/com/moral/service/impl/ForecastWeatherServiceImpl.java
+++ b/src/main/java/com/moral/service/impl/ForecastWeatherServiceImpl.java
@@ -17,17 +17,17 @@
 import org.springframework.stereotype.Service;
 import org.springframework.web.client.RestTemplate;
 
-import com.alibaba.fastjson.JSON;
-import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.moral.entity.Area;
 import com.moral.entity.City;
 import com.moral.entity.MonitorPoint;
+import com.moral.mapper.AQIMapper;
 import com.moral.mapper.AreaMapper;
 import com.moral.mapper.CityMapper;
 import com.moral.mapper.ForecastWeatherMapper;
 import com.moral.mapper.MonitorPointMapper;
 import com.moral.mapper.ProvinceMapper;
+import com.moral.mapper.RealWeatherMapper;
 import com.moral.service.ForecastWeatherService;
 import com.moral.util.Dom4jUtils;
 
@@ -49,8 +49,14 @@
     @Resource
     private ProvinceMapper provinceMapper;
 
+    @Resource
+    private RealWeatherMapper realWeatherMapper;
+
+    @Resource
+    private AQIMapper aqiMapper;
+
     @Override
-    public void insertForecastWeather() throws ParseException {
+    public int insertForecastWeather() throws ParseException {
         Calendar c = Calendar.getInstance();
         SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
         SimpleDateFormat sdf1 = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
@@ -64,39 +70,40 @@
         RestTemplate restTemplate = new RestTemplate();
         List<MonitorPoint> monitorPointList = monitorPointMapper.getMonitorPointList();
         Set<Map<String, Object>> hashSet = new HashSet<>();
-        List<Map<String, Object>> resultList = new ArrayList<>();
         for (MonitorPoint monitorPoint : monitorPointList) {
-            Map<String, Object> hashMap1 = new HashMap<>();
             Map<String, Object> hashMap = new HashMap<>();
             Integer areaCode = monitorPoint.getAreaCode();
             Integer cityCode = monitorPoint.getCityCode();
             Integer provinceCode = monitorPoint.getProvinceCode();
             String parentName = "";
             String name1 = "";
+            Integer code = 0;
             if (areaCode != null) {
                 Area area = areaMapper.getAreaByAreaCode(areaCode);
                 City city = cityMapper.getCityByCityCode(cityCode);
                 name1 = area.getAreaName();
                 parentName = city.getCityName();
+                code = areaCode;
                 if ("���������".equals(name1)) {
                     name1 = parentName;
                     parentName = provinceMapper.getProvinceByProvinceCode(provinceCode).getProvinceName();
+                    code = cityCode;
                 }
             } else {
                 City city = cityMapper.getCityByCityCode(cityCode);
                 name1 = city.getCityName();
                 parentName = provinceMapper.getProvinceByProvinceCode(provinceCode).getProvinceName();
+                code = cityCode;
                 if ("���������".equals(name1)) {
                     name1 = parentName;
                     parentName = "";
+                    code = provinceCode;
                 }
             }
             hashMap.put("name1", name1);
             hashMap.put("parentName", parentName);
+            hashMap.put("cityCode", code);
             hashSet.add(hashMap);
-            hashMap1.put("monitorPointId", monitorPoint.getId());
-            hashMap1.put("name", name1);
-            resultList.add(hashMap1);
         }
         List<Element> elements = Dom4jUtils.readDocument();
         String cityID = "101190404";
@@ -123,11 +130,96 @@
             }
         }
 
+        c.setTime(new Date());
+        Date endTime = sdf.parse(sdf.format(c.getTime()));
+        c.add(Calendar.MONTH, -2);
+        Date startTime = sdf.parse(sdf.format(c.getTime()));
+
+        for (Map<String, Object> hashmap : hashSet) {
+            hashmap.put("typeFormat", "%Y-%m-%d %H:%i:%s");
+            hashmap.put("startTime", startTime);
+            hashmap.put("endTime", endTime);
+            List<List<Map<String, Object>>> list1 = new ArrayList<>();
+            List<List<Map<String, Object>>> list2 = new ArrayList<>();
+
+            for (int i = 0; i < 24; i++) {
+                List<Map<String, Object>> list3 = new ArrayList<>();
+                List<Map<String, Object>> list4 = new ArrayList<>();
+                if (i < 10) {
+                    hashmap.put("time", "0" + i + ":00:00");
+                } else {
+                    hashmap.put("time", i + ":00:00");
+                }
+                List<Map<String, Object>> tempList = realWeatherMapper.getTemp(hashmap);
+                List<Map<String, Object>> O3List = aqiMapper.getO3(hashmap);
+                list3.addAll(tempList);
+                list4.addAll(O3List);
+                list1.add(list3);
+                list2.add(list4);
+            }
+
+            List<List<Map<String, Object>>> list = new ArrayList<>();
+            if (list2.get(0).size() == 0) {
+                list = list1;
+            } else {
+                for (List<Map<String, Object>> tempList : list1) {
+                    List<Map<String, Object>> list3 = new ArrayList<>();
+                    for (List<Map<String, Object>> O3List : list2) {
+                        for (Map<String, Object> tempMap : tempList) {
+                            for (Map<String, Object> O3Map : O3List) {
+                                if (O3Map.get("O3") != null) {
+                                    if (tempMap.get("time").equals(O3Map.get("time"))) {
+                                        Map<String, Object> map = new HashMap<>();
+                                        map.put("time", tempMap.get("time"));
+                                        map.put("temp", tempMap.get("temp").toString());
+                                        map.put("O3", O3Map.get("O3").toString());
+                                        list3.add(map);
+                                    }
+                                }
+                            }
+                        }
+                    }
+                    list.add(list3);
+                }
+            }
+
+            if (list.get(0).get(0).size() > 2) {
+                List<Map<String, Object>> constantList = new ArrayList<>();
+                for (int i = 0; i < list.size(); i++) {
+                    List<Map<String, Object>> mapList = list.get(i);
+                    double tempSum = 0;//���������
+                    double O3Sum = 0;//O3���
+                    double sum = 0;//
+                    double sum1 = 0;
+                    int length = mapList.size();
+                    Map<String, Object> cmap = new HashMap<>();
+                    cmap.put("time", i);
+                    for (Map<String, Object> map : mapList) {
+                        double temp = Double.valueOf(map.get("temp").toString().replace("\"", "").split("\\.")[0]);
+                        double O3 = Double.valueOf(map.get("O3").toString().replace("\"", "").split("\\.")[0]);
+                        tempSum = tempSum + temp;
+                        O3Sum = O3Sum + O3;
+                        sum = sum + temp * O3;
+                        sum1 = sum1 + temp * temp;
+                    }
+                    double tempAvg = tempSum / length;
+                    double O3Avg = O3Sum / length;
+                    double b = (sum - length * tempAvg * O3Avg) / (sum1 - length * tempAvg * tempAvg);
+                    double a = O3Avg - (b * tempAvg);
+                    cmap.put("a", a);
+                    cmap.put("b", b);
+                    constantList.add(cmap);
+                }
+                hashmap.put("constants", constantList);
+            }
+        }
+
+        int count = 0;
         for (Map<String, Object> map : hashSet) {
             String id = map.get("cityID").toString();
             Map<String, Object> dataMap = restTemplate.getForObject("https://api.heweather.net/v7/weather/72h?key=da05c6c4852d4f7aa3364a9236ee9e26&gzip=n&location={1}", Map.class, id);
             if (dataMap == null) {
-                dataMap = new HashMap<>();
+                return count;
             }
             String json = JSONObject.toJSONString(dataMap);
             dataMap = (Map<String, Object>) JSONObject.parse(json);
@@ -139,45 +231,57 @@
                 String hour1 = hourlyMap.get("fxTime").toString().split("\\+")[0].split(":")[0].replace("T", " ");
                 if (hour.equals(nextDay)) {
                     nextDayList.add(hourlyMap);
+
                 }
                 if (hour1.equals(nextNextDay)) {
                     nextDayList.add(hourlyMap);
                 }
+
                 map.put("data", nextDayList);
             }
         }
 
-        for (Map<String, Object> resultMap : resultList) {
-            ArrayList<Map<String, Object>> arrayList = new ArrayList<>();
-            for (Map<String, Object> hashMap : hashSet) {
-                if (resultMap.get("name").equals(hashMap.get("name1"))) {
-                    ArrayList<Map<String, Object>> list = (ArrayList<Map<String, Object>>) hashMap.get("data");
-                    for (Map<String, Object> hourMap : list) {
-                        Map<String, Object> map = new HashMap<>();
-                        map.put("monitorPointId", resultMap.get("monitorPointId").toString());
-                        Map<String, Object> jsonMap = new HashMap<>();
-                        String time = hourMap.get("fxTime").toString().split("\\+")[0].replace("T", " ") + ":00";
-                        map.put("time", sdf1.parse(time));
-                        hourMap.remove("icon");
-                        jsonMap.putAll(hourMap);
-                        map.put("json", JSONObject.toJSONString(jsonMap));
-                        arrayList.add(map);
-                    }
+        for (Map<String, Object> map : hashSet) {
+            String cityCode = map.get("cityCode").toString();
+            List<Map<String, Object>> resultList = new ArrayList<>();
+            ArrayList<Map<String, Object>> dataList = (ArrayList<Map<String, Object>>) map.get("data");
+            for (Map<String, Object> dataMap : dataList) {
+                Map<String, Object> hashMap = new HashMap<>();
+                hashMap.put("cityCode", cityCode);
+                String time = dataMap.get("fxTime").toString().split("\\+")[0].replace("T", " ") + ":00";
+                hashMap.put("time", sdf1.parse(time));
 
+                String temp = dataMap.get("temp").toString();
+                int hour = Integer.valueOf(time.substring(11, 13));
+                if (map.get("constants") != null) {
+                    List<Map<String, Object>> list = (List<Map<String, Object>>) map.get("constants");
+                    for (Map<String, Object> listMap : list) {
+                        int value = Integer.valueOf(listMap.get("time").toString());
+                        Double a = Double.valueOf(listMap.get("a").toString());
+                        Double b = Double.valueOf(listMap.get("b").toString());
+                        if (hour == value) {
+                            double O3C = b * Double.valueOf(temp) + a;
+                            dataMap.put("O3C", String.valueOf(Math.round(O3C)));
+                        }
+                    }
                 }
+                hashMap.put("json", JSONObject.toJSONString(dataMap));
+                resultList.add(hashMap);
             }
-            arrayList.remove(0);
+            resultList.remove(0);
+            for (Map<String, Object> resultMap : resultList) {
+                Map<String, Object> jsonMap = (Map<String, Object>) JSONObject.parse(resultMap.get("json").toString());
+                jsonMap.remove("fxTime");
+                jsonMap.remove("icon");
+                resultMap.put("json", JSONObject.toJSONString(jsonMap));
+            }
             Map<String, Object> parameters = new HashMap<>();
             parameters.put("start", start);
             parameters.put("end", end);
-            parameters.put("monitorPointId", resultMap.get("monitorPointId").toString());
+            parameters.put("cityCode", cityCode);
             forecastWeatherMapper.deleteByTime(parameters);
-            for (Map<String, Object> map : arrayList) {
-                Map<String, Object> jsonMap = (Map<String, Object>) JSONObject.parse(map.get("json").toString());
-                jsonMap.remove("fxTime");
-                map.put("json", JSONObject.toJSONString(jsonMap));
-            }
-            forecastWeatherMapper.insertForecastWeather(arrayList);
+            count = count + forecastWeatherMapper.insertForecastWeather(resultList);
         }
+        return count;
     }
 }

--
Gitblit v1.8.0