From ebf10ccf6217dfeebad750b4f4bb9ef4d9af68b0 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Fri, 02 Jun 2023 16:14:52 +0800
Subject: [PATCH] 新增告警模块,告警相关(告警信息,短信相关)代码从 sm 模块抽离

---
 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/AttAdBaseServiceImpl.java |   53 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 53 insertions(+), 0 deletions(-)

diff --git a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/AttAdBaseServiceImpl.java b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/AttAdBaseServiceImpl.java
index 7456274..633c1e2 100644
--- a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/AttAdBaseServiceImpl.java
+++ b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/AttAdBaseServiceImpl.java
@@ -1,5 +1,8 @@
 package cn.gistack.sm.sjztmd.service.impl;
 
+import cn.gistack.common.node.TreeNode;
+import cn.gistack.common.utils.NodeTreeUtil;
+import cn.gistack.sm.sjztmd.vo.PersonVO;
 import cn.gistack.sm.sjztmd.entity.AttAdBase;
 import cn.gistack.sm.sjztmd.mapper.AttAdBaseMapper;
 import cn.gistack.sm.sjztmd.service.IAttAdBaseService;
@@ -8,8 +11,10 @@
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.springframework.stereotype.Service;
 
+import java.util.Comparator;
 import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
 
 /**
  * @ClassName AttAdBaseServiceImpl
@@ -25,4 +30,52 @@
 	public List<AttAdBaseVO> lazyTree(String parentCode, Map<String, Object> param) {
 		return baseMapper.lazyTree(parentCode, param);
 	}
+
+	@Override
+	public List<AttAdBaseVO> lazyTreeAll(String parentCode, Map<String, Object> param) {
+		return baseMapper.lazyTreeAll(parentCode, param);
+	}
+
+	/**
+	 * 查询区域菜单树-带层级
+	 * @param parentCode
+	 * @param obj
+	 * @return
+	 */
+	@Override
+	public Object selectAttAdBaseTree(String parentCode, Map<String, Object> obj) {
+		List<TreeNode> nodeTree = NodeTreeUtil.getNodeTree(baseMapper.selectAttAdBaseTree(parentCode, obj));
+		// 排序并返回
+		return sortTree(nodeTree);
+	}
+
+	/**
+	 * 排序--只排第一,二个子集
+	 * @param nodeTree
+	 */
+	private List<TreeNode> sortTree(List<TreeNode> nodeTree) {
+		for (TreeNode node : nodeTree) {
+			if (node.getHasChildren()){
+				List<TreeNode> children = node.getChildren();
+				// 遍历子节点
+				for (TreeNode child : children) {
+					if (child.getHasChildren()){
+						child.setChildren(child.getChildren().stream().sorted(Comparator.comparing(TreeNode::getId)).collect(Collectors.toList()));
+					}
+				}
+				node.setChildren(node.getChildren().stream().sorted(Comparator.comparing(TreeNode::getId)).collect(Collectors.toList()));
+			}
+		}
+		return nodeTree;
+	}
+
+	/**
+	 * 根据水库编号查询除责任人之外的人员信息
+	 * @param res_cd
+	 * @return
+	 */
+	@Override
+	public List<PersonVO> getPersonListByResGuid(String res_cd) {
+		return baseMapper.getPersonListByResGuid(res_cd);
+	}
 }

--
Gitblit v1.9.3