screen-api

kaiyu

2022-03-24 a827480e5e74cb6550c5f7b842bccac3023c5bdc

 screen-job/src/main/java/com/moral/api/service/impl/CityAqiMonthlyServiceImpl.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.CityAqiMonthly;
import com.moral.api.mapper.CityAqiMonthlyMapper;
@@ -57,6 +58,20 @@
        //结束时间，本月1号
        Date end = DateUtils.addMonths(start, 1);

        //如果是1号，先删除上月数据再统计
        //不是1号，统计的是本月累计值，先删除本月数据
        int day = DateUtils.getDay(new Date());
        UpdateWrapper<CityAqiMonthly> cityAqiMonthlyUpdateWrapper = new UpdateWrapper<>();
        if (day == 1) {
            cityAqiMonthlyUpdateWrapper.eq("time", start);
        } else {
            start = end;
            end = DateUtils.addMonths(start, 1);
            lastLastMonth = DateUtils.addMonths(start, -1);
            cityAqiMonthlyUpdateWrapper.eq("time", start);
        }
        cityAqiMonthlyMapper.delete(cityAqiMonthlyUpdateWrapper);


        //获取所有城市aqi小时数据
        QueryWrapper<CityAqiDaily> wrapper = new QueryWrapper<>();
@@ -82,11 +97,24 @@
                .collect(Collectors.groupingBy(CityAqiMonthly::getCityCode));


        CityAqiMonthly cityAqiMonthly = new CityAqiMonthly();
        //获取去年本月数据
        Date thisMonthOfLastYear = DateUtils.addYears(start, -1);
        queryWrapper.clear();
        queryWrapper.select("city_code", "value")
                .eq("time", DateUtils.dateToDateString(thisMonthOfLastYear));
        List<CityAqiMonthly> thisMonthOfLastYearList = cityAqiMonthlyMapper.selectList(queryWrapper);
        Map<Integer, List<CityAqiMonthly>> thisMonthOfLastYearData = thisMonthOfLastYearList.stream()
                .collect(Collectors.groupingBy(CityAqiMonthly::getCityCode));


        List<CityAqiMonthly> cityAqiMonthlyList = new ArrayList<>();

        Date finalStart = start;
        data.forEach((cityCode, value) -> {
            CityAqiMonthly cityAqiMonthly = new CityAqiMonthly();
            Map<String, Object> jsonMap = new HashMap<>();
            cityAqiMonthly.setCityCode(Integer.parseInt(cityCode));
            cityAqiMonthly.setTime(start);
            cityAqiMonthly.setTime(finalStart);

            Map<String, Object> params = new HashMap<>();
            List<Map<String, Object>> temp = new ArrayList<>();
@@ -128,7 +156,7 @@
            Double compositeIndex = ComprehensiveIndexUtils.dailyData(jsonMap);
            jsonMap.put("compositeIndex", compositeIndex);

            //本月综指同上月对比
            //本月综指同上月对比(综合指数环比)
            List<CityAqiMonthly> cityAqiMonthlies = lastMonthData.get(Integer.parseInt(cityCode));
            if (!ObjectUtils.isEmpty(cityAqiMonthlies)) {
                CityAqiMonthly monthly = cityAqiMonthlies.get(0);
@@ -138,8 +166,45 @@
                String format = decimalFormat.format((compositeIndex - lastCompositeIndex) / lastCompositeIndex);
                jsonMap.put("monthContrast", format);
            }

            //获取去年本月数据，用于计算同比
            List<CityAqiMonthly> thisMonthOfLastYears = thisMonthOfLastYearData.get(Integer.parseInt(cityCode));

            //各因子同比计算
            Map<String, Object> yearOnYearValue = yearOnYearOfSensor(thisMonthOfLastYears, jsonMap);
            if (yearOnYearValue != null) {
                jsonMap.putAll(yearOnYearValue);
            }

            cityAqiMonthly.setValue(JSONObject.toJSONString(jsonMap));
            cityAqiMonthlyMapper.insert(cityAqiMonthly);
            cityAqiMonthlyList.add(cityAqiMonthly);
        });
        cityAqiMonthlyMapper.insertCityAqiMonthly(cityAqiMonthlyList);
    }


    /**
     * @param thisMonthOfLastYearData 去年本月数据
     * @param currentData             当前月该数据
     */
    private Map<String, Object> yearOnYearOfSensor(List<CityAqiMonthly> thisMonthOfLastYearData, Map<String, Object> currentData) {
        Map<String, Object> result = null;
        //需要计算同比的因子
        List<String> sensors = Arrays.asList("PM2_5", "PM10", "SO2", "NO2", "CO", "O3", "compositeIndex");
        if (!ObjectUtils.isEmpty(thisMonthOfLastYearData)) {
            result = new HashMap<>();
            CityAqiMonthly monthly = thisMonthOfLastYearData.get(0);
            Map<String, Object> map = JSONObject.parseObject(monthly.getValue(), Map.class);
            for (String sensor : sensors) {
                //去年本月该因子值
                double thisMonthOfLeastYearValue = Double.parseDouble(map.get(sensor).toString());
                //当前该因子值
                double currentValue = Double.parseDouble(currentData.get(sensor).toString());
                DecimalFormat decimalFormat = new DecimalFormat("0.00%");
                String format = decimalFormat.format((currentValue - thisMonthOfLeastYearValue) / thisMonthOfLeastYearValue);
                result.put(sensor + "_yearOnYear", format);
            }
        }
        return result;
    }
}