From 20f693a9af9e90931ecb004700ec96453d05cecb Mon Sep 17 00:00:00 2001 From: cjl <276999030@qq.com> Date: Thu, 27 Jul 2023 10:31:37 +0800 Subject: [PATCH] fix: 远程数据提交修改 --- screen-job/src/main/java/com/moral/api/service/impl/CityAqiDailyServiceImpl.java | 60 +++++++++++++++++++++++++++++++++++------------------------- 1 files changed, 35 insertions(+), 25 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 92b325d..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,6 +8,9 @@ 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; @@ -45,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(); //��������������������� @@ -55,43 +59,35 @@ //������������������aqi������������ QueryWrapper<CityAqi> wrapper = new QueryWrapper<>(); wrapper.select("city_code", "time", "value") - .ge("time", DateUtils.dateToDateString(start)) - .le("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); - - //��������������������������������������������������� - 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); + //O3��������������������������������������� + if (!ObjectUtils.isEmpty(value)) { + 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 -> { + 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; @@ -102,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); @@ -111,12 +107,26 @@ } }); + //���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