From c5c10c1bb95f5bb7fdc8b714c5c998e78c26e0f7 Mon Sep 17 00:00:00 2001
From: 龚焕茏 <2842157468@qq.com>
Date: 星期二, 09 七月 2024 09:47:43 +0800
Subject: [PATCH] feat:标签新增、下拉增加部门id

---
 src/main/java/com/mindskip/xzs/service/impl/DepartmentServiceImpl.java |  127 +++++++++++++++++++++++++++++------------
 1 files changed, 89 insertions(+), 38 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 d138d98..6f296cb 100644
--- a/src/main/java/com/mindskip/xzs/service/impl/DepartmentServiceImpl.java
+++ b/src/main/java/com/mindskip/xzs/service/impl/DepartmentServiceImpl.java
@@ -1,12 +1,9 @@
 package com.mindskip.xzs.service.impl;
 
-import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
-import com.github.pagehelper.PageHelper;
-import com.github.pagehelper.PageInfo;
 import com.mindskip.xzs.domain.Department;
-import com.mindskip.xzs.domain.UserDepartment;
 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;
@@ -14,13 +11,15 @@
 import com.mindskip.xzs.repository.UserMapper;
 import com.mindskip.xzs.service.DepartmentService;
 import com.mindskip.xzs.utility.ModelMapperSingle;
+import com.mindskip.xzs.utility.TreeUtils;
 import com.mindskip.xzs.viewmodel.admin.department.DepartmentResponseVM;
 import org.modelmapper.ModelMapper;
 import org.springframework.stereotype.Service;
 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
 public class DepartmentServiceImpl extends BaseServiceImpl<Department> implements DepartmentService {
@@ -39,11 +38,9 @@
     }
 
     @Override
-    public Integer add(String name) {
-        Department department = new Department();
-        department.setDeleted("0");
-        department.setName(name);
-        return departmentMapper.add(department);
+    public Integer add(Department model) {
+        model.setDeleted("0");
+        return departmentMapper.add(model);
     }
 
     @Override
@@ -62,27 +59,48 @@
         if (Objects.isNull(entity)) {
             throw new RuntimeException("璇ラ儴闂ㄤ笉瀛樺湪");
         }
-        if (Objects.nonNull(department.getAdminId())) {
-            if (! Objects.equals(entity.getAdminId(), department.getAdminId())) {
-                // 鍙栨秷鍘熷厛鐨勭鐞嗗憳鏍囪瘑
-                List<Integer> depIds = departmentMapper.selectByAdminId(entity.getAdminId());
-                depIds.remove(department.getId());
-                if (depIds.size() < 1) {
-                    userMapper.updateDeptAdmin(entity.getAdminId(), DeptAdminEnum.NO.getValue());
-                }
+        if (! CollectionUtils.isEmpty(department.getAdminId())) {
+            List<BaseSelect> deptUserList = userDepartmentMapper.getDeptUserList(entity.getId());
+            List<Integer> userIds = deptUserList.stream().map(BaseSelect::getId).collect(Collectors.toList());
+            if (! CollectionUtils.isEmpty(userIds)) {
+                // 鍏堟竻闄や箣鍓嶇殑绠$悊鍛樻爣璇�
+                userMapper.clearDeptAdmin(userIds, entity.getId());
             }
             // 鏍囪瘑閮ㄩ棬绠$悊鍛�
-            userMapper.updateDeptAdmin(department.getAdminId(), DeptAdminEnum.YES.getValue());
+            userMapper.updateDeptAdmin(department.getAdminId(), DeptAdminEnum.YES.getValue(), entity.getId());
         }
         Integer update = departmentMapper.update(department);
         return update;
     }
 
     @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
@@ -91,8 +109,8 @@
     }
 
     @Override
-    public List<Department> gets() {
-        return departmentMapper.gets();
+    public List<Department> gets(List<Integer> deptId) {
+        return TreeUtils.build(departmentMapper.gets(deptId));
     }
 
     @Override
@@ -111,24 +129,57 @@
     }
 
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public void updateAdmin(UpdateDeptAdminVO form) {
         Department entity = departmentMapper.getById(form.getId());
-        if (Objects.isNull(form)) {
+        if (Objects.isNull(entity)) {
             throw new RuntimeException("璇ラ儴闂ㄤ笉瀛樺湪");
         }
-        if (Objects.nonNull(form.getAdminId())) {
-            if (! Objects.equals(entity.getAdminId(), form.getAdminId())) {
-                // 鍙栨秷鍘熷厛鐨勭鐞嗗憳鏍囪瘑
-                List<Integer> depIds = departmentMapper.selectByAdminId(entity.getAdminId());
-                depIds.remove(form.getId());
-                if (depIds.size() < 1) {
-                    userMapper.updateDeptAdmin(entity.getAdminId(), DeptAdminEnum.NO.getValue());
-                }
+        if (! CollectionUtils.isEmpty(form.getAdminIds())) {
+            List<BaseSelect> deptUserList = userDepartmentMapper.getDeptUserList(entity.getId());
+            List<Integer> userIds = deptUserList.stream().map(BaseSelect::getId).collect(Collectors.toList());
+            if (! CollectionUtils.isEmpty(userIds)) {
+                // 鍏堟竻闄や箣鍓嶇殑绠$悊鍛樻爣璇�
+                userMapper.clearDeptAdmin(userIds, entity.getId());
             }
             // 鏍囪瘑閮ㄩ棬绠$悊鍛�
-            userMapper.updateDeptAdmin(form.getAdminId(), DeptAdminEnum.YES.getValue());
+            userMapper.updateDeptAdmin(form.getAdminIds(), DeptAdminEnum.YES.getValue(), entity.getId());
         }
-        entity.setAdminId(form.getAdminId());
-        departmentMapper.update(entity);
+        // 淇敼琚�夋嫨鐨勭敤鎴蜂负閮ㄩ棬绠$悊鍛�
+        userMapper.updateUserDeptAdmin(form);
+        // 淇敼娌¤閫夋嫨鐨勭敤鎴峰苟涓斾笉鏄埆鐨勯儴闂ㄧ鐞嗗憳鐨勭敤鎴疯鑹蹭负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