洪水预报接口对接

guoshilong

2024-03-05 2f8eaa7c850445fde0cdca0e8ecddcb7e2a936a3

 skjcmanager/skjcmanager-service/skjcmanager-nky/src/main/java/cn/gistack/nky/service/impl/HsybServiceImpl.java

@@ -1,19 +1,25 @@
package cn.gistack.nky.service.impl;

import cn.gistack.common.utils.CommonUtil;
import cn.gistack.common.utils.HttpClientUtils;
import cn.gistack.common.utils.SpringContextUtil;
import cn.gistack.nky.resultpojo.*;
import cn.gistack.nky.service.IHsybService;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.parser.Feature;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springblade.core.tool.utils.DateUtil;
import org.springblade.core.tool.utils.StringUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.*;
import org.springframework.stereotype.Service;
import org.springframework.util.MultiValueMap;
import org.springframework.web.client.RestTemplate;

import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.*;
import java.util.stream.Collectors;

@@ -22,12 +28,22 @@
@Slf4j
public class HsybServiceImpl implements IHsybService {

   private RestTemplate restTemplate;
   private static String PREFIX = "/hsybApi";

   private static String ONLINE = "http://10.42.7.148:50001";
   private static String LOCAL = "https://sk.hubeishuiyi.cn";
   private static String GET_FUTURE = "/hsybApi/api/fh-admin/skkr/getFuture";
   private static String GET_NEW_FUTURE = "/hsybApi/api/fh-admin/skkr/getSkFutureSw";
   private static String GET_FUTURE_ONLINE = "/api/fh-admin/skkr/getFuture";
   private static String GET_NEW_FUTURE_ONLINE = "/api/fh-admin/skkr/getSkFutureSw";
   private static String GET_FUTURE = "/api/fh-admin/skkr/getFuture";
   //新水位接口
   private static String GET_NEW_FUTURE = "/api/fh-admin/skkr/getSkFutureSw";
   //获取是否有预测模型接口
   private static String GET_SK_GXZT = "/api/fh-admin/skkr/getSkgxzt";
   //同步模型接口
   private static String UPDATE_CJYSJ_BY_TM = "/api/fh-admin/skkr/updateCjysjByTm";


//   private static String GET_FUTURE_ONLINE = "/api/fh-admin/skkr/getFuture";
//   private static String GET_NEW_FUTURE_ONLINE = "/api/fh-admin/skkr/getSkFutureSw";

   private static String AUTHORIZATION = "Bearer eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX25hbWUiOiJBRE1JTiIsIlVzZXJJZCI6IjEiLCJzY29wZSI6WyJhbGwiXSwiVXNlclJlYWxOYW1lIjoi6LaF57qn566h55CG5ZGYIiwiVXNlclh6cWhkbSI6IjQyMDUiLCJleHAiOjI1MTk3NzIwMTAsImp0aSI6ImExNTcxYzk1LTBkNjMtNDkyYi1iOWEyLTE2ZTIzNTQ5ZTY1ZiIsImNsaWVudF9pZCI6InVzZXItc2VydmljZSJ9.ebNVZrw9LbhKaj2w6RR8b2wccQiDkhvBeq79SxxCK-yWiOlIFqBkotTN4TNJg8umcpyYvLILwvqXWRJhffEtgi25sX2y6MqLIWM4kMZ9d8ptdnSmTpBPhltSiQOM0KFa1kl5nSDCBwYOLn-pESJglam76cjpgZNoC88x3iNHacdiXDItY0rtY85HrQ26uyJu9UovKtmYmZRHsIbGMpDta5Q1p4vfaCIr-YUayDrCweZJiQDEEcOSpWJ7O7RMk3pRkX_4UmPHFzrOI2lMp1jQIhxnSTE7EVAz_4z8h8r46muSkpF54Ic4XSawHKqdSLHx8T05LB0MpvOzWMPS6c1uHA";

@@ -37,7 +53,7 @@
      HsybGetFuturePo hsybGetFuturePo = apiRequest(GET_FUTURE, resCd);

