From 7a1051f10a563edc2dab5d3180458439f6518801 Mon Sep 17 00:00:00 2001 From: jinpengyong <jpy123456> Date: Wed, 27 Oct 2021 16:26:07 +0800 Subject: [PATCH] 城市气象数据存入redis --- screen-job/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java | 66 +++++++++++++++++++++++++++++--- 1 files changed, 59 insertions(+), 7 deletions(-) diff --git a/screen-job/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java b/screen-job/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java index 1816336..40921a6 100644 --- a/screen-job/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java +++ b/screen-job/src/main/java/com/moral/api/service/impl/CityAqiServiceImpl.java @@ -4,21 +4,25 @@ 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.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.List; import java.util.Map; +import java.util.stream.Collectors; /** * <p> @@ -40,23 +44,71 @@ @Autowired private RestTemplate restTemplate; + @Autowired + private RedisTemplate redisTemplate; + @Override public void insertCityAqi() { //������������������ 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); + Date time = DateUtils.dataToTimeStampTime(new Date(), DateUtils.yyyy_MM_dd_HH_EN); 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"); + 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")); CityAqi cityAqi = new CityAqi(); cityAqi.setCityCode(cityCode); - 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, aqi); } } + + @Override + public void insertCityAqiDaily() { + 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", "value") + .ge("time", DateUtils.dateToDateString(start)) + .lt("time", DateUtils.dateToDateString(end)); + List<Map<String, Object>> dailyData = cityAqiMapper.selectMaps(wrapper); + + if (dailyData.size() == 0) { + return; + } + //���city_code������ + Map<String, List<Map<String, Object>>> data = dailyData.parallelStream().collect(Collectors.groupingBy(o -> (String) o.get("city_code"))); + data.forEach((cityCode, value) -> { + }); + + } } -- Gitblit v1.8.0