| | |
|---|
| | | 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.CityWeather; |
|---|
| | | 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.constant.Constants; |
|---|
| | | import com.moral.constant.RedisConstants; |
|---|
| | | 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.data.redis.core.RedisTemplate; |
|---|
| | | import org.springframework.stereotype.Service; |
|---|
| | | import org.springframework.util.ObjectUtils; |
|---|
| | | import org.springframework.web.client.RestTemplate; |
|---|
| | | |
|---|
| | | import java.util.Date; |
|---|
| | | import java.util.HashMap; |
|---|
| | | import java.util.List; |
|---|
| | | import java.util.Map; |
|---|
| | | import java.util.OptionalDouble; |
|---|
| | | import java.util.stream.DoubleStream; |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * <p> |
|---|
| | |
|---|
| | | @Autowired |
|---|
| | | private RestTemplate restTemplate; |
|---|
| | | |
|---|
| | | @Autowired |
|---|
| | | private RedisTemplate redisTemplate; |
|---|
| | | |
|---|
| | | @Autowired |
|---|
| | | |
|---|
| | | |
|---|
| | | @Override |
|---|
| | | public void insertCityAqi() { |
|---|
| | | Date now = new Date(); |
|---|
| | | Date time = DateUtils.dataToTimeStampTime(now, DateUtils.yyyy_MM_dd_HH_EN); |
|---|
| | | String format = DateUtils.yyyy_MM_dd_EN; |
|---|
| | | |
|---|
| | | Date start = null; |
|---|
| | | if (DateUtils.getHour(time) == 0 || DateUtils.getHour(time) >= 8) { |
|---|
| | | start = DateUtils.addHours(time, -7); |
|---|
| | | } |
|---|
| | | |
|---|
| | | //获取城市配置 |
|---|
| | | QueryWrapper<CityConfig> wrapper = new QueryWrapper<>(); |
|---|
| | | wrapper.select("city_code", "location_id").eq("is_delete", Constants.NOT_DELETE); |
|---|
| | | wrapper.select("city_code", "city_id").eq("is_delete", Constants.NOT_DELETE); |
|---|
| | | List<CityConfig> list = cityConfigService.list(wrapper); |
|---|
| | | |
|---|
| | | for (CityConfig cityConfig : list) { |
|---|
| | | String cityCode = cityConfig.getCityCode(); |
|---|
| | | Integer locationId = cityConfig.getLocationId(); |
|---|
| | | Map<String, Object> data = restTemplate.getForObject("https://api.qweather.com/v7/air/now?key=da05c6c4852d4f7aa3364a9236ee9e26&gzip=n&location={1}", Map.class, locationId); |
|---|
| | | Map<String, Object> now = (Map<String, Object>) data.get("now"); |
|---|
| | | Integer cityCode = cityConfig.getCityCode(); |
|---|
| | | Map<String, Object> data; |
|---|
| | | try { |
|---|
| | | //从第三方接口获取数据 |
|---|
| | | data = restTemplate.getForObject("http://sapi.7drlb.com/api/mj?cityID={1}&apiKey=aqi", Map.class, cityConfig.getCityId()); |
|---|
| | | } catch (Exception e) { |
|---|
| | | continue; |
|---|
| | | } |
|---|
| | | if (ObjectUtils.isEmpty(data)) { |
|---|
| | | continue; |
|---|
| | | } |
|---|
| | | |
|---|
| | | Map<String, Object> map = (Map<String, Object>) data.get("data"); |
|---|
| | | Map<String, Object> aqi = (Map<String, Object>) map.get("aqi"); |
|---|
| | | aqi.put("PM2_5", aqi.remove("pm25C")); |
|---|
| | | aqi.remove("pm25"); |
|---|
| | | aqi.put("PM10", aqi.remove("pm10C")); |
|---|
| | | aqi.put("SO2", aqi.remove("so2C")); |
|---|
| | | aqi.put("NO2", aqi.remove("no2C")); |
|---|
| | | aqi.put("CO", aqi.remove("coC")); |
|---|
| | | aqi.put("O3", aqi.remove("o3C")); |
|---|
| | | 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); |
|---|
| | | Map<String, Object> params = new HashMap<>(); |
|---|
| | | params.put("time", DateUtils.dateToDateString(time,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 -> { |
|---|
| | | Map<String, Object> dataValue = JSONObject.parseObject((String) v.get("value"), Map.class); |
|---|
| | | double o3 = Double.parseDouble(dataValue.get("o3").toString()); |
|---|
| | | return DoubleStream.of(o3); |
|---|
| | | }).average(); |
|---|
| | | if (average.isPresent()) { |
|---|
| | | double o3_8h = AmendUtils.sciCal(average.getAsDouble(), 0); |
|---|
| | | aqi.put("o3_8h", o3_8h); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | |
|---|
| | | //综合指数计算 |
|---|
| | | Map<String, Object> temp = new HashMap<>(aqi); |
|---|
| | | temp.put("o3", temp.get("o3_8h")); |
|---|
| | | Double compositeIndex = ComprehensiveIndexUtils.dailyData(temp); |
|---|
| | | aqi.put("compositeIndex", compositeIndex); |
|---|
| | | |
|---|
| | | //首要污染物计算 |
|---|
| | | Map<String,Object> sixParamMap = new HashMap<>(); |
|---|
| | | sixParamMap.put("a34002",aqi.get("PM10")); |
|---|
| | | sixParamMap.put("a34004",aqi.get("PM2_5")); |
|---|
| | | sixParamMap.put("a21004",aqi.get("NO2")); |
|---|
| | | sixParamMap.put("a21005",aqi.get("CO")); |
|---|
| | | sixParamMap.put("a05024",aqi.get("O3")); |
|---|
| | | sixParamMap.put("a21026",aqi.get("SO2")); |
|---|
| | | AQI aqiAndPollutant = AQIUtils.hourlyAQI(sixParamMap); |
|---|
| | | if (!ObjectUtils.isEmpty(aqiAndPollutant.getPrimaryPollutantNames())) { |
|---|
| | | aqi.put("pollutant",aqiAndPollutant.getPrimaryPollutantNames()); |
|---|
| | | } |
|---|
| | | |
|---|
| | | CityAqi cityAqi = new CityAqi(); |
|---|
| | | cityAqi.setCityCode(cityCode); |
|---|
| | | now.put("pm2_5", now.get("pm2p5")); |
|---|
| | | cityAqi.setValue(JSONObject.toJSONString(now)); |
|---|
| | | Date time = DateUtils.dataToTimeStampTime(new Date(), DateUtils.yyyy_MM_dd_HH_EN); |
|---|
| | | cityAqi.setTime(time); |
|---|
| | | cityAqi.setValue(JSONObject.toJSONString(aqi)); |
|---|
| | | cityAqiMapper.insert(cityAqi); |
|---|
| | | //存入redis |
|---|
| | | redisTemplate.opsForHash().put(RedisConstants.CITY_AQI, cityCode.toString(), aqi); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
