From 3570bc18d5daed8079dd67891062ad8585a4e696 Mon Sep 17 00:00:00 2001
From: zhangk <zhangk@gisuni.com>
Date: Sat, 22 Jun 2024 19:20:44 +0800
Subject: [PATCH] 修改监测水库降雨模型

---
 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/word/service/impl/SjztmdServiceImpl.java |   31 ++++++++++++++++++++++++++-----
 1 files changed, 26 insertions(+), 5 deletions(-)

diff --git a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/word/service/impl/SjztmdServiceImpl.java b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/word/service/impl/SjztmdServiceImpl.java
index 8c2db7e..0e25681 100644
--- a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/word/service/impl/SjztmdServiceImpl.java
+++ b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/word/service/impl/SjztmdServiceImpl.java
@@ -2,6 +2,7 @@
 
 import cn.gistack.common.utils.SpringContextUtil;
 import cn.gistack.sm.sjztmd.entity.*;
+import cn.gistack.sm.sjztmd.vo.AttResWithRainfall;
 import cn.gistack.sm.sjztmd.vo.StationExport;
 import cn.gistack.sm.sjztmd.vo.ZtResultInfo;
 import cn.gistack.sm.sjztmd.vo.StationParams;
@@ -17,6 +18,7 @@
 import com.alibaba.fastjson.JSONObject;
 import com.alibaba.fastjson.parser.Feature;
 import com.baomidou.dynamic.datasource.annotation.DS;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import lombok.AllArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springblade.core.tool.utils.ObjectUtil;
@@ -42,14 +44,11 @@
 public class SjztmdServiceImpl implements ISjztmdService {
 
 	//总计
-//	private final String RIBAO_SHEET1_TOTAL_URL = "/services/1234567890ABCDEFGHIJKLMN/ribao_sheet1/api?small_start_stag=0&small_end_stag=2&mid_start_stag=0&mid_end_stag=2&big_start_stag=0&big_end_stag=2";
 	private final String RIBAO_SHEET1_TOTAL_URL = "/services/1234567890ABCDEFGHIJKLMN/ribao_sheet1/api?small_start_stag=0&mid_start_stag=0&big_start_stag=0";
 	private final String RIBAO_SHEET1_OVER_NORM_POOL_STAG = "/services/1234567890ABCDEFGHIJKLMN/ribao/sheet1_over_norm_pool_stag/api?";
 	//中大型水库
-//	private final String RIBAO_SHEET2_ZDX_URL = "/services/1234567890ABCDEFGHIJKLMN/ribao/sheet2_zdx/api?mid_start_stag=0&mid_end_stag=2&big_start_stag=0&big_end_stag=2";
 	private final String RIBAO_SHEET2_ZDX_URL = "/services/1234567890ABCDEFGHIJKLMN/ribao/sheet2_zdx/api?mid_start_stag=0&big_start_stag=0";
 	//超汛明细
-//	private final String RIBAO_SHEET3_OVER_STAG_LIST_URL = "/services/1234567890ABCDEFGHIJKLMN/ribao/sheet3_over_stag_list/api?small_start_stag=0&small_end_stag=2&mid_start_stag=0&mid_end_stag=2&big_start_stag=0&big_end_stag=2";
 	private final String RIBAO_SHEET3_OVER_STAG_LIST_URL = "/services/1234567890ABCDEFGHIJKLMN/ribao/sheet3_over_stag_list/api?small_start_stag=0&mid_start_stag=0&big_start_stag=0";
 	//省直管
 	private final String SZ_INFO = "/services/1234567890ABCDEFGHIJKLMN/ribao/shengzh_new_rz/api";
@@ -67,7 +66,7 @@
 	@Autowired
 	private RestTemplate restTemplate;
 	private final SjztMdMapper mapper;
-
+	private final IAttResBaseService attResBase;
 	private final ITbAttResBaseService attResBaseService;
 	private final ITbAttResStagCharService attResStagCharService;
 	private final ITbAttResWaterDeliveryService attResWaterDeliveryService;
@@ -78,7 +77,7 @@
 	private final ITbAttResWaterBlockService attResWaterBlockService;
 	private final ITbAttResRsbNorspiService attResRsbNorspiService;
 	private final ITbAttResPowerStationService attResPowerStationService;
-
+	private final IStResRainfallService stResRainfallService;
 	private final ITbAttResProjectUtilizeService attResProjectUtilizeService;
 	private final ITbAttResInformationProjectService attResInformationProjectService;
 
@@ -619,6 +618,28 @@
 
 	}
 
+	@Override
+	public void getReservoirRainfallForecast(List<String> dayIndexList) {
+		List<StResRainfall> stResRainfalls = new ArrayList<>();
+		for (String dayIndex : dayIndexList) {
+			List<AttResWithRainfall> attResByRainfall = attResBase.getAttResByRainfall(null, null, dayIndex);
+			for (AttResWithRainfall attResWithRainfall : attResByRainfall) {
+				StResRainfall stResRainfall = new StResRainfall();
+				stResRainfall.setId(UUID.randomUUID().toString().replaceAll("-", ""));
+				stResRainfall.setResCd(attResWithRainfall.getCode());
+				stResRainfall.setDrp(attResWithRainfall.getRainfall());
+				stResRainfall.setFileName(attResWithRainfall.getFileName());
+				stResRainfall.setDayIndex(dayIndex);
+				stResRainfall.setCreateTime(new Date());
+				stResRainfalls.add(stResRainfall);
+			}
+		}
+		if (!stResRainfalls.isEmpty()) {
+			stResRainfallService.remove(Wrappers.<StResRainfall>query().lambda());
+			stResRainfallService.saveBatch(stResRainfalls);
+		}
+	}
+
 
 	@Override
 	public List<DzkInfo> getDzkInfo(String isShow) {

--
Gitblit v1.9.3