jinpengyong
2020-08-19 aafe57b555359e4a5ff347ce5e495d9d40d7dd42
src/main/java/com/moral/service/impl/ForecastWeatherServiceImpl.java
@@ -17,17 +17,18 @@
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.entity.Province;
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 +50,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 +71,47 @@
        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;
            Integer parentCode = 0;
            if (areaCode != null) {
                Area area = areaMapper.getAreaByAreaCode(areaCode);
                City city = cityMapper.getCityByCityCode(cityCode);
                name1 = area.getAreaName();
                parentName = city.getCityName();
                parentCode = city.getCityCode();
                code = areaCode;
                if ("市辖区".equals(name1)) {
                    name1 = parentName;
                    parentName = provinceMapper.getProvinceByProvinceCode(provinceCode).getProvinceName();
                    Province province = provinceMapper.getProvinceByProvinceCode(provinceCode);
                    parentName = province.getProvinceName();
                    parentCode = province.getProvinceCode();
                    code = cityCode;
                }
            } else {
                City city = cityMapper.getCityByCityCode(cityCode);
                name1 = city.getCityName();
                parentName = provinceMapper.getProvinceByProvinceCode(provinceCode).getProvinceName();
                Province province = provinceMapper.getProvinceByProvinceCode(provinceCode);
                parentName = province.getProvinceName();
                code = cityCode;
                parentCode = province.getProvinceCode();
                if ("市辖区".equals(name1)) {
                    name1 = parentName;
                    parentName = "";
                    code = provinceCode;
                }
            }
            hashMap.put("name1", name1);
            hashMap.put("parentName", parentName);
            hashMap.put("cityCode", code);
            hashMap.put("parentCode", parentCode);
            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 +138,104 @@
            }
        }
        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) {
            Map<String, Object> map1 = new HashMap<>();
            map1.put("typeFormat", "%Y-%m-%d %H:%i:%s");
            map1.put("startTime", startTime);
            map1.put("endTime", endTime);
            map1.put("cityCode", Integer.valueOf(hashmap.get("cityCode").toString()));
            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) {
                    map1.put("time", "0" + i + ":00:00");
                } else {
                    map1.put("time", i + ":00:00");
                }
                List<Map<String, Object>> tempList = realWeatherMapper.getTemp(map1);
                List<Map<String, Object>> O3List = aqiMapper.getO3(map1);
                if (O3List.size() == 0) {
                    map1.put("cityCode", Integer.valueOf(hashmap.get("parentCode").toString()));
                    O3List = aqiMapper.getO3(map1);
                }
                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).size()!=0) {
                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,6 +247,7 @@
                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);
@@ -147,37 +256,47 @@
            }
        }
        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;
    }
}