package cn.gistack.nky.service;
|
|
import cn.gistack.nky.requestpojo.ArimaPredictReqPo;
|
import cn.gistack.nky.requestpojo.HstPredictReqPo;
|
import cn.gistack.nky.requestpojo.QueryReqPo;
|
import com.alibaba.fastjson.JSON;
|
import org.springframework.http.*;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.client.RestTemplate;
|
|
|
/**
|
* @PROJECT_NAME: skjcmanager
|
* @DESCRIPTION:
|
* @USER: aix
|
* @DATE: 2023/7/31 14:19
|
*/
|
@Service
|
public class NkyServiceImpl implements INkyService {
|
|
// ARIMA预测接口URL
|
private static final String ARIMA_PREDICT = "https://sk.hubeishuiyi.cn/warnIp/predict/arimaPredict";
|
// ARIMA预测结果接口
|
private static final String ARIMA_QUERY = "https://sk.hubeishuiyi.cn/warnIp/predict/arimaQuery";
|
// HST预测接口URL
|
private static final String HST_PREDICT = "https://sk.hubeishuiyi.cn/warnIp/predict/hstPredict";
|
// HST预测结果接口
|
private static final String HST_QUERY = "https://sk.hubeishuiyi.cn/warnIp/predict/hstQuery";
|
|
@Override
|
public void arimaPredict(ArimaPredictReqPo po) {
|
setPredict(ARIMA_PREDICT, JSON.toJSONString(po));
|
}
|
|
@Override
|
public void hstPredict(HstPredictReqPo po) {
|
setPredict(HST_PREDICT, JSON.toJSONString(po));
|
}
|
|
@Override
|
public Object arimaQuery(QueryReqPo po) {
|
return null;
|
}
|
|
@Override
|
public Object hstQuery(QueryReqPo po) {
|
return null;
|
}
|
|
private void setPredict(String url, String jsonparams) {
|
// 声明一个header变量
|
HttpHeaders headers = new HttpHeaders();
|
// 设置为json格式
|
MediaType mediaType = MediaType.parseMediaType("application/json;charset=utf-8");
|
headers.setContentType(mediaType);
|
headers.add("Accept", MediaType.APPLICATION_JSON.toString());
|
|
HttpEntity<String> httpEntity = new HttpEntity(jsonparams, headers);
|
|
RestTemplate template = new RestTemplate();
|
String resultStr = template.postForObject(url, httpEntity, String.class);
|
System.out.println(resultStr);
|
|
}
|
|
}
|