From 5912494a3298c006e47c0d81c8feb59ea978d43c Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期日, 07 七月 2024 22:14:15 +0800
Subject: [PATCH] 部门管理树状:分页

---
 src/main/java/com/mindskip/xzs/service/impl/DepartmentServiceImpl.java |   67 ++++++++++++++++++++++++++++++---
 1 files changed, 61 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/mindskip/xzs/service/impl/DepartmentServiceImpl.java b/src/main/java/com/mindskip/xzs/service/impl/DepartmentServiceImpl.java
index 503fd0b..c548f23 100644
--- a/src/main/java/com/mindskip/xzs/service/impl/DepartmentServiceImpl.java
+++ b/src/main/java/com/mindskip/xzs/service/impl/DepartmentServiceImpl.java
@@ -5,6 +5,7 @@
 import com.mindskip.xzs.domain.Department;
 import com.mindskip.xzs.domain.enums.DeptAdminEnum;
 import com.mindskip.xzs.domain.vo.BaseSelect;
+import com.mindskip.xzs.domain.vo.CascaderDataVO;
 import com.mindskip.xzs.domain.vo.UpdateDeptAdminVO;
 import com.mindskip.xzs.repository.BaseMapper;
 import com.mindskip.xzs.repository.DepartmentMapper;
@@ -18,8 +19,7 @@
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.CollectionUtils;
 
-import java.util.List;
-import java.util.Objects;
+import java.util.*;
 import java.util.stream.Collectors;
 
 @Service
@@ -77,10 +77,33 @@
     }
 
     @Override
-    public PageInfo<DepartmentResponseVM> gets(DepartmentResponseVM departmentResponseVM) {
-        return PageHelper.startPage(departmentResponseVM.getPageIndex(), departmentResponseVM.getPageSize(), "id desc").doSelectPageInfo(() ->
-                departmentMapper.page(departmentResponseVM)
-        );
+    public List<DepartmentResponseVM> gets(DepartmentResponseVM departmentResponseVM) {
+        List<DepartmentResponseVM> deptList = departmentMapper.page(departmentResponseVM);
+        return this.buildPageTree(deptList);
+    }
+
+    public List<DepartmentResponseVM> buildPageTree(List<DepartmentResponseVM> departments) {
+        // 鐢ㄤ簬瀛樺偍宸茬粡鏀惧叆鏍戠殑鑺傜偣锛屼互value涓洪敭
+        Map<Integer, DepartmentResponseVM> map = new HashMap<>();
+        List<DepartmentResponseVM> rootNodes = new ArrayList<>();
+
+        // 绗竴姝ワ細灏嗘墍鏈夎妭鐐规斁鍏ap涓紝骞跺垵濮嬪寲children鍒楄〃
+        for (DepartmentResponseVM dept : departments) {
+            map.put(dept.getId(), dept);
+            dept.setChildren(new ArrayList<>());
+        }
+        // 绗簩姝ワ細鏋勫缓鏍戝舰缁撴瀯
+        for (DepartmentResponseVM dept : departments) {
+            if (dept.getParentId() == null) {
+                rootNodes.add(dept);
+            } else {
+                DepartmentResponseVM parent = map.get(dept.getParentId());
+                if (parent != null) {
+                    parent.getChildren().add(dept);
+                }
+            }
+        }
+        return rootNodes;
     }
 
     @Override
@@ -130,4 +153,36 @@
         // 淇敼娌¤閫夋嫨鐨勭敤鎴峰苟涓斾笉鏄埆鐨勯儴闂ㄧ鐞嗗憳鐨勭敤鎴疯鑹蹭负1
         userMapper.cancelUserDeptAdmin(form);
     }
+
+    @Override
+    public List<CascaderDataVO> cascader() {
+        List<CascaderDataVO> deptList = departmentMapper.list();
+        List<CascaderDataVO> tree = this.buildTree(deptList);
+        return tree;
+    }
+
+
+    public List<CascaderDataVO> buildTree(List<CascaderDataVO> departments) {
+        // 鐢ㄤ簬瀛樺偍宸茬粡鏀惧叆鏍戠殑鑺傜偣锛屼互value涓洪敭
+        Map<Integer, CascaderDataVO> map = new HashMap<>();
+        List<CascaderDataVO> rootNodes = new ArrayList<>();
+
+        // 绗竴姝ワ細灏嗘墍鏈夎妭鐐规斁鍏ap涓紝骞跺垵濮嬪寲children鍒楄〃
+        for (CascaderDataVO dept : departments) {
+            map.put(dept.getValue(), dept);
+            dept.setChildren(new ArrayList<>());
+        }
+        // 绗簩姝ワ細鏋勫缓鏍戝舰缁撴瀯
+        for (CascaderDataVO dept : departments) {
+            if (dept.getParentId() == null) { // 鍋囪椤剁骇閮ㄩ棬鐨刾arentId涓簄ull鎴�0
+                rootNodes.add(dept);
+            } else {
+                CascaderDataVO parent = map.get(dept.getParentId());
+                if (parent != null) {
+                    parent.getChildren().add(dept);
+                }
+            }
+        }
+        return rootNodes;
+    }
 }

--
Gitblit v1.8.0