From a54daffd42f8d62eee0ac01b79d126d100f08e1d Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Tue, 09 Nov 2021 10:32:02 +0800
Subject: [PATCH] screen-api            增加获取今年天气污染等级比例接口

---
 screen-api/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java |   74 +++++++++++++++++++++++++++++++++++++
 1 files changed, 74 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..d4c1c90 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,19 @@
 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.PollutionLevelProportionDTO;
 import com.moral.api.service.CityAqiDailyService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+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>
@@ -17,4 +26,69 @@
 @Service
 public class CityAqiDailyServiceImpl extends ServiceImpl<CityAqiDailyMapper, CityAqiDaily> implements CityAqiDailyService {
 
+    @Autowired
+    CityAqiDailyMapper cityAqiDailyMapper;
+
+    @Override
+    public PollutionLevelProportionDTO queryPollutionLevelsForThisYear(Integer regionCode) {
+        //������������  ��������������� ���
+        Date startDate = DateUtils.getFirstDayOfCurrYear();
+        Date endDate = DateUtils.getYesterdayDate();
+        //������������
+        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 HashMap<>();
+        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);
+            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<>();
+        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 proprtionD = MathUtils.mul(proportionTmp,100);
+            String proportion = proprtionD.toString()+"%";
+            pollutionMap.put("proportion",proportion);
+            pollutionMap.put("pollution",pollution);
+            pollutionMap.put("days",days);
+            valueMap.add(pollutionMap);
+        }
+        //������������������
+        PollutionLevelProportionDTO dto = new PollutionLevelProportionDTO();
+        dto.setTime(time);
+        dto.setValues(valueMap);
+        return dto;
+    }
 }

--
Gitblit v1.8.0