From fc4f8786cbc684bda0937fa54fbb3db569b537fc Mon Sep 17 00:00:00 2001 From: cjl <276999030@qq.com> Date: Fri, 13 Oct 2023 11:53:47 +0800 Subject: [PATCH] fix:秒级数据调整补充提交 --- screen-job/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java | 60 ++++++++++++++++++++++++++++++++++++++++++------------------ 1 files changed, 42 insertions(+), 18 deletions(-) diff --git a/screen-job/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java b/screen-job/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java index 61a58ab..b98b8fc 100644 --- a/screen-job/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java +++ b/screen-job/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java @@ -8,13 +8,18 @@ import com.moral.api.service.CityAqiDailyService; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.moral.api.service.CityAqiService; +import com.moral.constant.Constants; +import com.moral.pojo.AQI; +import com.moral.util.AQIUtils; 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; @@ -43,7 +48,8 @@ @Override public void insertCityAqiDaily() { - List<String> sensors = Arrays.asList("pm2_5", "pm10", "so2", "no2", "co", "o3", "aqi"); + //��������������������������� + List<String> sensors = Arrays.asList("PM2_5", "PM10", "SO2", "NO2", "CO"); String format = DateUtils.yyyy_MM_dd_EN; Date now = new Date(); //��������������������� @@ -53,36 +59,35 @@ //������������������aqi������������ QueryWrapper<CityAqi> wrapper = new QueryWrapper<>(); wrapper.select("city_code", "time", "value") - .ge("time", DateUtils.dateToDateString(start)) - .lt("time", DateUtils.dateToDateString(end)); + .ge("time", start) + .lt("time", 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())); + Map<String, List<Map<String, Object>>> data = dailyData.stream() + .collect(Collectors.groupingBy(o -> o.get("city_code").toString())); + + List<CityAqiDaily> cityAqiDailies = new ArrayList<>(); data.forEach((cityCode, value) -> { - Map<String, Object> jsonMap = new HashMap<>(); CityAqiDaily cityAqiDaily = new CityAqiDaily(); + Map<String, Object> jsonMap = new HashMap<>(); cityAqiDaily.setCityCode(Integer.parseInt(cityCode)); cityAqiDaily.setTime(start); - - //��������������������������� - Double o3OfDay = AmendUtils.o3OfDay(value); - if (o3OfDay != null) { - jsonMap.put("o3", o3OfDay); + //O3��������������������������������������� + if (!ObjectUtils.isEmpty(value)) { + Double o3OfDay = AmendUtils.o3OfDay(value); + if (o3OfDay != null) { + jsonMap.put("O3", o3OfDay); + } } sensors.forEach(sensor -> { - OptionalDouble optionalDouble = value.parallelStream().flatMapToDouble(v -> { + OptionalDouble optionalDouble = value.stream().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; @@ -93,7 +98,7 @@ if (optionalDouble.isPresent()) { //��������������������� double sciCal; - if ("co".equals(sensor)) { + if ("CO".equals(sensor)) { sciCal = AmendUtils.sciCal(optionalDouble.getAsDouble(), 1); } else { sciCal = AmendUtils.sciCal(optionalDouble.getAsDouble(), 0); @@ -101,8 +106,27 @@ jsonMap.put(sensor, sciCal); } }); + + //���aqi,��������������������� + Map<String, Object> sixParamMap = new HashMap<>(); + sixParamMap.put(Constants.SENSOR_CODE_PM25, jsonMap.get("PM2_5")); + sixParamMap.put(Constants.SENSOR_CODE_PM10, jsonMap.get("PM10")); + sixParamMap.put(Constants.SENSOR_CODE_SO2, jsonMap.get("SO2")); + sixParamMap.put(Constants.SENSOR_CODE_NO2, jsonMap.get("NO2")); + sixParamMap.put(Constants.SENSOR_CODE_CO, jsonMap.get("CO")); + sixParamMap.put(Constants.SENSOR_CODE_O3, jsonMap.get("O3")); + + AQI aqi = AQIUtils.dailyAQI(sixParamMap); + jsonMap.put("AQI", aqi.getAQIValue()); + jsonMap.put("primaryPollutant", aqi.getPrimaryPollutantNames()); + + //��������������������� + Double compositeIndex = ComprehensiveIndexUtils.dailyData(jsonMap); + jsonMap.put("compositeIndex", compositeIndex); + cityAqiDaily.setValue(JSONObject.toJSONString(jsonMap)); - cityAqiDailyMapper.insert(cityAqiDaily); + cityAqiDailies.add(cityAqiDaily); }); + cityAqiDailyMapper.insertCityAqiDaily(cityAqiDailies); } } -- Gitblit v1.8.0