      if (hsybGetFuturePo != null && hsybGetFuturePo.getRespCode().equals("200")) {
         List<List<String>> data = hsybGetFuturePo.getData();
         List<List<String>> data = (List<List<String>>)hsybGetFuturePo.getData();

         List<List<String>> collect = filterPredictData(data);
         return collect;
@@ -48,29 +64,68 @@

   @Override
   public List<NewSwFuturePo> getNewFuture(String resId) {
      Map<String, Object> params = new HashMap<>();
      params.put("resId", resId);
      HsybGetFuturePo get = apiRequest(GET_NEW_FUTURE, "GET", params);
      //获取预测水位之前需要确认是否有预测水位模型
//      Boolean isUpdate =  isResHasModel(resId);

      //程序转为hashmap,手动转换类型
      List<NewSwFuturePo> data =JSON.parseArray(JSON.toJSONString(get.getData()), NewSwFuturePo.class);
      List<NewSwFuturePo> filterData = filterPredict(data);
//      if (isUpdate){
         //数据模型已经更新过，就可以去拿预测水位
         //设置预测73小时未来水位,因为最后一条是当前时间的整点数据，获取不到三天，多加一小时就可以获取到下一个整点的未来数据
         String urlParams = StringUtil.format("?resId={}&yjq=73",resId);

      log.info(StringUtil.format("过滤后的预测水位数据：{}", JSON.toJSONString(filterData)));
      return filterData;
         HsybGetFuturePo hsybGetFuturePo = sendRequestToHsyb(GET_NEW_FUTURE, urlParams);

         //程序转为hashmap,手动转换类型
         List<NewSwFuturePo> data = JSON.parseArray(JSON.toJSONString(hsybGetFuturePo.getData()), NewSwFuturePo.class);
         List<NewSwFuturePo> filterData = filterPredict(data);

         log.info(StringUtil.format("过滤后的预测水位数据：{}", JSON.toJSONString(filterData)));
         return filterData;
//      }else{
//         //拿不到预测水位
//         return null;
//      }
   }

   public Boolean isResHasModel(String resId){
      String urlParams = StringUtil.format("?skbm={}",resId);
      HsybGetFuturePo hsybGetFuturePo = sendRequestToHsyb(GET_SK_GXZT, urlParams);
      String message = hsybGetFuturePo.getData().toString();

      if (message.equals("请更新模型数据！")){
         return false;
      }else{
         return true;
      }
   }

   @Override
   public String updateCjysjByTm(List<String> resIds) {
      // 获取当前时间
      LocalDate currentDate = LocalDate.now();
      // 获取前30天的时间
      LocalDate thirtyDaysBefore = currentDate.minusDays(30);
      // 输出结果
      DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd 00:00:00");
      String startTime = thirtyDaysBefore.format(formatter);
      String endTime = currentDate.format(formatter);

      String urlParams = StringUtil.format("?startTime={}&endTime={}", startTime, endTime);
      String skbmListStr = String.join("&skbmList=", resIds);
      HsybGetFuturePo res = sendRequestToHsyb(UPDATE_CJYSJ_BY_TM, urlParams + "&skbmList=" + skbmListStr);
      return res.getData().toString();
   }

   private HsybGetFuturePo apiRequest(String url, String type, Map<String, Object> params) {
      // 获取环境
      String activeProfile = SpringContextUtil.getActiveProfile();
      if (activeProfile.equals("dev")) {
         url = LOCAL + url;
         url = LOCAL + PREFIX + url;
      }
      if (activeProfile.equals("prod")) {
         url = ONLINE + GET_NEW_FUTURE_ONLINE;
         url = ONLINE + url;
      }
      if (activeProfile.equals("test")) {
         url = ONLINE + GET_NEW_FUTURE_ONLINE;
         url = ONLINE + url;
      }

      HttpMethod method;
@@ -99,13 +154,13 @@
      // 获取环境
      String activeProfile = SpringContextUtil.getActiveProfile();
      if (activeProfile.equals("dev")) {
         url = LOCAL + url;
         url = LOCAL + PREFIX + url;
      }
      if (activeProfile.equals("prod")) {
         url = ONLINE + GET_FUTURE_ONLINE;
         url = ONLINE + GET_FUTURE;
      }
      if (activeProfile.equals("test")) {
         url = ONLINE + GET_FUTURE_ONLINE;
         url = ONLINE + GET_FUTURE;
      }


@@ -131,6 +186,35 @@
         System.out.println(e);
      }

      return null;
   }

   public HsybGetFuturePo sendRequestToHsyb(String url, String urlParam) {
      // 获取环境
      String activeProfile = SpringContextUtil.getActiveProfile();
      if (activeProfile.equals("dev")) {
         url = LOCAL + PREFIX + url;
      }
      if (activeProfile.equals("prod")) {
         url = ONLINE + url;
      }
      if (activeProfile.equals("test")) {
         url = ONLINE + url;
      }
      url = url + urlParam;
      log.info("洪水预报请求地址：{}", url);
      //设置请求头
      HttpHeaders headers = new HttpHeaders();
      headers.add("Authorization", AUTHORIZATION);
      //封装请求头
      HttpEntity<MultiValueMap<String, Object>> formEntity = new HttpEntity<MultiValueMap<String, Object>>(headers);

      try {
         ResponseEntity<HsybGetFuturePo> exchange = restTemplate.exchange(url, HttpMethod.GET, formEntity, HsybGetFuturePo.class);
         return exchange.getBody();
      } catch (Exception e) {
         e.printStackTrace();
      }
      return null;
   }

@@ -215,7 +299,7 @@
//      return distinctList;
   }

   public List<NewSwFuturePo> filterPredict(List<NewSwFuturePo> list){
   public List<NewSwFuturePo> filterPredict(List<NewSwFuturePo> list) {

      /**
       * {
@@ -229,7 +313,7 @@
       *             "resId": "42092250024"
       *         },
       */
      if (list.size() == 0){
      if (list.size() == 0) {
         return null;
      }

@@ -240,7 +324,7 @@
      List<NewSwFuturePo> filterList = list.stream().filter(item -> item.getTime().indexOf(newSwFuturePo.getTime().split(" ")[1]) > -1).filter(item->!item.getSw().equals("0.0")).collect(Collectors.toList());

      //给时间重新赋值，因为南科院hst预测只需要日期
      filterList.forEach(e->{
      filterList.forEach(e -> {
         e.setTime(e.getTime().split(" ")[0]);
      });