diff --git a/laws-modules/src/main/java/com/jero/modules/laws/standard/controller/LawsPartNameController.java b/laws-modules/src/main/java/com/jero/modules/laws/standard/controller/LawsPartNameController.java index 57436e65..55d7df66 100644 --- a/laws-modules/src/main/java/com/jero/modules/laws/standard/controller/LawsPartNameController.java +++ b/laws-modules/src/main/java/com/jero/modules/laws/standard/controller/LawsPartNameController.java @@ -38,6 +38,12 @@ public class LawsPartNameController { @RequestParam(name = "nodeName", required = false, defaultValue = "") @ApiParam("一级节点名称") String nodeName, @RequestParam(name = "pageNo", defaultValue = "1") @ApiParam("页码") Integer pageNo, @RequestParam(name = "pageSize", defaultValue = "10") @ApiParam("页面大小") Integer pageSize) { + if (pageNo == null) { + pageNo = 1; + } + if (pageSize == null) { + pageSize = 5; + } TreePage list = partNameService.queryTreeList(nodeName, pageNo, pageSize); return Result.OK(list); } diff --git a/laws-modules/src/main/java/com/jero/modules/laws/standard/service/impl/PartNameServiceImpl.java b/laws-modules/src/main/java/com/jero/modules/laws/standard/service/impl/PartNameServiceImpl.java index 3ac0c7dd..547794c0 100644 --- a/laws-modules/src/main/java/com/jero/modules/laws/standard/service/impl/PartNameServiceImpl.java +++ b/laws-modules/src/main/java/com/jero/modules/laws/standard/service/impl/PartNameServiceImpl.java @@ -48,7 +48,7 @@ public class PartNameServiceImpl extends ServiceImpl // 不做筛选的时候查询加速 if (StrUtil.isBlank(nodeName)) { // 构建树结构 - topLevelNodes = this.buildTree(new HashSet<>(allNodes)); + topLevelNodes = this.buildTree(allNodes); } else { // 做筛选的时候需要自己组树 queryWrapper.clear(); @@ -85,11 +85,11 @@ public class PartNameServiceImpl extends ServiceImpl return result; } - private List buildTree(Set nodesSet) { + private List buildTree(Collection nodesCollection) { List topLevelNodes = new ArrayList<>(); - Map nodeMap = nodesSet.stream() + Map nodeMap = nodesCollection.stream() .collect(Collectors.toMap(PartName::getCode, Function.identity())); - for (PartName node : nodesSet) { + for (PartName node : nodesCollection) { PartName parentNode = nodeMap.get(node.getParentCode()); if (parentNode != null) { if (parentNode.getChildPartNameList() == null) {