修复device_channel的Id改变带来的tree查询bug

This commit is contained in:
朱俊杰 2022-03-15 16:31:06 +08:00
parent 0dfce85d2f
commit 70ada79c7a
5 changed files with 14 additions and 22 deletions

View File

@ -16,7 +16,7 @@ public class BaseNode<T> implements INode<T> {
/** /**
* 主键ID * 主键ID
*/ */
protected int id; protected String channelId;
/** /**
* 父节点ID * 父节点ID
@ -50,12 +50,8 @@ public class BaseNode<T> implements INode<T> {
} }
@Override @Override
public int getId() { public String getChannelId() {
return id; return channelId;
}
public void setId(int id) {
this.id = id;
} }
@Override @Override
@ -63,10 +59,6 @@ public class BaseNode<T> implements INode<T> {
return parentId; return parentId;
} }
public void setParentId(String parentId) {
this.parentId = parentId;
}
@Override @Override
public List<T> getChildren() { public List<T> getChildren() {
return children; return children;

View File

@ -15,8 +15,8 @@ public class ForestNode extends BaseNode<ForestNode> {
*/ */
private Object content; private Object content;
public ForestNode(int id, String parentId, Object content) { public ForestNode(String id, String parentId, Object content) {
this.id = id; this.channelId = id;
this.parentId = parentId; this.parentId = parentId;
this.content = content; this.content = content;
} }

View File

@ -17,15 +17,15 @@ public class ForestNodeManager<T extends INode<T>> {
/** /**
* 森林的所有节点 * 森林的所有节点
*/ */
private final ImmutableMap<Integer, T> nodeMap; private final ImmutableMap<String, T> nodeMap;
/** /**
* 森林的父节点ID * 森林的父节点ID
*/ */
private final Map<Integer, Object> parentIdMap = Maps.newHashMap(); private final Map<String, Object> parentIdMap = Maps.newHashMap();
public ForestNodeManager(List<T> nodes) { public ForestNodeManager(List<T> nodes) {
nodeMap = Maps.uniqueIndex(nodes, INode::getId); nodeMap = Maps.uniqueIndex(nodes, INode::getChannelId);
} }
/** /**
@ -46,7 +46,7 @@ public class ForestNodeManager<T extends INode<T>> {
* *
* @param parentId 父节点ID * @param parentId 父节点ID
*/ */
public void addParentId(int parentId) { public void addParentId(String parentId) {
parentIdMap.put(parentId, ""); parentIdMap.put(parentId, "");
} }
@ -58,7 +58,7 @@ public class ForestNodeManager<T extends INode<T>> {
public List<T> getRoot() { public List<T> getRoot() {
List<T> roots = new ArrayList<>(); List<T> roots = new ArrayList<>();
nodeMap.forEach((key, node) -> { nodeMap.forEach((key, node) -> {
if (node.getParentId() == null || parentIdMap.containsKey(node.getId())) { if (node.getParentId() == null || parentIdMap.containsKey(node.getChannelId())) {
roots.add(node); roots.add(node);
} }
}); });

View File

@ -25,7 +25,7 @@ public class ForestNodeMerger {
if (node != null) { if (node != null) {
node.getChildren().add(forestNode); node.getChildren().add(forestNode);
} else { } else {
forestNodeManager.addParentId(forestNode.getId()); forestNodeManager.addParentId(forestNode.getChannelId());
} }
} }
}); });
@ -37,8 +37,8 @@ public class ForestNodeMerger {
items.forEach(forestNode -> { items.forEach(forestNode -> {
if (forestNode.getParentId() != null) { if (forestNode.getParentId() != null) {
INode<T> node = forestNodeManager.getTreeNodeAt(forestNode.getParentId()); INode<T> node = forestNodeManager.getTreeNodeAt(forestNode.getParentId());
if (CollectionUtil.contains(parentIds, forestNode.getId())){ if (CollectionUtil.contains(parentIds, forestNode.getChannelId())){
forestNodeManager.addParentId(forestNode.getId()); forestNodeManager.addParentId(forestNode.getChannelId());
} else { } else {
if (node != null){ if (node != null){
node.getChildren().add(forestNode); node.getChildren().add(forestNode);

View File

@ -14,7 +14,7 @@ public interface INode<T> extends Serializable {
* *
* @return String * @return String
*/ */
int getId(); String getChannelId();
/** /**
* 父主键 * 父主键