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.CityAqi;
|
import com.moral.api.entity.Forecast;
|
import com.moral.api.mapper.CityAqiMapper;
|
import com.moral.api.mapper.ForecastMapper;
|
import com.moral.api.service.CityAqiService;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.moral.constant.RedisConstants;
|
import com.moral.util.AQIUtils;
|
import com.moral.util.DateUtils;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.redis.core.RedisTemplate;
|
import org.springframework.stereotype.Service;
|
import org.springframework.util.ObjectUtils;
|
|
import java.util.*;
|
|
/**
|
* <p>
|
* 城市aqi实测小时数据表 服务实现类
|
* </p>
|
*
|
* @author moral
|
* @since 2021-09-28
|
*/
|
@Service
|
public class CityAqiServiceImpl extends ServiceImpl<CityAqiMapper, CityAqi> implements CityAqiService {
|
|
@Autowired
|
private CityAqiMapper cityAqiMapper;
|
|
@Autowired
|
private ForecastMapper forecastMapper;
|
|
@Autowired
|
private RedisTemplate redisTemplate;
|
|
@Override
|
public List<Map<String, Object>> measuredCompareForecastOfO3(Map<String, Object> params) {
|
String regionCode = params.get("regionCode").toString();
|
String time = params.get("time").toString();
|
//预测数据
|
QueryWrapper<Forecast> forecastQueryWrapper = new QueryWrapper<>();
|
forecastQueryWrapper.select("time", "value")
|
.eq("city_code", regionCode)
|
.likeRight("time", time);
|
List<Map<String, Object>> forecastData = forecastMapper.selectMaps(forecastQueryWrapper);
|
//实测数据
|
QueryWrapper<CityAqi> cityAqiQueryWrapper = new QueryWrapper<>();
|
cityAqiQueryWrapper.select("time", "value")
|
.eq("city_code", regionCode)
|
.likeRight("time", time);
|
List<Map<String, Object>> measuredData = cityAqiMapper.selectMaps(cityAqiQueryWrapper);
|
|
List<Map<String, Object>> result = new ArrayList<>();
|
for (int i = 0; i < 48; i++) {
|
Map<String, Object> map = new HashMap<>();
|
if (i % 2 == 0) {
|
map.put("type", "预测");
|
Date d = DateUtils.addHours(DateUtils.getDate(time), i / 2);
|
map.put("time", DateUtils.dateToDateString(d, DateUtils.yyyy_MM_dd_HH_EN));
|
for (Map<String, Object> forecastDatum : forecastData) {
|
Date date = (Date) forecastDatum.get("time");
|
String value = forecastDatum.get("value").toString();
|
Map<String, Object> data = JSONObject.parseObject(value, Map.class);
|
Object o3 = data.get("o3");
|
if (i == DateUtils.getHour(date) * 2) {
|
if (!ObjectUtils.isEmpty(o3)) {
|
map.put("O3", o3);
|
}
|
}
|
}
|
} else {
|
map.put("type", "实测");
|
Date d = DateUtils.addHours(DateUtils.getDate(time), (i - 1) / 2);
|
map.put("time", DateUtils.dateToDateString(d, DateUtils.yyyy_MM_dd_HH_EN));
|
for (Map<String, Object> measuredDatum : measuredData) {
|
Date date = (Date) measuredDatum.get("time");
|
String value = measuredDatum.get("value").toString();
|
Map<String, Object> data = JSONObject.parseObject(value, Map.class);
|
Object o3 = data.get("o3");
|
if (i == (DateUtils.getHour(date) * 2 + 1)) {
|
if (!ObjectUtils.isEmpty(o3)) {
|
map.put("O3", o3);
|
}
|
}
|
}
|
}
|
result.add(map);
|
}
|
return result;
|
}
|
|
@Override
|
public Map<String, Object> queryCityAqiByRegionCode(Integer regionCode) {
|
Map<String, Object> value = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.CITY_AQI, String.valueOf(regionCode));
|
if (value == null)
|
value = queryCityAqiByRegionCodeFromDB(regionCode);
|
//根据AQI计算污染等级
|
if (value == null || value.get("aqi") == null)
|
return null;
|
Integer aqi = Integer.parseInt(value.get("aqi").toString());
|
String category = AQIUtils.classOfPollutionByAqi(aqi);
|
value.put("category", category);
|
return value;
|
}
|
|
@Override
|
public Map<String, Object> query24HoursAqiByRegionCode(Integer regionCode) {
|
//查询最新一条数据，用于获取最新的时间
|
QueryWrapper<CityAqi> lastDataWrapper = new QueryWrapper<>();
|
lastDataWrapper.eq("city_code", regionCode);
|
lastDataWrapper.orderByDesc("time");
|
lastDataWrapper.last(true, "limit 1");
|
CityAqi cityAqi = cityAqiMapper.selectOne(lastDataWrapper);
|
//算出前24小时的时间点
|
Date endDate = cityAqi.getTime();
|
Date startDate = DateUtils.addHours(endDate, -23);
|
//查询数据
|
QueryWrapper<CityAqi> wrapper = new QueryWrapper<>();
|
wrapper.between("time", startDate, endDate);
|
wrapper.eq("city_code", regionCode);
|
List<CityAqi> cityAqis = cityAqiMapper.selectList(wrapper);
|
//如果数据不足24小时则补全
|
if (cityAqis.size() != 24) {
|
Map<Date, CityAqi> dateCityAqiMap = new HashMap<>();
|
cityAqis.forEach(value -> dateCityAqiMap.put(value.getTime(), value));
|
for (int i = 0; i < 24; i++) {
|
Date date = DateUtils.addHours(startDate, i);
|
CityAqi cityAqi1 = dateCityAqiMap.get(date);
|
if (cityAqi1 == null) {
|
CityAqi newCityAqi = new CityAqi();
|
newCityAqi.setTime(date);
|
cityAqis.add(newCityAqi);
|
}
|
}
|
//按照时间进行排序
|
cityAqis.sort(Comparator.comparing(CityAqi::getTime));
|
}
|
//封装返回数据,map的key为HH:mm格式的时间，value为aqi的数值
|
Map<String, Object> result = new LinkedHashMap<>();
|
for (CityAqi aqi : cityAqis) {
|
String key = DateUtils.dateToDateString(aqi.getTime(), "HH:mm");
|
String allDataJson = aqi.getValue();
|
if (allDataJson == null) {
|
result.put(key, "");
|
continue;
|
}
|
Map<String, Object> allDataMap = JSON.parseObject(allDataJson, Map.class);
|
Object aqiData = allDataMap.get("aqi");
|
if (aqiData == null)
|
result.put(key, "");
|
else
|
result.put(key, aqiData);
|
}
|
return result;
|
}
|
|
/**
|
* @Description: 从数据库查询数据
|
* @Param: [regionCode]
|
* @return: java.util.Map<java.lang.String , java.lang.Object>
|
* @Author: 陈凯裕
|
* @Date: 2021/10/28
|
*/
|
private Map<String, Object> queryCityAqiByRegionCodeFromDB(Integer regionCode) {
|
QueryWrapper<CityAqi> wrapper = new QueryWrapper();
|
wrapper.eq("city_code", regionCode);
|
wrapper.orderByDesc("time");
|
wrapper.last(true, "limit 1");
|
CityAqi cityAqi = cityAqiMapper.selectOne(wrapper);
|
if (cityAqi == null)
|
return null;
|
String value = cityAqi.getValue();
|
redisTemplate.opsForHash().put(RedisConstants.CITY_AQI, regionCode, value);
|
return JSON.parseObject(value, Map.class);
|
}
|
}
|
