update

jinpengyong

2020-08-20 f73c5a24b25280030deaaafe38b35b1e7439b9c3

 src/main/java/com/moral/service/impl/ForecastWeatherServiceImpl.java

@@ -7,7 +7,6 @@
import java.util.Date;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
@@ -18,16 +17,18 @@
import org.springframework.stereotype.Service;
import org.springframework.web.client.RestTemplate;

import com.alibaba.fastjson.JSON;
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,136 +50,257 @@
    @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");
        String now = sdf.format(c.getTime());
        c.add(Calendar.DAY_OF_MONTH, 1);
        String nextDay = sdf.format(c.getTime());
        c.setTime(c.getTime());
        Date start = sdf1.parse(nextDay + " 01:00:00");
        c.add(Calendar.DAY_OF_MONTH, 1);
        String nextNextDay = sdf.format(c.getTime());

        String nextNextDay = sdf.format(c.getTime()) + " 00";
        Date end = sdf1.parse(sdf.format(c.getTime()) + " 01:00:00");
        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";
        List<Map<String, Object>> insertList = new ArrayList<>();
        for (Map<String, Object> map : hashSet) {
            Map<String, Object> hashMap = new HashMap<>();
            String name1 = map.get("name1").toString();
            hashMap.put("name", name1);
            for (Element element : elements) {
                String name2 = element.element("name").getText();
                String parentName = map.get("parentName").toString();
                if (name2.equals("大丰市")) {
                    name2 = "大丰区";
                }
                if (name1.equals(name2)) {
                    cityID = element.element("Fid").getText();
                    cityID = element.element("Fweathercn").getText();
                    map.put("cityID", cityID);
                    break;
                }
                if (name2.endsWith(name1)) {
                    if (name2.startsWith(parentName)) {
                        cityID = element.element("Fid").getText();
                        cityID = element.element("Fweathercn").getText();
                        map.put("cityID", cityID);
                        break;
                    }
                }
            }
            Map<String, Object> map1 = restTemplate.getForObject("https://yiketianqi.com/api?version=v9&appid=68884517&appsecret=kqUYTBk3&cityid={1}", Map.class, cityID);
            if (map1 == null) {
                map1=new HashMap<>();
            }

            List<Map<String, Object>> dataList = (ArrayList<Map<String, Object>>) map1.get("data");
            List<Map<String, Object>> arrayList = new ArrayList<>();

            for (Map<String, Object> dataMap : dataList) {
                String date = dataMap.get("date").toString();
                ArrayList<Map<String, Object>> list = (ArrayList<Map<String, Object>>) dataMap.get("hours");
                if (date.equals(now)) {
                    for (int i = list.size() - 7; i < list.size(); i++) {
                        Map<String, Object> resultMap = list.get(i);
                        arrayList.add(resultMap);
                    }
                }
                if (date.equals(nextDay)) {
                    for (int i = 0; i < 17; i++) {
                        Map<String, Object> resultMap = list.get(i);
                        arrayList.add(resultMap);
                    }
                }
            }
            hashMap.put("data", arrayList);
            insertList.add(hashMap);
        }

        for (Map<String, Object> resultMap : resultList) {
            List<Map<String, Object>> list = new ArrayList<>();
            String monitorPointId = resultMap.get("monitorPointId").toString();
            for (Map<String, Object> insertMap : insertList) {
                if (resultMap.get("name").equals(insertMap.get("name"))) {
                    ArrayList<Map<String, Object>> dataList = (ArrayList<Map<String, Object>>) insertMap.get("data");
                    for (Map<String, Object> dataMap : dataList) {
                        dataMap.remove("wea_img");
                        Map<String, Object> map = new HashMap<>();
                        map.put("monitorPointId", Integer.valueOf(monitorPointId));
                        String hour = dataMap.get("hours").toString().replaceAll("[\u4e00-\u9fa5]+", "");
                        Date date = sdf1.parse(nextDay + " " + hour + ":00:00");
                        if (hour.equals("00")) {
                            date = sdf1.parse(nextNextDay + " 00:00:00");
        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);
            hashmap.put("typeFormat", "%Y-%m-%d %H:%i:%s");
            hashmap.put("startTime", startTime);
            hashmap.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");
                    hashmap.put("time", "0" + i + ":00:00");
                } else {
                    map1.put("time", i + ":00:00");
                    hashmap.put("time", i + ":00:00");
                }
                List<Map<String, Object>> tempList = realWeatherMapper.getTemp(hashmap);
                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);
                                    }
                                }
                            }
                        }
                        map.put("time", date);
                        String json = JSONObject.toJSONString(dataMap);
                        map.put("json", json);
                        list.add(map);
                    }
                    list.add(list3);
                }
            }

            for (Map<String, Object> map : list) {
                String json = map.get("json").toString();
                Map<String, Object> jsonMap = JSON.parseObject(json, Map.class);
                Iterator<String> iterator = jsonMap.keySet().iterator();
                while (iterator.hasNext()) {
                    if (iterator.next().equals("hours")) {
                        iterator.remove();
                        json = JSONObject.toJSONString(jsonMap);
                        map.put("json",json);
            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);
                }
            }
            forecastWeatherMapper.insertForecastWeather(list);
        }

        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) {
                return count;
            }
            String json = JSONObject.toJSONString(dataMap);
            dataMap = (Map<String, Object>) JSONObject.parse(json);
            List<Map<String, Object>> nextDayList = new ArrayList<>();
            String string = dataMap.get("hourly").toString();
            List<Map<String, Object>> hourlyList = (List<Map<String, Object>>) JSONObject.parse(string);
            for (Map<String, Object> hourlyMap : hourlyList) {
                String hour = hourlyMap.get("fxTime").toString().split("T")[0];
                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> 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);
            }
            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("cityCode", cityCode);
            forecastWeatherMapper.deleteByTime(parameters);
            count = count + forecastWeatherMapper.insertForecastWeather(resultList);
        }
        return count;
    }
}