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/controller/DeptController.java | 49 +++++++++++++++++++++++++++++++++++++++++++++---- 1 files changed, 45 insertions(+), 4 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("淇敼澶辫触"); } -- Gitblit v1.9.1