| | |
|---|
| | | 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.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; |
|---|
| | | import org.springframework.beans.factory.annotation.Autowired; |
|---|
| | | import org.springframework.stereotype.Service; |
|---|
| | | |
|---|
| | | import java.util.*; |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * <p> |
|---|
| | |
|---|
| | | @Service |
|---|
| | | public class CityAqiDailyServiceImpl extends ServiceImpl<CityAqiDailyMapper, CityAqiDaily> implements CityAqiDailyService { |
|---|
| | | |
|---|
| | | @Autowired |
|---|
| | | CityAqiDailyMapper cityAqiDailyMapper; |
|---|
| | | |
|---|
| | | @Override |
|---|
| | | 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); |
|---|
| | | 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(); |
|---|
| | | 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); |
|---|
| | | } |
|---|
| | | //封装返回对象 |
|---|
| | | PieChartOfPollutionLevelDTO dto = new PieChartOfPollutionLevelDTO(); |
|---|
| | | dto.setTime(time); |
|---|
| | | dto.setValues(valueMap); |
|---|
| | | return dto; |
|---|
| | | } |
|---|
| | | } |
|---|
