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
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);
}
}