From 00165e5d50f47c99d9eb7c2fc784bef4d3f72aca Mon Sep 17 00:00:00 2001
From: guoshilong <123456>
Date: Thu, 16 Nov 2023 15:05:24 +0800
Subject: [PATCH] 预测返回测站名称

---
 skjcmanager/skjcmanager-service/skjcmanager-nky/src/main/java/cn/gistack/nky/controller/ArimaPredictController.java |   33 +++++++++++++++++++++++++++++++++
 1 files changed, 33 insertions(+), 0 deletions(-)

diff --git a/skjcmanager/skjcmanager-service/skjcmanager-nky/src/main/java/cn/gistack/nky/controller/ArimaPredictController.java b/skjcmanager/skjcmanager-service/skjcmanager-nky/src/main/java/cn/gistack/nky/controller/ArimaPredictController.java
index f81d75b..223933e 100644
--- a/skjcmanager/skjcmanager-service/skjcmanager-nky/src/main/java/cn/gistack/nky/controller/ArimaPredictController.java
+++ b/skjcmanager/skjcmanager-service/skjcmanager-nky/src/main/java/cn/gistack/nky/controller/ArimaPredictController.java
@@ -1,10 +1,20 @@
 package cn.gistack.nky.controller;
 
+import cn.gistack.nky.entity.ArimaPredict;
+import cn.gistack.nky.entity.HstPredict;
+import cn.gistack.nky.service.IArimaPredictService;
+import cn.gistack.nky.service.IHstPredictService;
+import cn.gistack.nky.vo.ArimaPredictVO;
+import cn.gistack.nky.vo.HstPredictVO;
 import lombok.AllArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springblade.core.boot.ctrl.BladeController;
+import org.springblade.core.tool.api.R;
+import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
+
+import java.util.List;
 
 @Slf4j
 @RestController
@@ -12,4 +22,27 @@
 @RequestMapping("/arimaPredict/arimaPredict")
 public class ArimaPredictController extends BladeController {
 
+	private final IArimaPredictService arimaPredictService;
+	private final IHstPredictService hstPredictService;
+
+	/**
+	 * 获取预测数据
+	 * @param resCd 水库id
+	 * @param type 1:渗压,2:渗流,3:位移
+	 * @param predictFlag 预测方式
+	 * @return
+	 */
+	@GetMapping("/futureData")
+	public R futureData(String resCd,String type,String predictFlag){
+		if (predictFlag.equals("Arima")){
+			List<ArimaPredictVO> list =  arimaPredictService.getFutureData(resCd,type);
+			return R.data(list);
+		}else if (predictFlag.equals("Hst")){
+			List<HstPredictVO> list = hstPredictService.getFutureData(resCd,type);
+			return R.data(list);
+		}else {
+			return R.fail("无该预测方式");
+		}
+	}
+
 }

--
Gitblit v1.9.3