ZhouXY108/plusone-commons
1
0
Fork 0
forked from plusone/plusone-commons
Code Pull Requests Activity

完善 TreeBuilder。

Browse Source
This commit is contained in:
zhouxy108 2023-07-19 01:25:21 +08:00
parent 47eb91d015
commit 201cf39e0c
2 changed files with 6 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/main/java/xyz/zhouxy/plusone/commons/util/TreeBuilder.java
View File

@ -8,14 +8,14 @@ import java.util.function.BiConsumer;
import java.util.function.Function;
import java.util.stream.Collectors;
public class TreeBuilder<T, TIdentity> {
public class TreeBuilder<T, TSubTree extends T, TIdentity> {
private final Collection<T> nodes;
private final Function<T, TIdentity> identityGetter;
private final Function<T, Optional<TIdentity>> parentIdentityGetter;
private final BiConsumer<T, T> addChildrenMethod;
private final BiConsumer<TSubTree, T> addChildrenMethod;
public TreeBuilder(Collection<T> nodes, Function<T, TIdentity> identityGetter,
Function<T, Optional<TIdentity>> parentIdentityGetter, BiConsumer<T, T> addChildren) {
Function<T, Optional<TIdentity>> parentIdentityGetter, BiConsumer<TSubTree, T> addChildren) {
this.nodes = nodes;
this.identityGetter = identityGetter;
this.parentIdentityGetter = parentIdentityGetter;
@ -30,7 +30,8 @@ public class TreeBuilder<T, TIdentity> {
for (T node : this.nodes) {
Optional<TIdentity> parentIdentity = parentIdentityGetter.apply(node);
if (parentIdentity.isPresent() && identityNodeMap.containsKey(parentIdentity.get())) {
T parentNode = identityNodeMap.get(parentIdentity.get());
@SuppressWarnings("all")
TSubTree parentNode = (TSubTree) identityNodeMap.get(parentIdentity.get());
addChildrenMethod.accept(parentNode, node);
}
}

11
src/test/java/xyz/zhouxy/plusone/commons/util/TreeBuilderTests.java
View File

@ -35,7 +35,7 @@ class TreeBuilderTests {
menus,
Menu::getMenuCode,
Menu::getParentMenuCode,
Menu::addChild)
MenuList::addChild)
.buildTree();
log.info("menuTree: {}", menuTree);
}
@ -64,8 +64,6 @@ abstract class Menu {
public String getTitle() {
return title;
}
public abstract void addChild(Menu child);
}
@ToString(callSuper = true)
@ -89,12 +87,6 @@ class MenuItem extends Menu {
public String getUrl() {
return url;
}
@Override
@Deprecated
public void addChild(Menu child) {
throw new UnsupportedOperationException("Unimplemented method 'addChild'");
}
}
@ToString(callSuper = true)
@ -126,7 +118,6 @@ class MenuList extends Menu {
return instance;
}
@Override
public void addChild(Menu child) {
if (this.children == null) {
this.children = Lists.newArrayList();