ZhouXY108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-05-09 23:51:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

:trollface: 优化EasyStream中toTree对predicate的处理，过滤掉id为null的元素，避免StackOverflow

Browse Source
This commit is contained in:
VampireAchao 2022-10-07 11:53:31 +08:00
parent 2c2ad12972
commit c1acb2e148
3 changed files with 14 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
hutool-core/src/main/java/cn/hutool/core/stream/CollectorUtil.java
View File

@ -373,7 +373,7 @@ public class CollectorUtil {
* <p>将集合转换为树，默认用 {@code parentId == null} 来判断树的根节点
* 因为需要在当前传入数据里查找，所以这是一个结束操作 <br>
*
* @param idGetter id的getter对应的lambda，可以写作 {@code Student::getId}
* @param idGetter id的getter对应的lambda，可以写作 {@code Student::getId} 会过滤掉id为null的元素
* @param pIdGetter parentId的getter对应的lambda，可以写作 {@code Student::getParentId}
* @param childrenSetter children的setter对应的lambda，可以写作{ @code Student::setChildren}
* @param isParallel 是否并行去组装，数据量特别大时使用
@ -397,7 +397,7 @@ public class CollectorUtil {
* <p>将集合转换为树，默认用 {@code parentId == pidValue} 来判断树的根节点，可以为null
* 因为需要在当前传入数据里查找，所以这是一个结束操作 <br>
*
* @param idGetter id的getter对应的lambda，可以写作 {@code Student::getId}
* @param idGetter id的getter对应的lambda，可以写作 {@code Student::getId} 会过滤掉id为null的元素
* @param pIdGetter parentId的getter对应的lambda，可以写作 {@code Student::getParentId}
* @param pidValue pid的值
* @param childrenSetter children的setter对应的lambda，可以写作{ @code Student::setChildren}
@ -417,7 +417,7 @@ public class CollectorUtil {
final R pidValue,
final BiConsumer<T, List<T>> childrenSetter,
final boolean isParallel) {
return Collectors.collectingAndThen(groupingBy(pIdGetter, Collectors.toList()),
return Collectors.collectingAndThen(filtering(e -> idGetter.apply(e) != null, groupingBy(pIdGetter, Collectors.toList())),
getChildrenFromMapByPidAndSet(idGetter, pIdValuesMap -> pIdValuesMap.get(pidValue), childrenSetter, isParallel));
}
@ -425,7 +425,7 @@ public class CollectorUtil {
* 将集合转换为树，自定义根节点的判断条件
* 因为需要在当前传入数据里查找，所以这是一个结束操作
*
* @param idGetter id的getter对应的lambda，可以写作 {@code Student::getId}
* @param idGetter id的getter对应的lambda，可以写作 {@code Student::getId} 会过滤掉id为null的元素
* @param pIdGetter parentId的getter对应的lambda，可以写作 {@code Student::getParentId}
* @param childrenSetter children的setter对应的lambda，可以写作 {@code Student::setChildren}
* @param parentPredicate 树顶部的判断条件，可以写作 {@code s -> Objects.equals(s.getParentId(),0L) }
@ -447,21 +447,12 @@ public class CollectorUtil {
final Predicate<T> parentPredicate,
final boolean isParallel) {
final List<T> parents = new ArrayList<>();
return Collectors.collectingAndThen(groupingBy(pIdGetter,
new SimpleCollector<>(ArrayList::new,
(acc, e) -> {
if (parentPredicate.test(e)) {
parents.add(e);
}
if (idGetter.apply(e) != null) {
acc.add(e);
}
},
(left, right) -> {
left.addAll(right);
return left;
},
CH_ID)),
return Collectors.collectingAndThen(filtering(e -> {
if (parentPredicate.test(e)) {
parents.add(e);
}
return idGetter.apply(e) != null;
}, groupingBy(pIdGetter)),
getChildrenFromMapByPidAndSet(idGetter, pIdValuesMap -> parents, childrenSetter, isParallel));
}

4
hutool-core/src/main/java/cn/hutool/core/stream/TerminableWrappedStream.java
View File

@ -188,7 +188,7 @@ public interface TerminableWrappedStream<T, S extends TerminableWrappedStream<T,
* <p>将集合转换为树，默认用 {@code parentId == null} 来判断树的根节点
* 因为需要在当前传入数据里查找，所以这是一个结束操作 <br>
*
* @param idGetter id的getter对应的lambda，可以写作 {@code Student::getId}
* @param idGetter id的getter对应的lambda，可以写作 {@code Student::getId} 会过滤掉id为null的元素
* @param pIdGetter parentId的getter对应的lambda，可以写作 {@code Student::getParentId}
* @param childrenSetter children的setter对应的lambda，可以写作{ @code Student::setChildren}
* @param <R> 此处是id、parentId的泛型限制
@ -211,7 +211,7 @@ public interface TerminableWrappedStream<T, S extends TerminableWrappedStream<T,
* 将集合转换为树，自定义根节点的判断条件
* 因为需要在当前传入数据里查找，所以这是一个结束操作
*
* @param idGetter id的getter对应的lambda，可以写作 {@code Student::getId}
* @param idGetter id的getter对应的lambda，可以写作 {@code Student::getId} 会过滤掉id为null的元素
* @param pIdGetter parentId的getter对应的lambda，可以写作 {@code Student::getParentId}
* @param childrenSetter children的setter对应的lambda，可以写作 {@code Student::setChildren}
* @param parentPredicate 树顶部的判断条件，可以写作 {@code s -> Objects.equals(s.getParentId(),0L) }

2
hutool-core/src/test/java/cn/hutool/core/stream/EasyStreamTest.java
View File

@ -434,6 +434,8 @@ public class EasyStreamTest {
Consumer<Object> test = o -> {
final List<Student> studentTree = EasyStream
.of(
// 会过滤掉id为null的元素
Student.builder().name("top").build(),
Student.builder().id(1L).name("dromara").build(),
Student.builder().id(2L).name("baomidou").build(),
Student.builder().id(3L).name("hutool").parentId(1L).build(),