From 695b18282b20c2bd954b3dfdcbe51bc28dc46f64 Mon Sep 17 00:00:00 2001
From: guoshilong <123456>
Date: Sat, 03 Jun 2023 16:13:00 +0800
Subject: [PATCH] 行政区划修改

---
 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztmd/service/impl/AttAdBaseServiceImpl.java |   63 +++++++++++++++++++++++++++++++
 1 files changed, 63 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..7df55ce 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,62 @@
 	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);
+	}
+
+	@Override
+	public AttAdBaseVO getDetail(AttAdBase attAdBase) {
+		return baseMapper.getDetail(attAdBase);
+	}
+
+	@Override
+	public boolean save(AttAdBase entity) {
+		return baseMapper.customizeSave(entity);
+	}
 }

--
Gitblit v1.9.3