kaiyu
2021-11-12 6a8b22c541448dbe15eb1b046dbf456131c2f325
screen-api/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java
@@ -4,9 +4,11 @@
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.moral.api.entity.CityAqiDaily;
import com.moral.api.mapper.CityAqiDailyMapper;
import com.moral.api.pojo.dto.dataDisplay.PollutionLevelProportionDTO;
import com.moral.api.pojo.dto.dataDisplay.PieChartOfPollutionLevelDTO;
import com.moral.api.pojo.form.aqi.QueryPieChartOfPollutionLevelForm;
import com.moral.api.service.CityAqiDailyService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.moral.constant.Constants;
import com.moral.util.AQIUtils;
import com.moral.util.DateUtils;
import com.moral.util.MathUtils;
@@ -30,10 +32,11 @@
    CityAqiDailyMapper cityAqiDailyMapper;
    @Override
    public PollutionLevelProportionDTO queryPollutionLevelsForThisYear(Integer regionCode) {
        //获取时间  今年第一天 至
        Date startDate = DateUtils.getFirstDayOfCurrYear();
        Date endDate = DateUtils.getYesterdayDate();
    public PieChartOfPollutionLevelDTO queryPieChartOfPollutionLevels(QueryPieChartOfPollutionLevelForm form) {
        //取参
        Integer regionCode = form.getRegionCode();
        Date startDate = form.getStartDate();
        Date endDate = form.getEndDate();
        //查询数据
        QueryWrapper<CityAqiDaily> wrapper = new QueryWrapper<>();
        wrapper.eq("city_code",regionCode);
@@ -45,7 +48,7 @@
        String startTime = DateUtils.dateToDateString(cityAqiDailies.get(0).getTime(),"yyyy-MM-dd");
        String endTime = DateUtils.dateToDateString(cityAqiDailies.get(cityAqiDailies.size()-1).getTime(),"yyyy-MM-dd");
        String time = startTime+" 至 "+endTime;
        //如果时间有重复则去重
        //如果数据的时间有重复则去重
        Map<String,CityAqiDaily> tmpMap = new HashMap<>();
        for (CityAqiDaily cityAqiDaily : cityAqiDailies) {
            String tmpTime = DateUtils.dateToDateString(cityAqiDaily.getTime(),"yyyy-MM-dd");
@@ -53,20 +56,22 @@
        }
        cityAqiDailies = new ArrayList<CityAqiDaily>(tmpMap.values());
        //统计各污染等级的天数
        Map<String,Integer> pollutionDaysMap = new HashMap<>();
        Map<String,Integer> pollutionDaysMap = new LinkedHashMap<>();
        pollutionDaysMap.put(Constants.EXCELLENT_WEATHER_CODE,0);
        pollutionDaysMap.put(Constants.GOOD_WEATHER_CODE,0);
        pollutionDaysMap.put(Constants.MILD_WEATHER_CODE,0);
        pollutionDaysMap.put(Constants.MIDDLE_WEATHER_CODE,0);
        pollutionDaysMap.put(Constants.SERIOUS_WEATHER_CODE,0);
        pollutionDaysMap.put(Constants.SERVER_WEATHER_CODE,0);
        for (CityAqiDaily cityAqiDaily : cityAqiDailies) {
            Map<String,Object> valueMap = JSON.parseObject(cityAqiDaily.getValue(),Map.class);
            if(valueMap.get("AQI")==null)
                continue;
            Integer aqi = Integer.valueOf(valueMap.get("AQI").toString());
            //计算AQI污染等级
            String pollutionClass = AQIUtils.classOfPollutionByAqi(aqi);
            String pollutionClass = AQIUtils.classCodeOfPollutionByAqi(aqi);
            Integer days = pollutionDaysMap.get(pollutionClass);
            if(days==null){
                pollutionDaysMap.put(pollutionClass,1);
            }else{
                pollutionDaysMap.put(pollutionClass,days+1);
            }
        }
        //计算比例
        List<Map<String,Object>> valueMap = new ArrayList<>();
@@ -78,15 +83,15 @@
            String pollution = entry.getKey();
            Integer days = entry.getValue();
            Double proportionTmp = MathUtils.division(Double.valueOf(days),Double.valueOf(cityAqiDailies.size()),3);
            Double proprtionD = MathUtils.mul(proportionTmp,100);
            String proportion = proprtionD.toString()+"%";
            Double proportionD = MathUtils.mul(proportionTmp,100);
            String proportion = proportionD.toString()+"%";
            pollutionMap.put("proportion",proportion);
            pollutionMap.put("pollution",pollution);
            pollutionMap.put("days",days);
            valueMap.add(pollutionMap);
        }
        //封装返回对象
        PollutionLevelProportionDTO dto = new PollutionLevelProportionDTO();
        PieChartOfPollutionLevelDTO dto = new PieChartOfPollutionLevelDTO();
        dto.setTime(time);
        dto.setValues(valueMap);
        return dto;