diff --git a/laws-module-system/src/main/java/com/jero/modules/system/entity/SysDictItem.java b/laws-module-system/src/main/java/com/jero/modules/system/entity/SysDictItem.java index 45cb2ac5..18c4e4b8 100644 --- a/laws-module-system/src/main/java/com/jero/modules/system/entity/SysDictItem.java +++ b/laws-module-system/src/main/java/com/jero/modules/system/entity/SysDictItem.java @@ -178,6 +178,13 @@ public class SysDictItem implements TreeNode { @TableField(exist = false) private List childrenList; + /** + * 是否还有子集 + */ + @TableField(exist = false) + @ApiModelProperty(value = "是否还有子集") + private boolean hasChild; + @Override public String obtainTreeNodeId() { return id; diff --git a/laws-modules/src/main/java/com/jero/modules/laws/standard/controller/LawsDictTreeController.java b/laws-modules/src/main/java/com/jero/modules/laws/standard/controller/LawsDictTreeController.java index c19eca37..d54277b7 100644 --- a/laws-modules/src/main/java/com/jero/modules/laws/standard/controller/LawsDictTreeController.java +++ b/laws-modules/src/main/java/com/jero/modules/laws/standard/controller/LawsDictTreeController.java @@ -33,9 +33,10 @@ public class LawsDictTreeController { @GetMapping("/page") public Result> queryTreeList( @RequestParam(name = "nodeName", required = false, defaultValue = "") @ApiParam("一级节点名称") String nodeName, + @RequestParam("dictId") @ApiParam("字典编码") String dictId, @RequestParam(name = "pageNo", defaultValue = "1") @ApiParam("页码") Integer pageNo, @RequestParam(name = "pageSize", defaultValue = "10") @ApiParam("页面大小") Integer pageSize) { - TreePage list = dictTreeService.queryTreeList(nodeName, pageNo, pageSize); + TreePage list = dictTreeService.queryTreeList(nodeName, dictId, pageNo, pageSize); return Result.OK(list); } @@ -47,11 +48,12 @@ public class LawsDictTreeController { return Result.OK(list); } - @AutoLog(value = "树形结构-零部件树历史数据处理") - @ApiOperation(value = "树形结构-零部件树历史数据处理", notes = "树形结构-零部件树历史数据处理") + @AutoLog(value = "树形结构-获取子集") + @ApiOperation(value = "树形结构-获取子集", notes = "树形结构-获取子集") @GetMapping("/getChild") - public Result> getChild(@RequestParam("dictItemId") String dictItemId) { - return Result.OK(dictTreeService.getChild(dictItemId)); + public Result> getChild(@RequestParam("dictItemId") String dictItemId, + @RequestParam("dictId") @ApiParam("字典编码") String dictId) { + return Result.OK(dictTreeService.getChild(dictItemId, dictId)); } diff --git a/laws-modules/src/main/java/com/jero/modules/laws/standard/entity/TreePage.java b/laws-modules/src/main/java/com/jero/modules/laws/standard/entity/TreePage.java index 1c846c31..d389851b 100644 --- a/laws-modules/src/main/java/com/jero/modules/laws/standard/entity/TreePage.java +++ b/laws-modules/src/main/java/com/jero/modules/laws/standard/entity/TreePage.java @@ -1,7 +1,9 @@ package com.jero.modules.laws.standard.entity; import com.jero.modules.system.service.TreeNode; +import lombok.AllArgsConstructor; import lombok.Data; +import lombok.NoArgsConstructor; import java.util.*; import java.util.stream.Collectors; @@ -11,6 +13,8 @@ import java.util.stream.Collectors; * @Date: 2024/1/19 15:24 */ @Data +@NoArgsConstructor +@AllArgsConstructor public class TreePage { private List records; diff --git a/laws-modules/src/main/java/com/jero/modules/laws/standard/service/DictTreeService.java b/laws-modules/src/main/java/com/jero/modules/laws/standard/service/DictTreeService.java index c723d8f2..0666066f 100644 --- a/laws-modules/src/main/java/com/jero/modules/laws/standard/service/DictTreeService.java +++ b/laws-modules/src/main/java/com/jero/modules/laws/standard/service/DictTreeService.java @@ -1,5 +1,8 @@ package com.jero.modules.laws.standard.service; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.jero.modules.laws.standard.entity.TreePage; import com.jero.modules.system.entity.SysDictItem; import com.jero.modules.system.service.ISysDictItemService; @@ -8,6 +11,7 @@ import org.springframework.stereotype.Service; import javax.annotation.Resource; import java.util.List; +import java.util.stream.Collectors; @Service public class DictTreeService { @@ -18,15 +22,51 @@ public class DictTreeService { @Resource private ISysDictItemService dictItemService; - public TreePage queryTreeList(String nodeName, Integer pageNo, Integer pageSize) { - return null; + public TreePage queryTreeList(String nodeName, String dictId, Integer pageNo, Integer pageSize) { + LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); + queryWrapper.eq(SysDictItem::getDictId, dictId); + queryWrapper.eq(SysDictItem::getParentId, "0"); + if (nodeName != null && !nodeName.isEmpty()) { + queryWrapper.like(SysDictItem::getItemText, nodeName); + } + Page page = new Page<>(pageNo, pageSize); + IPage pageList = dictItemService.page(page, queryWrapper); + + List dictItemIds = pageList.getRecords().stream().map(SysDictItem::getId).collect(Collectors.toList()); + // 判断有没有子级 + queryWrapper.clear(); + queryWrapper.eq(SysDictItem::getDictId, dictId); + queryWrapper.in(SysDictItem::getParentId, dictItemIds); + List childList = dictItemService.list(queryWrapper); + pageList.getRecords().forEach(item -> { + if (childList.stream().anyMatch(child -> child.getParentId().equals(item.getId()))) { + item.setHasChild(true); + } + }); + return new TreePage<>(pageList.getRecords(), pageList.getTotal(), pageList.getSize(), pageList.getCurrent()); } public List queryEcho(String selectNodeCodes) { return null; } - public List getChild(String dictItemId) { - return null; + public List getChild(String dictItemId, String dictId) { + LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); + queryWrapper.eq(SysDictItem::getDictId, dictId); + queryWrapper.eq(SysDictItem::getParentId, dictItemId); + List list = dictItemService.list(queryWrapper); + + List dictItemIds = list.stream().map(SysDictItem::getId).collect(Collectors.toList()); + // 判断有没有子级 + queryWrapper.clear(); + queryWrapper.eq(SysDictItem::getDictId, dictId); + queryWrapper.in(SysDictItem::getParentId, dictItemIds); + List childList = dictItemService.list(queryWrapper); + list.forEach(item -> { + if (childList.stream().anyMatch(child -> child.getParentId().equals(item.getId()))) { + item.setHasChild(true); + } + }); + return list; } }