From 33d898d8146223184fcaab82a9e16649434a1ab6 Mon Sep 17 00:00:00 2001
From: lizijie <lzjiiie@163.com>
Date: Fri, 09 Sep 2022 17:47:54 +0800
Subject: [PATCH] 千灯数据转发bug修改

---
 src/main/java/com/moral/task/AQIDataInsertTask.java |  145 +++++++++++++++++++++++++++++++----------------
 1 files changed, 95 insertions(+), 50 deletions(-)

diff --git a/src/main/java/com/moral/task/AQIDataInsertTask.java b/src/main/java/com/moral/task/AQIDataInsertTask.java
index 8ebb73d..8236f2d 100644
--- a/src/main/java/com/moral/task/AQIDataInsertTask.java
+++ b/src/main/java/com/moral/task/AQIDataInsertTask.java
@@ -1,83 +1,128 @@
 package com.moral.task;
+
+import java.text.SimpleDateFormat;
+import java.util.Calendar;
+import java.util.Collection;
+import java.util.Date;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import javax.annotation.Resource;
+
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
 import com.moral.service.AQIService;
 import com.moral.util.HttpUtils;
 import com.moral.util.WxMappingJackson2HttpMessageConverter;
+import com.xxl.job.core.biz.model.ReturnT;
+import com.xxl.job.core.handler.annotation.XxlJob;
 
 import org.apache.commons.lang3.time.DateUtils;
 import org.apache.http.HttpResponse;
 import org.apache.http.util.EntityUtils;
+import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Component;
 import org.springframework.util.ObjectUtils;
 import org.springframework.web.client.RestTemplate;
