From 01ebd6b361611c0a7e8ed80fe5aed79a88b2da6d Mon Sep 17 00:00:00 2001
From: CMM <2198256324@qq.com>
Date: Sun, 1 Sep 2024 17:51:30 +0800
Subject: [PATCH] =?UTF-8?q?=E9=A2=84=E8=AD=A6=E8=A7=84=E5=88=99=E4=BF=9D?=
 =?UTF-8?q?=E5=AD=98=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/hz/pm/api/sys/controller/EarlyWarningController.java    | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/hz-pm-api/src/main/java/com/hz/pm/api/sys/controller/EarlyWarningController.java b/hz-pm-api/src/main/java/com/hz/pm/api/sys/controller/EarlyWarningController.java
index b3b62d3..111a918 100644
--- a/hz-pm-api/src/main/java/com/hz/pm/api/sys/controller/EarlyWarningController.java
+++ b/hz-pm-api/src/main/java/com/hz/pm/api/sys/controller/EarlyWarningController.java
@@ -1,5 +1,6 @@
 package com.hz.pm.api.sys.controller;
 
+import com.hz.pm.api.user.helper.MhUnitCache;
 import com.ningdatech.basic.model.PageVo;
 import com.ningdatech.log.annotation.WebLog;
 import com.hz.pm.api.sys.model.req.WarningListReq;
@@ -34,6 +35,7 @@ public class EarlyWarningController {
     private final IEarlyWarningService earlyWarningService;
 
     private final IEarlyWarningRecordsService earlyWarningRecordsService;
+    private final MhUnitCache mhUnitCache;
 
     @ApiOperation(value = "预警规则获取", notes = "预警规则获取")
     @GetMapping("/detail/{mhUnitId}")
@@ -45,6 +47,9 @@ public class EarlyWarningController {
     @PostMapping("/save")
     @WebLog("预警规则保存")
     public String save(@Validated @RequestBody WflowEarlyWarningDTO dto) {
+        Long mhUnitId = dto.getMhUnitId();
+        List<Long> viewChildIds = mhUnitCache.getViewChildIdsRecursion(mhUnitId);
+        dto.setViewChildUnitIds(viewChildIds);
         return earlyWarningService.saveByMhUnit(dto);
     }