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 87d65dda..e8c68970 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 @@ -44,15 +44,15 @@ public class PartNameServiceImpl extends ServiceImpl // 将所有节点转换为映射,以便快速查找 Map nodeMap = allNodes.stream() - .collect(Collectors.toMap(PartName::getId, Function.identity())); + .collect(Collectors.toMap(PartName::getCode, Function.identity())); // 构建树形结构 List topLevelNodes = new ArrayList<>(); allNodes.forEach(node -> { - if (node.getParentId() == null) { + if (node.getParentCode().equals("Part")) { topLevelNodes.add(node); } else { - PartName parentNode = nodeMap.get(node.getParentId()); + PartName parentNode = nodeMap.get(node.getParentCode()); if (parentNode != null) { if (parentNode.getChildPartNameList() == null) { parentNode.setChildPartNameList(new ArrayList<>());