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()
|
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<>();
|
List<PartName> topLevelNodes = new ArrayList<>();
|
||||||
allNodes.forEach(node -> {
|
allNodes.forEach(node -> {
|
||||||
if (node.getParentId() == null) {
|
if (node.getParentCode().equals("Part")) {
|
||||||
topLevelNodes.add(node);
|
topLevelNodes.add(node);
|
||||||
} else {
|
} else {
|
||||||
PartName parentNode = nodeMap.get(node.getParentId());
|
PartName parentNode = nodeMap.get(node.getParentCode());
|
||||||
if (parentNode != null) {
|
if (parentNode != null) {
|
||||||
if (parentNode.getChildPartNameList() == null) {
|
if (parentNode.getChildPartNameList() == null) {
|
||||||
parentNode.setChildPartNameList(new ArrayList<>());
|
parentNode.setChildPartNameList(new ArrayList<>());
|
||||||
|
|||||||
Reference in New Issue
Block a user