fix: 修复线上环境零部件树问题

This commit is contained in:
2024-02-07 19:11:50 +08:00
parent 6ca2e3c7cc
commit b872bc4b14
@@ -44,15 +44,15 @@ public class PartNameServiceImpl extends ServiceImpl<PartNameMapper, PartName>
// 将所有节点转换为映射,以便快速查找
Map<String, PartName> nodeMap = allNodes.stream()
.collect(Collectors.toMap(PartName::getId, Function.identity()));
.collect(Collectors.toMap(PartName::getCode, Function.identity()));
// 构建树形结构
List<PartName> 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<>());