From c9c6942fb773e1c33ba88a88e843d624218f87fb Mon Sep 17 00:00:00 2001
From: zhangk <zhangk@gisuni.com>
Date: Fri, 21 Jun 2024 16:19:34 +0800
Subject: [PATCH] 添加水库降雨量预报定时任务

---
 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/word/service/impl/SjztmdServiceImpl.java |   28 ++++++++++++++++++++++++++--
 1 files changed, 26 insertions(+), 2 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 f2260ba..9d31569 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;
@@ -64,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;
@@ -75,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;
 
@@ -616,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(System.currentTimeMillis());
+				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