From 733ca7545a278fe9ea6fa0be906cf28e3f036c60 Mon Sep 17 00:00:00 2001
From: guoshilong <123456>
Date: Mon, 29 Jan 2024 15:13:26 +0800
Subject: [PATCH] 南科院预测修改,大坝专题汛限水位返回

---
 skjcmanager/skjcmanager-service/skjcmanager-nky/src/main/java/cn/gistack/nky/service/impl/HstPredictServiceImpl.java |  182 ++++++++++++++++++++++++++++++++++-----------
 1 files changed, 137 insertions(+), 45 deletions(-)

diff --git a/skjcmanager/skjcmanager-service/skjcmanager-nky/src/main/java/cn/gistack/nky/service/impl/HstPredictServiceImpl.java b/skjcmanager/skjcmanager-service/skjcmanager-nky/src/main/java/cn/gistack/nky/service/impl/HstPredictServiceImpl.java
index 0e6ed16..5979ff0 100644
--- a/skjcmanager/skjcmanager-service/skjcmanager-nky/src/main/java/cn/gistack/nky/service/impl/HstPredictServiceImpl.java
+++ b/skjcmanager/skjcmanager-service/skjcmanager-nky/src/main/java/cn/gistack/nky/service/impl/HstPredictServiceImpl.java
@@ -49,7 +49,7 @@
 		});
 	}
 