-import redis.clients.jedis.Jedis;
-
-import javax.annotation.Resource;
-import java.text.SimpleDateFormat;
-import java.util.*;
 
 @Component
 public class AQIDataInsertTask {
 
-
+    @Resource
+    private AQIService aqiService;
 
     @Resource
-        private AQIService aqiService;
+    private RedisTemplate redisTemplate;
 
-        public void insertData(){
-            SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
-            RestTemplate restTemplate = new RestTemplate();
-            restTemplate.getMessageConverters().add(new WxMappingJackson2HttpMessageConverter());
-            Date pubtime = DateUtils.truncate(new Date(),Calendar.HOUR);
+    @XxlJob("insertData")
+    public ReturnT insertData(String param) {
+        SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+        RestTemplate restTemplate = new RestTemplate();
+        restTemplate.getMessageConverters().add(new WxMappingJackson2HttpMessageConverter());
+        Date pubtime = DateUtils.truncate(new Date(), Calendar.HOUR);
+        List<Map<String, Object>> CityAqiConfigs = aqiService.getCityAqiConfig();
+        ReturnT returnT = null;
 
-            Jedis jedis = new Jedis("r-bp1672d21a422a14161.redis.rds.aliyuncs.com", 6379);
-            jedis.auth("KtElFcI1sYm9NP3");
-            jedis.select(1);
+        for (Map<String, Object> cityAqiConfig : CityAqiConfigs) {
+            String entity = null;
+            Collection<Object> values = null;
+            Map<String, Object> data = null;
+         /*   try {
+                HttpResponse response = HttpUtils.doGet("https://api.epmap.org", "/api/v1/air/city", "GET",
+                        new HashMap<String, String>() {{put("Authorization", "APPCODE " + "31b6ea8f804a4472be3b633cfee44849");}},
+                        new HashMap<String, String>() {{put("city", cityAqiConfig.get("city_name").toString());}}
+                );
+                entity = EntityUtils.toString(response.getEntity());
+                JSONObject json = JSON.parseObject(entity);
 
-            List<Map<String,Object>> CityAqiConfigs =aqiService.getCityAqiConfig();
-            for (Map<String, Object> cityAqiConfig : CityAqiConfigs) {
-                String entity = null ;
-                Collection<Object> values = null;
-                Map<String, Object> data = null;
+                data = (Map<String, Object>) json.get("data");
+                if (!ObjectUtils.isEmpty(data)) {
+                    values = data.values();
+                    pubtime = format.parse(data.get("pubtime").toString());
+                    int hours = pubtime.getHours();
+                    Calendar calendar = Calendar.getInstance();
+                    int hour = calendar.get(Calendar.HOUR_OF_DAY);
+                    if (hour != hours){
+                        pubtime = null;
+                    }
+                }
+            } catch (Exception e) {
+                e.printStackTrace();
+            }*/
+            if (ObjectUtils.isEmpty(entity) || ObjectUtils.isEmpty(values) || values.contains("None") || pubtime == null) {
+                Map<String, Object> mjMap = restTemplate.getForObject("http://sapi.7drlb.com/api/mj?cityID={1}&apiKey=aqi", Map.class, cityAqiConfig.get("city_id"));
+                if (ObjectUtils.isEmpty(mjMap)) {
+                    continue;
+                } else {
+                    data = (Map<String, Object>) ((Map) mjMap.get("data")).get("aqi");
+                    data.put("aqi", data.remove("value"));
+                    data.put("PM2_5", data.remove("pm25"));
+                    pubtime = new Date(Long.valueOf(data.get("pubtime").toString()));
+                    String string = JSONObject.toJSONString(data).toUpperCase();
+                    data = (Map<String, Object>) JSON.parse(string);
+                }
+            }
+            Map<String, Object> parameters = new HashMap<>();
+            parameters.put("time", pubtime);
+            parameters.put("data", data.toString());
+            parameters.put("code", cityAqiConfig.get("city_code"));
+            if (!ObjectUtils.isEmpty(data)) {
                 try {
-                    HttpResponse response = HttpUtils.doGet("https://api.epmap.org", "/api/v1/air/city", "GET",
-                            new HashMap<String, String>() {{put("Authorization", "APPCODE " + "31b6ea8f804a4472be3b633cfee44849");}},
-                            new HashMap<String, String>() {{put("city", cityAqiConfig.get("city_name").toString());}}
-                    );
-                    entity = EntityUtils.toString(response.getEntity());
-                    JSONObject json = JSON.parseObject(entity);
-                    data = (Map<String, Object>) json.get("data");
-                    if (!ObjectUtils.isEmpty(data)) {
-                        values = data.values();
-                        pubtime = format.parse(data.get("pubtime").toString());
+                    int i = aqiService.insertAQIData(parameters);
+                    if (i > 0) {
+                        returnT = new ReturnT(200, " ���������������������");
+                    } else {
+                        returnT = new ReturnT(500, " ���������������������");
                     }
                 } catch (Exception e) {
                     e.printStackTrace();
                 }
-
-                Map<String, Object> parameters = new HashMap<>();
-                parameters.put("time",pubtime);
-                parameters.put("data",data.toString());
-                parameters.put("code",cityAqiConfig.get("city_code"));
-                if (!ObjectUtils.isEmpty(data)) {
-                    try {
-                        aqiService.insertAQIData(parameters);
-                    } catch (Exception e) {
-                        e.printStackTrace();
+                if ("1".equals(cityAqiConfig.get("is_compensate"))) {
+                    Map<String, String> map = new HashMap<String, String>();
+                    map.put("e1", data.containsKey("PM25C") ? data.get("PM25C").toString() : data.get("PM2_5").toString());
+                    map.put("e2", data.containsKey("PM10C") ? data.get("PM10C").toString() : data.get("PM10").toString());
+                    map.put("e10", data.containsKey("COC") ? data.get("COC").toString() : data.get("CO").toString());
+                    map.put("e11", data.containsKey("SO2C") ? data.get("SO2C").toString() : data.get("SO2").toString());
+                    map.put("e15", data.containsKey("O3C") ? data.get("O3C").toString() : data.get("O3").toString());
+                    map.put("e16", data.containsKey("NO2C") ? data.get("NO2C").toString() : data.get("NO2").toString());
+                    if (data.get("CITYNAME").equals("���������")){
+                        map.put("e1", data.get("PM2_5").toString());
                     }
-                    if ("1".equals(cityAqiConfig.get("is_compensate"))) {
-                        Map<String, String> map = new HashMap<String, String>();
-                        map.put("e1", data.containsKey("PM25C") ? data.get("PM25C").toString() : data.get("PM2_5").toString());
-                        map.put("e2", data.containsKey("PM10C") ? data.get("PM10C").toString() : data.get("PM10").toString());
-                        map.put("e10", data.containsKey("COC") ? data.get("COC").toString() : data.get("CO").toString());
-                        map.put("e11", data.containsKey("SO2C") ? data.get("SO2C").toString() : data.get("SO2").toString());
-                        map.put("e15", data.containsKey("O3C") ? data.get("O3C").toString() : data.get("O3").toString());
-                        map.put("e16", data.containsKey("NO2C") ? data.get("NO2C").toString() : data.get("NO2").toString());
-                        jedis.hmset("aqi_" + cityAqiConfig.get("city_code"), map);
-                    }
+                    redisTemplate.opsForHash().putAll("aqi_" + cityAqiConfig.get("city_code"), map);
                 }
             }
         }
+        return returnT;
+    }
 
+    @XxlJob("insertStationData")
+    public ReturnT insertStationData(String param) {
+        int state = aqiService.insertStationData();
+        if (state == 200) {
+            return new ReturnT(200, "������������");
+        } else {
+            return new ReturnT(500, "������������������������������������������������");
+        }
+    }
 }

--
Gitblit v1.8.0