package com.moral.api.service.impl;
|
|
import com.alibaba.fastjson.JSON;
|
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.cityAQI.PieChartOfPollutionLevelVO;
|
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;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import java.util.*;
|
|
/**
|
* <p>
|
* 城市aqi日数据表 服务实现类
|
* </p>
|
*
|
* @author moral
|
* @since 2021-11-01
|
*/
|
@Service
|
public class CityAqiDailyServiceImpl extends ServiceImpl<CityAqiDailyMapper, CityAqiDaily> implements CityAqiDailyService {
|
|
@Autowired
|
CityAqiDailyMapper cityAqiDailyMapper;
|
|
@Override
|
public PieChartOfPollutionLevelVO 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);
|
wrapper.between("time",startDate,endDate);
|
List<CityAqiDaily> cityAqiDailies = cityAqiDailyMapper.selectList(wrapper);
|
//取开始结束时间,拼装返回time
|
if(cityAqiDailies.size()==0)
|
return null;
|
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");
|
tmpMap.put(tmpTime,cityAqiDaily);
|
}
|
cityAqiDailies = new ArrayList<CityAqiDaily>(tmpMap.values());
|
//统计各污染等级的天数
|
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.classCodeOfPollutionByAqi(aqi);
|
Integer days = pollutionDaysMap.get(pollutionClass);
|
pollutionDaysMap.put(pollutionClass,days+1);
|
}
|
//计算比例
|
List<Map<String,Object>> valueMap = new ArrayList<>();
|
Set<Map.Entry<String, Integer>> entries = pollutionDaysMap.entrySet();
|
Iterator<Map.Entry<String, Integer>> iterator = entries.iterator();
|
Integer totalDays = 0;
|
while(iterator.hasNext()){
|
Map.Entry<String, Integer> entry = iterator.next();
|
Map<String,Object> pollutionMap = new HashMap<>();
|
String pollution = entry.getKey();
|
Integer days = entry.getValue();
|
Double proportionTmp = MathUtils.division(Double.valueOf(days),Double.valueOf(cityAqiDailies.size()),3);
|
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);
|
totalDays+=days;
|
}
|
//封装返回对象
|
PieChartOfPollutionLevelVO VO = new PieChartOfPollutionLevelVO();
|
VO.setTime(time);
|
VO.setValues(valueMap);
|
VO.setTotalDays(totalDays);
|
return VO;
|
}
|
}
|
