From 7e7dac82d82f3273cb709e404aa179787d561011 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期二, 20 二月 2024 13:02:48 +0800
Subject: [PATCH] #

---
 zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/entity/Dept.java               |    9 +++++----
 zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/DeptController.java |   49 +++++++++++++++++++++++++++++++++++++++++++++----
 2 files changed, 50 insertions(+), 8 deletions(-)

diff --git a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/DeptController.java b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/DeptController.java
index 5be3efb..f138f48 100644
--- a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/DeptController.java
+++ b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/controller/DeptController.java
@@ -11,15 +11,14 @@
 import com.zy.asrs.wcs.system.entity.Dept;
 import com.zy.asrs.wcs.system.service.DeptService;
 import com.zy.asrs.wcs.utils.ExcelUtil;
+import com.zy.asrs.wcs.utils.NodeUtils;
+import com.zy.asrs.wcs.utils.Utils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.web.bind.annotation.*;
 
 import javax.servlet.http.HttpServletResponse;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 @RestController
 @RequestMapping("/api")
@@ -43,6 +42,14 @@
     }
 
     @PreAuthorize("hasAuthority('system:dept:list')")
+    @PostMapping("/dept/tree")
+    public R tree(@RequestBody Map<String, Object> map) {
+        PageParam<Dept, BaseParam> param = new PageParam<>(buildParam(map, BaseParam.class), Dept.class);
+        List<Dept> deptList = deptService.list(param.buildWrapper(true));
+        return R.ok().add(Utils.toTreeData(deptList, 0L, Dept::getParentId, Dept::getId, Dept::setChildren));
+    }
+
+    @PreAuthorize("hasAuthority('system:dept:list')")
     @GetMapping("/dept/{id}")
     public R get(@PathVariable("id") Long id) {
         return R.ok().add(deptService.getById(id));
@@ -52,6 +59,24 @@
     @OperationLog("娣诲姞閮ㄩ棬")
     @PostMapping("/dept/save")
     public R save(@RequestBody Dept dept) {
+        if (dept.getParentId() != null && dept.getParentId() > 0) {
+            Dept parent = deptService.getById(dept.getParentId());
+            if (parent != null) {
+                dept.setParentName(parent.getName());
+            }
+        } else {
+            dept.setParentId(0L);
+        }
+
+        NodeUtils nodeUtils = new NodeUtils();
+        nodeUtils.generatePath0(item -> deptService.getById(item.getParentId()), dept, Dept::getId, Dept::getName, Dept::getParentId);
+        dept.setPath(nodeUtils.path.toString());
+        dept.setPathName(nodeUtils.pathName.toString());
+
+        dept.setCreateBy(getLoginUserId());
+        dept.setCreateTime(new Date());
+        dept.setUpdateBy(getLoginUserId());
+        dept.setUpdateTime(new Date());
         if (!deptService.save(dept)) {
             return R.error("娣诲姞澶辫触");
         }
@@ -62,6 +87,22 @@
     @OperationLog("淇敼閮ㄩ棬")
     @PostMapping("/dept/update")
     public R update(@RequestBody Dept dept) {
+        if (dept.getParentId() != null && dept.getParentId() > 0) {
+            Dept parent = deptService.getById(dept.getParentId());
+            if (parent != null) {
+                dept.setParentName(parent.getName());
+            }
+        } else {
+            dept.setParentId(0L);
+        }
+
+        NodeUtils nodeUtils = new NodeUtils();
+        nodeUtils.generatePath0(item -> deptService.getById(item.getParentId()), dept, Dept::getId, Dept::getName, Dept::getParentId);
+        dept.setPath(nodeUtils.path.toString());
+        dept.setPathName(nodeUtils.pathName.toString());
+
+        dept.setUpdateBy(getLoginUserId());
+        dept.setUpdateTime(new Date());
         if (!deptService.updateById(dept)) {
             return R.error("淇敼澶辫触");
         }
diff --git a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/entity/Dept.java b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/entity/Dept.java
index 5f6d8b4..bfee95f 100644
--- a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/entity/Dept.java
+++ b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/system/entity/Dept.java
@@ -1,9 +1,6 @@
 package com.zy.asrs.wcs.system.entity;
 
-import com.baomidou.mybatisplus.annotation.IdType;
-import com.baomidou.mybatisplus.annotation.TableId;
-import com.baomidou.mybatisplus.annotation.TableLogic;
-import com.baomidou.mybatisplus.annotation.TableName;
+import com.baomidou.mybatisplus.annotation.*;
 import com.zy.asrs.framework.common.Cools;
 import com.zy.asrs.framework.common.SpringUtils;
 import com.zy.asrs.wcs.system.service.HostService;
@@ -15,6 +12,7 @@
 import java.io.Serializable;
 import java.text.SimpleDateFormat;
 import java.util.Date;
+import java.util.List;
 
 @Data
 @TableName("sys_dept")
@@ -152,6 +150,9 @@
     @ApiModelProperty(value= "澶囨敞")
     private String memo;
 
+    @TableField(exist = false)
+    private List<Dept> children;
+
     public Dept() {}
 
     public Dept(String name,Long parentId,String parentName,String path,String pathName,String fullName,String brief,String code,Integer type,String leader,Integer count,Integer sort,Long hostId,Integer status,Integer deleted,Date createTime,Long createBy,Date updateTime,Long updateBy,String memo) {

--
Gitblit v1.9.1