From 56e65455f7c4761a73e75c2f8e93e679d0c2e0af Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Thu, 30 Dec 2021 11:49:54 +0800
Subject: [PATCH] screen-api 添加城市优良天气对比接口
---
screen-api/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java | 323 +++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 323 insertions(+), 0 deletions(-)
diff --git a/screen-api/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java b/screen-api/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java
index 42e8956..44e67e4 100644
--- a/screen-api/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java
+++ b/screen-api/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java
@@ -1,10 +1,30 @@
package com.moral.api.service.impl;
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.moral.api.entity.CityAqiDaily;
+import com.moral.api.entity.SysArea;
import com.moral.api.mapper.CityAqiDailyMapper;
+import com.moral.api.pojo.dto.cityAQI.CityPollutionLevel;
+import com.moral.api.pojo.dto.cityAQI.MonthlyPollutionLevel;
+import com.moral.api.pojo.dto.cityAQI.PollutionDaysAndProportion;
+import com.moral.api.pojo.form.aqi.AreaPollutionLevelForm;
+import com.moral.api.pojo.vo.cityAQI.PieChartOfPollutionLevelVO;
+import com.moral.api.pojo.form.aqi.ChartOfPollutionLevelForm;
import com.moral.api.service.CityAqiDailyService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.moral.api.service.SysAreaService;
+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 org.springframework.util.ObjectUtils;
+
+import java.text.SimpleDateFormat;
+import java.util.*;
/**
* <p>
@@ -17,4 +37,307 @@
@Service
public class CityAqiDailyServiceImpl extends ServiceImpl<CityAqiDailyMapper, CityAqiDaily> implements CityAqiDailyService {
+ @Autowired
+ CityAqiDailyMapper cityAqiDailyMapper;
+ @Autowired
+ SysAreaService sysAreaService;
+
+ @Override
+ public PieChartOfPollutionLevelVO queryPieChartOfPollutionLevels(ChartOfPollutionLevelForm 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;
+ }
+
+ @Override
+ public List<MonthlyPollutionLevel> queryBarChartOfPollutionLevels(ChartOfPollutionLevelForm 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);
+ wrapper.orderByAsc("time");
+ List<CityAqiDaily> cityAqiDailies = cityAqiDailyMapper.selectList(wrapper);
+ //���������������������������������������
+ Map<String, CityAqiDaily> tmpMap = new LinkedHashMap<>();
+ 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, List<CityAqiDaily>> monthlyDataMap = new LinkedHashMap<>();
+ cityAqiDailies.forEach(value -> {
+ String month = DateUtils.dateToDateString(value.getTime(), "yyyy-MM");
+ List<CityAqiDaily> dataList = monthlyDataMap.get(month);
+ if (ObjectUtils.isEmpty(dataList))
+ dataList = new ArrayList<>();
+ dataList.add(value);
+ monthlyDataMap.put(month, dataList);
+ });
+ //���������������������������
+ List<MonthlyPollutionLevel> monthlyPollutionLevels = new ArrayList<>();
+ Set<Map.Entry<String, List<CityAqiDaily>>> monthlyEntries = monthlyDataMap.entrySet();
+ Iterator<Map.Entry<String, List<CityAqiDaily>>> monthlyIterator = monthlyEntries.iterator();
+ while (monthlyIterator.hasNext()) {
+ Map.Entry<String, List<CityAqiDaily>> monthlyEntry = monthlyIterator.next();
+ List<CityAqiDaily> monthlyValues = monthlyEntry.getValue();
+ //���������������������������������������
+ 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 : monthlyValues) {
+ 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<PollutionDaysAndProportion> pollutionDaysAndProportions = 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();
+ String pollution = entry.getKey();
+ Integer days = entry.getValue();
+ Double proportionTmp = MathUtils.division(Double.valueOf(days), Double.valueOf(monthlyValues.size()), 3);
+ Double proportionD = MathUtils.mul(proportionTmp, 100);
+ String proportion = proportionD.toString() + "%";
+ PollutionDaysAndProportion pollutionDaysAndProportion = new PollutionDaysAndProportion();
+ pollutionDaysAndProportion.setDays(days);
+ pollutionDaysAndProportion.setPollutionLevel(pollution);
+ pollutionDaysAndProportion.setProportion(proportion);
+ pollutionDaysAndProportions.add(pollutionDaysAndProportion);
+ }
+ //������������������
+ MonthlyPollutionLevel monthlyPollutionLevel = new MonthlyPollutionLevel();
+ monthlyPollutionLevel.setPollutionDaysAndProportions(pollutionDaysAndProportions);
+ monthlyPollutionLevel.setTime(monthlyEntry.getKey());
+ monthlyPollutionLevels.add(monthlyPollutionLevel);
+ }
+ return monthlyPollutionLevels;
+ }
+
+ @Override
+ public Map<String, Map<String, Object>> oneYearsData(Map map) {
+ int city_code = Integer.parseInt(map.get("cityCode").toString());
+ String year = map.get("year").toString();
+ String startTime = year + "-01-01 00:00:00";
+ String endTime = year + "-12-31 23:59:59";
+ QueryWrapper<CityAqiDaily> cityAqiDailyQueryWrapper = new QueryWrapper<>();
+ cityAqiDailyQueryWrapper.eq("city_code", city_code);
+ cityAqiDailyQueryWrapper.between("time", startTime, endTime);
+ List<CityAqiDaily> cityAqiDailies = cityAqiDailyMapper.selectList(cityAqiDailyQueryWrapper);
+ Map<String, Map<String, Object>> resultMap = new HashMap<>();
+ for (int i = 1; i < 13; i++) {
+ Map<String, Object> monthMap = new HashMap<>();
+ String month = null;
+ if (i < 10) {
+ month = "0" + i;
+ } else {
+ month = String.valueOf(i);
+ }
+ String monthDate = year + "-" + month;
+ String monthFirstDay = year + "-" + month + "-01";
+ int monthDay = DateUtils.getMonthDay(monthFirstDay);
+ for (int j = 1; j < monthDay + 1; j++) {
+ String day = null;
+ if (j < 10) {
+ day = "0" + j;
+ } else {
+ day = String.valueOf(j);
+ }
+ String dayDate = year + "-" + month + "-" + day;
+ monthMap.put(dayDate, null);
+ }
+ resultMap.put(monthDate, monthMap);
+ }
+ SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+ for (CityAqiDaily cityAqiDaily : cityAqiDailies) {
+ String time = sdf.format(cityAqiDaily.getTime());
+ String monthDate = time.substring(0, 7);
+ String dayDate = time.substring(0, 10);
+ Double PM2_5 = null;
+ Double PM10 = null;
+ Double SO2 = null;
+ Double NO2 = null;
+ Double CO = null;
+ Double O3 = null;
+ String primaryPollutant = null;
+ String AQI = null;
+ String value = cityAqiDaily.getValue();
+ JSONObject jsonObject = new JSONObject();
+ jsonObject = JSONObject.parseObject(value);
+ if (!ObjectUtils.isEmpty(jsonObject.get("PM2_5"))) {
+ PM2_5 = (Double.parseDouble(jsonObject.get("PM2_5").toString()));
+ }
+ if (!ObjectUtils.isEmpty(jsonObject.get("PM10"))) {
+ PM10 = (Double.parseDouble(jsonObject.get("PM10").toString()));
+ }
+ if (!ObjectUtils.isEmpty(jsonObject.get("SO2"))) {
+ SO2 = (Double.parseDouble(jsonObject.get("SO2").toString()));
+ }
+ if (!ObjectUtils.isEmpty(jsonObject.get("NO2"))) {
+ NO2 = (Double.parseDouble(jsonObject.get("NO2").toString()));
+ }
+ if (!ObjectUtils.isEmpty(jsonObject.get("CO"))) {
+ CO = (Double.parseDouble(jsonObject.get("CO").toString()));
+ }
+ if (!ObjectUtils.isEmpty(jsonObject.get("O3"))) {
+ O3 = (Double.parseDouble(jsonObject.get("O3").toString()));
+ }
+ if (!ObjectUtils.isEmpty(jsonObject.get("primaryPollutant"))) {
+ primaryPollutant = jsonObject.get("primaryPollutant").toString();
+ }
+ if (!ObjectUtils.isEmpty(jsonObject.get("AQI"))) {
+ AQI = jsonObject.get("AQI").toString();
+ }
+ Map<String, Object> valueMap = new HashMap<>();
+ valueMap.put("PM2_5", PM2_5);
+ valueMap.put("PM10", PM10);
+ valueMap.put("SO2", SO2);
+ valueMap.put("NO2", NO2);
+ valueMap.put("CO", CO);
+ valueMap.put("O3", O3);
+ valueMap.put("AQI", AQI);
+ valueMap.put("primaryPollutant", primaryPollutant);
+ Map<String, Object> monthMap = new HashMap<>();
+ monthMap = resultMap.get(monthDate);
+ monthMap.put(dayDate, valueMap);
+ resultMap.put(monthDate, monthMap);
+ }
+ return resultMap;
+ }
+
+ @Override
+ public List<CityPollutionLevel> queryAreaPollutionLevel(AreaPollutionLevelForm form) {
+ //������
+ Integer regionCode = form.getRegionCode();
+ Date startDate = form.getStartDate();
+ Date endDate = form.getEndDate();
+ //������������������������������
+ List<SysArea> allAreas = sysAreaService.getChildren(regionCode);
+ //������������������������
+ List<CityPollutionLevel> datas = new ArrayList<>();
+ //��������������������� ������������
+ for (SysArea area : allAreas) {
+ //������������
+ QueryWrapper<CityAqiDaily> wrapper = new QueryWrapper<>();
+ wrapper.eq("city_code", area.getAreaCode());
+ wrapper.between("time", startDate, endDate);
+ wrapper.orderByAsc("time");
+ List<CityAqiDaily> cityAqiDailies = cityAqiDailyMapper.selectList(wrapper);
+ //���������������������������������������
+ Map<String, CityAqiDaily> tmpMap = new LinkedHashMap<>();
+ 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);
+ }
+ //������������
+ CityPollutionLevel cityPollutionLevel = new CityPollutionLevel();
+ cityPollutionLevel.setRegionName(area.getAreaName());
+ cityPollutionLevel.setExcellentWeatherDays(pollutionDaysMap.get(Constants.EXCELLENT_WEATHER_CODE));
+ cityPollutionLevel.setGoodWeatherDays(pollutionDaysMap.get(Constants.GOOD_WEATHER_CODE));
+ cityPollutionLevel.setMildWeatherDays(pollutionDaysMap.get(Constants.MILD_WEATHER_CODE));
+ cityPollutionLevel.setMiddleWeatherDays(pollutionDaysMap.get(Constants.MIDDLE_WEATHER_CODE));
+ cityPollutionLevel.setSeriousWeatherDays(pollutionDaysMap.get(Constants.SERIOUS_WEATHER_CODE));
+ cityPollutionLevel.setServerWeatherDays(pollutionDaysMap.get(Constants.SERVER_WEATHER_CODE));
+ datas.add(cityPollutionLevel);
+ }
+ return datas;
+ }
+
}
--
Gitblit v1.8.0