package com.moral.api.service.impl;
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.moral.api.entity.CityAqi;
|
import com.moral.api.entity.CityAqiDaily;
|
import com.moral.api.mapper.CityAqiDailyMapper;
|
import com.moral.api.service.CityAqiDailyService;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.moral.api.service.CityAqiService;
|
import com.moral.util.AmendUtils;
|
import com.moral.util.ComprehensiveIndexUtils;
|
import com.moral.util.DateUtils;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.util.ObjectUtils;
|
|
import java.util.ArrayList;
|
import java.util.Arrays;
|
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;
|
|
/**
|
* <p>
|
* 城市aqi日数据表 服务实现类
|
* </p>
|
*
|
* @author moral
|
* @since 2021-10-28
|
*/
|
@Service
|
public class CityAqiDailyServiceImpl extends ServiceImpl<CityAqiDailyMapper, CityAqiDaily> implements CityAqiDailyService {
|
|
@Autowired
|
private CityAqiDailyMapper cityAqiDailyMapper;
|
|
@Autowired
|
private CityAqiService cityAqiService;
|
|
@Override
|
public void insertCityAqiDaily() {
|
List<String> sensors = Arrays.asList("pm2_5", "pm10", "so2", "no2", "co", "o3", "aqi");
|
String format = DateUtils.yyyy_MM_dd_EN;
|
Date now = new Date();
|
//开始时间，昨日
|
Date start = DateUtils.dataToTimeStampTime(DateUtils.getDateOfDay(now, -1), format);
|
//结束时间，今日
|
Date end = DateUtils.dataToTimeStampTime(now, format);
|
//获取所有城市aqi小时数据
|
QueryWrapper<CityAqi> wrapper = new QueryWrapper<>();
|
wrapper.select("city_code", "time", "value")
|
.ge("time", DateUtils.dateToDateString(start))
|
.le("time", DateUtils.dateToDateString(end));
|
List<Map<String, Object>> dailyData = cityAqiService.listMaps(wrapper);
|
|
if (dailyData.size() == 0) {
|
return;
|
}
|
//按city_code分组
|
Map<String, List<Map<String, Object>>> data = dailyData.parallelStream().collect(Collectors.groupingBy(o -> o.get("city_code").toString()));
|
|
data.forEach((cityCode, value) -> {
|
Map<String, Object> jsonMap = new HashMap<>();
|
CityAqiDaily cityAqiDaily = new CityAqiDaily();
|
cityAqiDaily.setCityCode(Integer.parseInt(cityCode));
|
cityAqiDaily.setTime(start);
|
|
//中间变量，用于计算除臭氧外其它因子
|
List<Map<String, Object>> tempValue = new ArrayList<>(value);
|
|
//移除第一天数据（0点的），O3滑动值第一条数据是从1点-8点
|
value.removeIf(map -> ((Date) map.get("time")).getTime() == start.getTime());
|
|
Double o3OfDay = AmendUtils.o3OfDay(value);
|
if (o3OfDay != null) {
|
jsonMap.put("o3", o3OfDay);
|
}
|
//除臭氧外其他因子均值计算
|
tempValue.removeIf(o -> ((Date) o.get("time")).getTime() == end.getTime());
|
|
sensors.forEach(sensor -> {
|
OptionalDouble optionalDouble = tempValue.parallelStream().flatMapToDouble(v -> {
|
Map<String, Object> dataValue = JSONObject.parseObject((String) v.get("value"), Map.class);
|
//臭氧单独计算
|
if ("o3".equals(sensor)) {
|
return null;
|
}
|
|
Object sensorValue = dataValue.get(sensor);
|
if (ObjectUtils.isEmpty(sensorValue)) {
|
return null;
|
}
|
double aDouble = Double.parseDouble(sensorValue.toString());
|
return DoubleStream.of(aDouble);
|
}).average();
|
if (optionalDouble.isPresent()) {
|
//银行家算法修约
|
double sciCal;
|
if ("co".equals(sensor)) {
|
sciCal = AmendUtils.sciCal(optionalDouble.getAsDouble(), 1);
|
} else {
|
sciCal = AmendUtils.sciCal(optionalDouble.getAsDouble(), 0);
|
}
|
jsonMap.put(sensor, sciCal);
|
}
|
});
|
|
//日综合指数计算
|
Double compositeIndex = ComprehensiveIndexUtils.dailyData(jsonMap);
|
jsonMap.put("compositeIndex", compositeIndex);
|
|
cityAqiDaily.setValue(JSONObject.toJSONString(jsonMap));
|
cityAqiDailyMapper.insert(cityAqiDaily);
|
});
|
}
|
}
|
