fix: 修复线上环境零部件树问题
This commit is contained in:
+3
-3
@@ -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<>());
|
||||
|
||||
Reference in New Issue
Block a user