forked from moral

blame | history | patch | commit | commitdiff | ignore whitespace
jinpengyong
2022-01-14 f6ecfd331ec959d7daea3fa5ebc262e7a50e4e01 
 screen-api/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java


@@ -3,6 +3,7 @@


import com.alibaba.fastjson.JSON;


import com.alibaba.fastjson.JSONObject;


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;


import com.moral.api.config.properties.SpecialCitiesProperties;


import com.moral.api.entity.CityAqi;


import com.moral.api.entity.CityAqiDaily;


import com.moral.api.entity.CityAqiMonthly;


@@ -60,9 +61,6 @@


    private RedisTemplate redisTemplate;




    @Autowired


    private OrganizationService organizationService;




    @Autowired


    private SysAreaService sysAreaService;




    @Autowired


@@ -73,6 +71,9 @@




    @Autowired


    private CityAqiYearlyService cityAqiYearlyService;




    @Autowired


    private SpecialCitiesProperties specialCitiesProperties;




    @Override


    public List<Map<String, Object>> measuredCompareForecastOfO3(Map<String, Object> params) {


@@ -135,6 +136,21 @@


        Map<String, Object> value = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.CITY_AQI, String.valueOf(regionCode));


        if (value == null)


            value = queryCityAqiByRegionCodeFromDB(regionCode);


        //如果地区码是县级市则转换到市级在进行查询


        if (value == null) {


            String regionCodeStr = String.valueOf(regionCode);


            String end = regionCodeStr.substring(regionCodeStr.length() - 2, regionCodeStr.length());


            if (!end.equals(00)) {


                regionCodeStr = regionCodeStr.substring(0, regionCodeStr.length() - 2);


                regionCodeStr += "00";


                regionCode = Integer.parseInt(regionCodeStr);


                value = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.CITY_AQI, String.valueOf(regionCode));


                if (value == null)


                    value = queryCityAqiByRegionCodeFromDB(regionCode);


            } else {


                return null;


            }


        }


        //根据AQI计算污染等级


        if (value == null || value.get("AQI") == null)


            return null;


@@ -152,6 +168,8 @@


        lastDataWrapper.orderByDesc("time");


        lastDataWrapper.last(true, "limit 1");


        CityAqi cityAqi = cityAqiMapper.selectOne(lastDataWrapper);


        if (cityAqi == null)


            return null;


        //算出前24小时的时间点


        Date endDate = cityAqi.getTime();


        Date startDate = DateUtils.addHours(endDate, -23);


