From 0aca12e52beb725c489a83a8624f00af355aa5fa Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Thu, 20 Jan 2022 17:28:25 +0800
Subject: [PATCH] scren_job定时任务城市臭氧滑动值计算逻辑修改
---
screen-job/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java | 64 +++++++++++++++++++++++---------
1 files changed, 46 insertions(+), 18 deletions(-)
diff --git a/screen-job/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java b/screen-job/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java
index a9c5c9b..88b5cff 100644
--- a/screen-job/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java
+++ b/screen-job/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java
@@ -3,11 +3,11 @@
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.moral.api.entity.CityAqi;
-import com.moral.api.entity.CityConfig;
+import com.moral.api.entity.CityConfigAqi;
import com.moral.api.mapper.CityAqiMapper;
import com.moral.api.service.CityAqiService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import com.moral.api.service.CityConfigService;
+import com.moral.api.service.CityConfigAqiService;
import com.moral.constant.Constants;
import com.moral.constant.RedisConstants;
import com.moral.pojo.AQI;
@@ -23,11 +23,13 @@
import org.springframework.util.ObjectUtils;
import org.springframework.web.client.RestTemplate;
+import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.OptionalDouble;
+import java.util.stream.Collectors;
import java.util.stream.DoubleStream;
/**
@@ -45,7 +47,7 @@
private CityAqiMapper cityAqiMapper;
@Autowired
- private CityConfigService cityConfigService;
+ private CityConfigAqiService cityConfigAqiService;
@Autowired
private RestTemplate restTemplate;
@@ -61,21 +63,42 @@
//������������������time������������������
Date dataTime = DateUtils.addHours(time, -1);
Date start = null;
- if (DateUtils.getHour(time) >= 8) {
+ if (DateUtils.getHour(time) >= 8 || DateUtils.getHour(time) == 0) {
start = DateUtils.addHours(time, -8);
}
//������������������
- QueryWrapper<CityConfig> wrapper = new QueryWrapper<>();
+ QueryWrapper<CityConfigAqi> wrapper = new QueryWrapper<>();
wrapper.select("city_code", "city_id").eq("is_delete", Constants.NOT_DELETE);
- List<CityConfig> list = cityConfigService.list(wrapper);
+ List<CityConfigAqi> list = cityConfigAqiService.list(wrapper);
- for (CityConfig cityConfig : list) {
- Integer cityCode = cityConfig.getCityCode();
+ //������������������������������������������������
+ Map<Integer, List<Map<String, Object>>> cityData = null;
+ if (start != null) {
+ QueryWrapper<CityAqi> queryWrapper = new QueryWrapper<>();
+ List<Integer> cityCodes = list.stream().map(CityConfigAqi::getCityCode)
+ .collect(Collectors.toList());
+
+ queryWrapper.select("city_code", "time", "value")
+ .ge("time", DateUtils.dateToDateString(start))
+ .in("city_code", cityCodes);
+ List<Map<String, Object>> selectMaps = cityAqiMapper.selectMaps(queryWrapper);
+
+ //���city_code������
+ cityData = selectMaps.stream()
+ .collect(Collectors.groupingBy(o -> Integer.parseInt(o.get("city_code").toString())));
+ }
+
+
+ List<CityAqi> cityAqiList = new ArrayList<>();
+
+ for (CityConfigAqi cityConfigAqi : list) {
+ CityAqi cityAqi = new CityAqi();
+ Integer cityCode = cityConfigAqi.getCityCode();
Map<String, Object> data;
try {
//������������������������������
- data = restTemplate.getForObject("http://sapi.7drlb.com/api/mj?cityID={1}&apiKey=aqi", Map.class, cityConfig.getCityId());
+ data = restTemplate.getForObject("http://sapi.7drlb.com/api/mj?cityID={1}&apiKey=aqi", Map.class, cityConfigAqi.getCityId());
} catch (Exception e) {
continue;
}
@@ -101,18 +124,15 @@
aqi.put("AQI", aqi.remove("value"));
//o3���������������
- if (start != null) {
- QueryWrapper<CityAqi> queryWrapper = new QueryWrapper<>();
- queryWrapper.select("time", "value")
- .eq("city_code", cityCode)
- .ge("time", DateUtils.dateToDateString(start));
- List<Map<String, Object>> cityAqis = cityAqiMapper.selectMaps(queryWrapper);
+ if (cityData != null) {
+ List<Map<String, Object>> cityAqis = cityData.get(cityCode);
+
Map<String, Object> params = new HashMap<>();
params.put("time", DateUtils.dateToDateString(dataTime, DateUtils.yyyy_MM_dd_HH_mm_ss_S_EN));
params.put("value", JSONObject.toJSONString(aqi));
cityAqis.add(params);
if (cityAqis.size() >= 6) {
- OptionalDouble average = cityAqis.parallelStream().flatMapToDouble(v -> {
+ OptionalDouble average = cityAqis.stream().flatMapToDouble(v -> {
Map<String, Object> dataValue = JSONObject.parseObject((String) v.get("value"), Map.class);
double o3 = Double.parseDouble(dataValue.get("O3").toString());
return DoubleStream.of(o3);
@@ -142,13 +162,21 @@
aqi.put("primaryPollutant", aqiAndPollutant.getPrimaryPollutantNames());
- CityAqi cityAqi = new CityAqi();
cityAqi.setCityCode(cityCode);
cityAqi.setTime(dataTime);
cityAqi.setValue(JSONObject.toJSONString(aqi));
- cityAqiMapper.insert(cityAqi);
+ cityAqiList.add(cityAqi);
//������redis
redisTemplate.opsForHash().put(RedisConstants.CITY_AQI, cityCode.toString(), aqi);
}
+ cityAqiMapper.insertCityAqi(cityAqiList);
+ }
+
+ @Override
+ public List<CityAqi> getCityAqi() {
+ String time = DateUtils.getDateStringOfHour(-1, DateUtils.yyyy_MM_dd_HH_EN) + ":00:00";
+ QueryWrapper<CityAqi> queryWrapper = new QueryWrapper<>();
+ queryWrapper.eq("time", time);
+ return cityAqiMapper.selectList(queryWrapper);
}
}
--
Gitblit v1.8.0