-	private String resCd = "";
+//	private String resCd = "";
 
 	@Override
 	public void predictData(String type) {
@@ -57,15 +57,122 @@
 		//暂时筛选出这两个水库的数据
 		List<DataResChildrenPo> collect = cdList.stream().filter(item -> item.getRes_cd().equals("42011640018") || item.getRes_cd().equals("42128140006")).collect(Collectors.toList());
 
+		predict(collect,type);
 
-		if (type.equals("1")) {
-			//渗压
-			collect.forEach(dataResChildrenPo -> {
-				//所有水库请求预报数据
-//				if (!resCd.equals(dataResChildrenPo.getRes_cd())) {
-				//请求预报水位
-				List<List<String>> future = hsybService.getFuture(dataResChildrenPo.getRes_cd());
-				resCd = dataResChildrenPo.getRes_cd();
+
+//
+//		if (type.equals("1")) {
+//			//渗压
+//			collect.forEach(dataResChildrenPo -> {
+//				//所有水库请求预报数据
+////				if (!resCd.equals(dataResChildrenPo.getRes_cd())) {
+//				//请求预报水位
+//				List<List<String>> future = hsybService.getFuture(dataResChildrenPo.getRes_cd());
+//				resCd = dataResChildrenPo.getRes_cd();
+//				if (future != null && future.size() > 0) {
+//					List<PredictWatersPo> po = new ArrayList<>();
+//					future.forEach(item -> {
+//						PredictWatersPo predictWatersPo = new PredictWatersPo();
+//						//设置时间
+//						predictWatersPo.setTime(item.get(0));
+//						//设置预测水位
+//						predictWatersPo.setValue(Double.parseDouble(item.get(2)));
+//						po.add(predictWatersPo);
+//					});
+//					HstPredictReqPo arimaPredictReqPo = new HstPredictReqPo(dataResChildrenPo, type, po);
+//					//循环预测请求
+//					nkyService.hstPredict(arimaPredictReqPo);
+//				}
+////				}
+//			});
+//		} else if (type.equals("2")) {
+//			//渗流
+//			collect.forEach(dataResChildrenPo -> {
+//				//所有水库请求预报数据
+////				if (!resCd.equals(dataResChildrenPo.getRes_cd())) {
+//				//请求预报水位
+//				List<List<String>> future = hsybService.getFuture(dataResChildrenPo.getRes_cd());
+//				resCd = dataResChildrenPo.getRes_cd();
+//				if (future != null && future.size() > 0) {
+//					List<PredictWatersPo> po = new ArrayList<>();
+//					future.forEach(item -> {
+//						PredictWatersPo predictWatersPo = new PredictWatersPo();
+//						//设置时间
+//						predictWatersPo.setTime(item.get(0));
+//						//设置预测水位
+//						predictWatersPo.setValue(Double.parseDouble(item.get(2)));
+//						po.add(predictWatersPo);
+//					});
+//					HstPredictReqPo arimaPredictReqPo = new HstPredictReqPo(dataResChildrenPo, type, po);
+//					//循环预测请求
+//					nkyService.hstPredict(arimaPredictReqPo);
+//				}
+////				}
+//			});
+//
+//
+//		} else if (type.equals("3") || type.equals("4") || type.equals("5")) {
+//			collect.forEach(dataResChildrenPo -> {
+//				//所有水库请求预报数据
+////				if (!resCd.equals(dataResChildrenPo.getRes_cd())) {
+//				//请求预报水位
+//				List<List<String>> future = hsybService.getFuture(dataResChildrenPo.getRes_cd());
+//				resCd = dataResChildrenPo.getRes_cd();
+//				if (future != null && future.size() > 0) {
+//					List<PredictWatersPo> po = new ArrayList<>();
+//					future.forEach(item -> {
+//						PredictWatersPo predictWatersPo = new PredictWatersPo();
+//						//设置时间
+//						predictWatersPo.setTime(item.get(0));
+//						//设置预测水位
+//						predictWatersPo.setValue(Double.parseDouble(item.get(2)));
+//						po.add(predictWatersPo);
+//					});
+//					//循环预测请求(位移一起请求减少请求次数)
+//					HstPredictReqPo arimaPredictReqPoX = new HstPredictReqPo(dataResChildrenPo, "3", po);
+//					nkyService.hstPredict(arimaPredictReqPoX);
+//
+//					HstPredictReqPo arimaPredictReqPoY = new HstPredictReqPo(dataResChildrenPo, "4", po);
+//					nkyService.hstPredict(arimaPredictReqPoY);
+//
+//					HstPredictReqPo arimaPredictReqPoZ = new HstPredictReqPo(dataResChildrenPo, "5", po);
+//					nkyService.hstPredict(arimaPredictReqPoZ);
+//				}
+////				}
+//			});
+
+
+
+//		}
+//		resCd = "";
+	}
+
+	private void predict(List<DataResChildrenPo> collect, String type) {
+		String resCd = "";
+		//无数据直接返回
+		if (collect.size() == 0){
+			return;
+		}
+		List<List<String>> future = new ArrayList<>();
+		if (type.equals("1") || type.equals("2")){
+
+			for (int i = 0; i < collect.size(); i++) {
+				DataResChildrenPo dataResChildrenPo = collect.get(i);
+				String currentResCd = dataResChildrenPo.getRes_cd();
+
+				//第一个水库
+				if (future.size() == 0 && resCd.equals("")){
+					future = hsybService.getFuture(currentResCd);
+					//当前水库
+					resCd = currentResCd;
+				} else if (!resCd.equals(currentResCd)) {
+					//其他水库,获取新的预测数据
+					future = hsybService.getFuture(currentResCd);
+					//更换为新水库
+					resCd = currentResCd;
+				}
+
+				//预测数据构造请求数据
 				if (future != null && future.size() > 0) {
 					List<PredictWatersPo> po = new ArrayList<>();
 					future.forEach(item -> {
@@ -80,41 +187,28 @@
 					//循环预测请求
 					nkyService.hstPredict(arimaPredictReqPo);
 				}
-//				}
-			});
-		} else if (type.equals("2")) {
-			//渗流
-			collect.forEach(dataResChildrenPo -> {
-				//所有水库请求预报数据
-//				if (!resCd.equals(dataResChildrenPo.getRes_cd())) {
-				//请求预报水位
-				List<List<String>> future = hsybService.getFuture(dataResChildrenPo.getRes_cd());
-				resCd = dataResChildrenPo.getRes_cd();
-				if (future != null && future.size() > 0) {
-					List<PredictWatersPo> po = new ArrayList<>();
-					future.forEach(item -> {
-						PredictWatersPo predictWatersPo = new PredictWatersPo();
-						//设置时间
-						predictWatersPo.setTime(item.get(0));
-						//设置预测水位
-						predictWatersPo.setValue(Double.parseDouble(item.get(2)));
-						po.add(predictWatersPo);
-					});
-					HstPredictReqPo arimaPredictReqPo = new HstPredictReqPo(dataResChildrenPo, type, po);
-					//循环预测请求
-					nkyService.hstPredict(arimaPredictReqPo);
+
+
+			}
+		}else if (type.equals("3") || type.equals("4") || type.equals("5")){
+
+			for (int i = 0; i < collect.size(); i++) {
+				DataResChildrenPo dataResChildrenPo = collect.get(i);
+				String currentResCd = dataResChildrenPo.getRes_cd();
+
+				//第一个水库
+				if (future.size() == 0 && resCd.equals("")){
+					future = hsybService.getFuture(currentResCd);
+					//当前水库
+					resCd = currentResCd;
+				} else if (!resCd.equals(currentResCd)) {
+					//其他水库,获取新的预测数据
+					future = hsybService.getFuture(currentResCd);
+					//更换为新水库
+					resCd = currentResCd;
 				}
-//				}
-			});
 
-
-		} else if (type.equals("3") || type.equals("4") || type.equals("5")) {
-			collect.forEach(dataResChildrenPo -> {
-				//所有水库请求预报数据
-//				if (!resCd.equals(dataResChildrenPo.getRes_cd())) {
-				//请求预报水位
-				List<List<String>> future = hsybService.getFuture(dataResChildrenPo.getRes_cd());
-				resCd = dataResChildrenPo.getRes_cd();
+				//预测数据构造请求数据
 				if (future != null && future.size() > 0) {
 					List<PredictWatersPo> po = new ArrayList<>();
 					future.forEach(item -> {
@@ -135,10 +229,8 @@
 					HstPredictReqPo arimaPredictReqPoZ = new HstPredictReqPo(dataResChildrenPo, "5", po);
 					nkyService.hstPredict(arimaPredictReqPoZ);
 				}
-//				}
-			});
+			}
 		}
-		resCd = "";
 	}
 
 	@Override

--
Gitblit v1.9.3