@@ -176,10 +194,10 @@


            //按照时间进行排序


            cityAqis.sort(Comparator.comparing(CityAqi::getTime));


        }


        //封装返回数据,map的key为HH:mm格式的时间，value为aqi的数值


        //封装返回数据,map的key为yyyy-MM-dd HH:mm格式的时间，value为aqi的数值


        Map<String, Object> result = new LinkedHashMap<>();


        for (CityAqi aqi : cityAqis) {


            String key = DateUtils.dateToDateString(aqi.getTime(), "HH:mm");


            String key = DateUtils.dateToDateString(aqi.getTime(), "yyyy-MM-dd HH:mm");


            String allDataJson = aqi.getValue();


            if (allDataJson == null) {


                result.put(key, "");


@@ -223,7 +241,7 @@


    @Override


    public List<Map<String, Object>> rankingDetails(Map<String, Object> params) {


        List<Map<String, Object>> result = new ArrayList<>();


        int orgId = Integer.parseInt(params.get("organizationId").toString());


        int regionCode = Integer.parseInt(params.get("regionCode").toString());


        String type = params.get("type").toString();


        String time = null;


        if (!ObjectUtils.isEmpty(params.get("time"))) {


@@ -237,45 +255,52 @@


        }


        String cityType = params.get("cityType").toString();




        //获取省,市code


        Organization organization = organizationService.getById(orgId);


        Integer curProvinceCode = organization.getProvinceCode();


        Integer curCityCode = organization.getCityCode();


        String s = String.valueOf(regionCode);


        //获取当前省,市code


        Integer curProvinceCode = Integer.parseInt(s.substring(0, 2) + "0000");


        Integer curCityCode = Integer.parseInt(s.substring(0, 4) + "00");




        QueryWrapper<SysArea> areaWrapper = new QueryWrapper<>();


        if ("province".equals(cityType)) {


            //获取省内所有市


            areaWrapper.select("area_code").eq("parent_code", curProvinceCode);




        List<SysArea> sysAreas;


        if ("28".equals(cityType)) {


            //获取2+26城市


            sysAreas = specialCitiesProperties.getTwentyEightCities();


        } else {


            //获取市内所有县区


            areaWrapper.select("area_code").eq("parent_code", curCityCode);


            if ("province".equals(cityType)) {


                //获取省内所有市


                areaWrapper.select("area_code", "area_name").eq("parent_code", curProvinceCode);


            } else {


                //获取市内所有县区


                areaWrapper.select("area_code", "area_name").eq("parent_code", curCityCode);


            }


            sysAreas = sysAreaService.list(areaWrapper);


        }


        List<Object> regionCodes = sysAreaService.listObjs(areaWrapper);




        switch (type) {


            case "today":


                result = accumulatedTodayRank(regionCodes);


                result = accumulatedTodayRank(sysAreas);


                break;


            case "hour":


                time = new StringBuilder(time).replace(10, 11, " ").toString() + ":00:00";


                result = hourRank(regionCodes, time);


                result = hourRank(sysAreas, time);


                break;


            case "day":


                time = time + " 00:00:00";


                result = dayRank(regionCodes, time);


                result = dayRank(sysAreas, time);


                break;


            case "month":


                time = time + "-01 00:00:00";


                result = monthRank(regionCodes, time);


                result = monthRank(sysAreas, time);


                break;


            case "year":


                time = time + "-01-01 00:00:00";


                result = yearRank(regionCodes, time);


                result = yearRank(sysAreas, time);


                break;


            case "custom":


                start = start + " 00:00:00";


                end = end + " :00:00";


                result = customRank(regionCodes, start, end);


                result = customRank(sysAreas, start, end);


                break;


            default:


                break;


@@ -284,10 +309,14 @@


    }




    /**


     * @param regionCodes 所要获取城市的地去编程集合


     * @param sysAreas 所要获取城市集合


     * @return 功能：今日累计排名


     */


    private List<Map<String, Object>> accumulatedTodayRank(List<Object> regionCodes) {


    private List<Map<String, Object>> accumulatedTodayRank(List<SysArea> sysAreas) {


        List<Integer> regionCodes = sysAreas.stream()


                .map(SysArea::getAreaCode)


                .collect(Collectors.toList());




        List<Map<String, Object>> result = new ArrayList<>();


        List<String> sensors = Arrays.asList("PM2_5", "PM10", "SO2", "NO2", "CO", "O3");


        Date now = new Date();


@@ -298,7 +327,7 @@


                .in("city_code", regionCodes);


        List<Map<String, Object>> cumulativeData = cityAqiMapper.selectMaps(wrapper);


        //按city_code分组


        Map<String, List<Map<String, Object>>> data = cumulativeData.parallelStream().collect(Collectors.groupingBy(o -> o.get("city_code").toString()));


        Map<Integer, List<Map<String, Object>>> data = cumulativeData.parallelStream().collect(Collectors.groupingBy(o -> Integer.parseInt(o.get("city_code").toString())));


        data.forEach((cityCode, value) -> {


            List<Double> doubles = new ArrayList<>();


            for (Map<String, Object> objectMap : value) {


@@ -349,7 +378,7 @@


            List<String> primaryPollutantNames = aqi.getPrimaryPollutantNames();


            String primaryPollutant = "";


            if (!ObjectUtils.isEmpty(primaryPollutantNames)) {


                primaryPollutant = primaryPollutantNames.toString().replace("[", "").replace("]", "");


                primaryPollutant = primaryPollutantNames.stream().map(String::valueOf).collect(Collectors.joining(","));


            }


            dataMap.put("primaryPollutant", primaryPollutant);




@@ -360,22 +389,27 @@


            dataMap.put("compositeIndex", compositeIndex);




            //城市名


            QueryWrapper<SysArea> queryWrapper = new QueryWrapper<>();


            queryWrapper.select("area_name").eq("area_code", cityCode);


            String areaName = sysAreaService.getOne(queryWrapper).getAreaName();


            dataMap.put("cityName", areaName);




            for (SysArea sysArea : sysAreas) {


                if (cityCode.equals(sysArea.getAreaCode())) {


                    dataMap.put("cityName", sysArea.getAreaName());


                    break;


                }


            }


            result.add(dataMap);


        });


        return result;


    }




    /**


     * @param regionCodes 所要获取城市的编码集合


     * @param time        所要获取数据的时间 2021-11-04 13:00:00


     * @param sysAreas 所要获取城市集合


     * @param time     所要获取数据的时间 2021-11-04 13:00:00


     * @return 功能：小时排名


     */


    private List<Map<String, Object>> hourRank(List<Object> regionCodes, String time) {


    private List<Map<String, Object>> hourRank(List<SysArea> sysAreas, String time) {


        List<Integer> regionCodes = sysAreas.stream()


                .map(SysArea::getAreaCode)


                .collect(Collectors.toList());




        List<Map<String, Object>> result = new ArrayList<>();


        QueryWrapper<CityAqi> wrapper = new QueryWrapper<>();


        wrapper.select("value")


@@ -387,7 +421,7 @@


            List<String> primaryPollutantNames = (List<String>) value.get("primaryPollutant");


            String primaryPollutant = "";


            if (!ObjectUtils.isEmpty(primaryPollutantNames)) {


                primaryPollutant = primaryPollutantNames.toString().replace("[", "").replace("]", "");


                primaryPollutant = primaryPollutantNames.stream().map(String::valueOf).collect(Collectors.joining(","));


            }


            value.put("primaryPollutant", primaryPollutant);


            value.remove("pubtime");


@@ -398,11 +432,15 @@


    }




    /**


     * @param regionCodes 所要获取城市的编码集合


     * @param time        所要获取数据的时间 2021-11-04 00:00:00


     * @param sysAreas 所要获取城市集合


     * @param time     所要获取数据的时间 2021-11-04 00:00:00


     * @return 功能：日排名


     */


    private List<Map<String, Object>> dayRank(List<Object> regionCodes, String time) {


    private List<Map<String, Object>> dayRank(List<SysArea> sysAreas, String time) {


        List<Integer> regionCodes = sysAreas.stream()


                .map(SysArea::getAreaCode)


                .collect(Collectors.toList());




        List<Map<String, Object>> result = new ArrayList<>();


        QueryWrapper<CityAqiDaily> wrapper = new QueryWrapper<>();


        wrapper.select("city_code", "value")


@@ -414,33 +452,40 @@


            List<String> primaryPollutantNames = (List<String>) value.get("primaryPollutant");


            String primaryPollutant = "";


            if (!ObjectUtils.isEmpty(primaryPollutantNames)) {


                primaryPollutant = primaryPollutantNames.toString().replace("[", "").replace("]", "");


                primaryPollutant = primaryPollutantNames.stream().map(String::valueOf).collect(Collectors.joining(","));


            }


            value.put("primaryPollutant", primaryPollutant);






            //城市名


            QueryWrapper<SysArea> queryWrapper = new QueryWrapper<>();


            queryWrapper.select("area_name")


                    .eq("area_code", dayDatum.get("city_code"));


            String areaName = sysAreaService.getOne(queryWrapper).getAreaName();


            value.put("cityName", areaName);


            for (SysArea sysArea : sysAreas) {


                if (dayDatum.get("city_code").equals(sysArea.getAreaCode())) {


                    value.put("cityName", sysArea.getAreaName());


                    break;


                }


            }


            result.add(value);


        }


        return result;


    }




    /**


     * @param regionCodes 所要获取城市的编码集合


     * @param time        所要获取数据的时间 2021-11-01 00:00:00 每月1号


     * @param sysAreas 所要获取城市集合


     * @param time     所要获取数据的时间 2021-11-01 00:00:00 每月1号


     * @return 功能：月排名


     */


    private List<Map<String, Object>> monthRank(List<Object> regionCodes, String time) {


    private List<Map<String, Object>> monthRank(List<SysArea> sysAreas, String time) {


        List<Integer> regionCodes = sysAreas.stream()


                .map(SysArea::getAreaCode)


                .collect(Collectors.toList());




        //需要均值计算的因子


        List<String> sensors = Arrays.asList("PM2_5", "PM10", "SO2", "NO2");


        List<Map<String, Object>> result = new ArrayList<>();


        //如果是本月,实时计算，其他直接从city_aqi_monthly获取


        //如果部是本月,实时计算，其他直接从city_aqi_monthly获取


        if (!time.substring(0, 7).equals(DateUtils.dateToDateString(new Date(), DateUtils.yyyy_MM_EN))) {


            QueryWrapper<CityAqiMonthly> cityAqiMonthlyQueryWrapper = new QueryWrapper<>();


            for (Object regionCode : regionCodes) {


            for (Integer regionCode : regionCodes) {


                cityAqiMonthlyQueryWrapper.clear();


                cityAqiMonthlyQueryWrapper.select("value")


                        .eq("city_code", regionCode)


@@ -451,13 +496,15 @@


                }


                String value = cityAqiMonthly.getValue();


                Map<String, Object> resultMap = JSONObject.parseObject(value, Map.class);






                //城市名


                QueryWrapper<SysArea> sysAreaQueryWrapper = new QueryWrapper<>();


                sysAreaQueryWrapper.select("area_name")


                        .eq("area_code", regionCode);


                String areaName = sysAreaService.getOne(sysAreaQueryWrapper).getAreaName();


                resultMap.put("cityName", areaName);


                result.add(resultMap);


                for (SysArea sysArea : sysAreas) {


                    if (regionCode.equals(sysArea.getAreaCode())) {


                        resultMap.put("cityName", sysArea.getAreaName());


                        break;


                    }


                }


            }


            return result;


        }


@@ -468,7 +515,9 @@


                .in("city_code", regionCodes);


        List<Map<String, Object>> thisMonthData = cityAqiDailyService.listMaps(cityAqiDailyQueryWrapper);


        //按city_code分组


        Map<String, List<Map<String, Object>>> thisMonthMap = thisMonthData.parallelStream().collect(Collectors.groupingBy(o -> o.get("city_code").toString()));


        Map<Integer, List<Map<String, Object>>> thisMonthMap = thisMonthData.parallelStream()


                .collect(Collectors.groupingBy(o -> Integer.parseInt(o.get("city_code").toString())));




        thisMonthMap.forEach((cityCode, value) -> {


            Map<String, Object> resultMap = new HashMap<>();




@@ -539,30 +588,34 @@


            resultMap.put("monthContrast", monthContrast);




            //城市名


            QueryWrapper<SysArea> sysAreaQueryWrapper = new QueryWrapper<>();


            sysAreaQueryWrapper.select("area_name")


                    .eq("area_code", cityCode);


            String areaName = sysAreaService.getOne(sysAreaQueryWrapper).getAreaName();


            resultMap.put("cityName", areaName);




            for (SysArea sysArea : sysAreas) {


                if (cityCode.equals(sysArea.getAreaCode())) {


                    resultMap.put("cityName", sysArea.getAreaName());


                    break;


                }


            }


            result.add(resultMap);


        });


        return result;


    }




    /**


     * @param regionCodes 所要获取城市的编码集合


     * @param time        所要获取数据的时间 2021-11-01 00:00:00 每年1月1号


     * @param sysAreas 所要获取城市集合


     * @param time     所要获取数据的时间 2021-11-01 00:00:00 每年1月1号


     * @return 功能：年排名


     */


    private List<Map<String, Object>> yearRank(List<Object> regionCodes, String time) {


    private List<Map<String, Object>> yearRank(List<SysArea> sysAreas, String time) {


        List<Integer> regionCodes = sysAreas.stream()


                .map(SysArea::getAreaCode)


                .collect(Collectors.toList());




        //需要均值计算的因子


        List<String> sensors = Arrays.asList("PM2_5", "PM10", "SO2", "NO2");


        List<Map<String, Object>> result = new ArrayList<>();


        //如果是本月,实时计算，其他直接从city_aqi_monthly获取


        if (!time.substring(0, 4).equals(DateUtils.dateToDateString(new Date(), DateUtils.yyyy))) {


            QueryWrapper<CityAqiYearly> cityAqiYearlyQueryWrapper = new QueryWrapper<>();


            for (Object regionCode : regionCodes) {


            for (Integer regionCode : regionCodes) {


                cityAqiYearlyQueryWrapper.clear();


                cityAqiYearlyQueryWrapper.select("value")


                        .eq("city_code", regionCode)


@@ -573,12 +626,15 @@


                }


                String value = cityAqiYearly.getValue();


                Map<String, Object> resultMap = JSONObject.parseObject(value, Map.class);






                //城市名


                QueryWrapper<SysArea> sysAreaQueryWrapper = new QueryWrapper<>();


                sysAreaQueryWrapper.select("area_name")


                        .eq("area_code", regionCode);


                String areaName = sysAreaService.getOne(sysAreaQueryWrapper).getAreaName();


                resultMap.put("cityName", areaName);


                for (SysArea sysArea : sysAreas) {


                    if (regionCode.equals(sysArea.getAreaCode())) {


                        resultMap.put("cityName", sysArea.getAreaName());


                        break;


                    }


                }


                result.add(resultMap);


            }


            return result;


@@ -590,7 +646,8 @@


                .in("city_code", regionCodes);


        List<Map<String, Object>> thisMonthData = cityAqiDailyService.listMaps(cityAqiDailyQueryWrapper);


        //按city_code分组


        Map<String, List<Map<String, Object>>> thisYearMap = thisMonthData.parallelStream().collect(Collectors.groupingBy(o -> o.get("city_code").toString()));


        Map<Integer, List<Map<String, Object>>> thisYearMap = thisMonthData.parallelStream()


                .collect(Collectors.groupingBy(o -> Integer.parseInt(o.get("city_code").toString())));


        thisYearMap.forEach((cityCode, value) -> {


            Map<String, Object> resultMap = new HashMap<>();




@@ -659,24 +716,31 @@


                yearContrast = decimalFormat.format((compositeIndex - lastCompositeIndex) / lastCompositeIndex);


            }


            resultMap.put("yearContrast", yearContrast);


            //城市名


            QueryWrapper<SysArea> sysAreaQueryWrapper = new QueryWrapper<>();


            sysAreaQueryWrapper.select("area_name").eq("area_code", cityCode);


            String areaName = sysAreaService.getOne(sysAreaQueryWrapper).getAreaName();


            resultMap.put("cityName", areaName);






            //城市名


            for (SysArea sysArea : sysAreas) {


                if (cityCode.equals(sysArea.getAreaCode())) {


                    resultMap.put("cityName", sysArea.getAreaName());


                    break;


                }


            }


            result.add(resultMap);


        });


        return result;


    }




    /**


     * @param regionCodes 所要获取城市的编码集合


     * @param start       所要获取数据的开始时间,精确到日，2021-11-03 00:00:00


     * @param end         所要获取数据的结束时间,精确到日，2021-11-25 00:00:00


     * @param sysAreas 所要获取城市集合


     * @param start    所要获取数据的开始时间,精确到日，2021-11-03 00:00:00


     * @param end      所要获取数据的结束时间,精确到日，2021-11-25 00:00:00


     * @return 功能：自定义排名


     */


    private List<Map<String, Object>> customRank(List<Object> regionCodes, String start, String end) {


    private List<Map<String, Object>> customRank(List<SysArea> sysAreas, String start, String end) {


        List<Integer> regionCodes = sysAreas.stream()


                .map(SysArea::getAreaCode)


                .collect(Collectors.toList());




        //需要均值计算的因子


        List<String> sensors = Arrays.asList("PM2_5", "PM10", "SO2", "NO2");


        List<Map<String, Object>> result = new ArrayList<>();


@@ -687,7 +751,8 @@


                .in("city_code", regionCodes);


        List<Map<String, Object>> thisMonthData = cityAqiDailyService.listMaps(cityAqiDailyQueryWrapper);


        //按city_code分组


        Map<String, List<Map<String, Object>>> customMap = thisMonthData.parallelStream().collect(Collectors.groupingBy(o -> o.get("city_code").toString()));


        Map<Integer, List<Map<String, Object>>> customMap = thisMonthData.parallelStream()


                .collect(Collectors.groupingBy(o -> Integer.parseInt(o.get("city_code").toString())));


        customMap.forEach((cityCode, value) -> {


            Map<String, Object> resultMap = new HashMap<>();




@@ -741,11 +806,12 @@


            resultMap.put("O3_8H", resultMap.remove("O3"));




            //城市名


            QueryWrapper<SysArea> sysAreaQueryWrapper = new QueryWrapper<>();


            sysAreaQueryWrapper.select("area_name").eq("area_code", cityCode);


            String areaName = sysAreaService.getOne(sysAreaQueryWrapper).getAreaName();


            resultMap.put("cityName", areaName);




            for (SysArea sysArea : sysAreas) {


                if (cityCode.equals(sysArea.getAreaCode())) {


                    resultMap.put("cityName", sysArea.getAreaName());


                    break;


                }


            }


            result.add(resultMap);


        });


        return result;


@@ -826,7 +892,7 @@


    }




    @Override


    public Map<String, Object> provincialRanking(Integer organizationId) {


    public Map<String, Object> provincialRanking(Integer regionCode) {


        //结果集


        Map<String, Object> result = new HashMap<>();




@@ -835,10 +901,10 @@


        Date yesterday = DateUtils.dataToTimeStampTime(DateUtils.getDateOfDay(now, -1), DateUtils.yyyy_MM_dd_EN);


        String dateString = DateUtils.dateToDateString(yesterday, DateUtils.yyyy_MM_dd_HH_mm_ss_EN);




        String s = String.valueOf(regionCode);


        //获取省,市code


        Organization organization = organizationService.getById(organizationId);


        Integer provinceCode = organization.getProvinceCode();


        Integer cityCode = organization.getCityCode();


        Integer provinceCode = Integer.parseInt(s.substring(0, 2) + "0000");


        Integer cityCode = Integer.parseInt(s.substring(0, 4) + "00");


        //获取省内所有city_code


        QueryWrapper<SysArea> wrapper = new QueryWrapper<>();


        wrapper.select("area_code").eq("parent_code", provinceCode);


@@ -921,8 +987,6 @@




        //年排名，按累计综指排


        sortByField(ranks, "compositeIndexYear");


        //年排名结果


        sortByField(ranks, "compositeIndexYear");


        Map<String, Object> yearMap = rankByField(ranks, cityCode, "compositeIndexYear", cityCodes.size());


        if (ObjectUtils.isEmpty(yearMap)) {


            yearMap.put("rank", null);


@@ -966,6 +1030,4 @@


        }


        return result;


    }






}