ZhouXY108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-04-19 03:01:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

:trollface: 递归改为遍历

Browse Source
This commit is contained in:
VampireAchao 2022-09-19 11:53:58 +08:00
parent e81585d7d0
commit 0dda333b07
1 changed files with 8 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,7 +1,6 @@
package cn.hutool.core.stream;
import cn.hutool.core.lang.Opt;
import cn.hutool.core.lang.mutable.MutableObj;
import cn.hutool.core.text.StrUtil;
import cn.hutool.core.util.ArrayUtil;
@ -481,18 +480,14 @@ public class CollectorUtil {
final BiConsumer<T, List<T>> childrenSetter,
final boolean isParallel) {
return pIdValuesMap -> {
final MutableObj<Consumer<List<T>>> recursiveRef = new MutableObj<>();
final Consumer<List<T>> recursive = parents -> EasyStream.of(parents, isParallel).forEach(parent -> {
final List<T> children = pIdValuesMap.get(idGetter.apply(parent));
childrenSetter.accept(parent, children);
recursiveRef.get().accept(children);
});
final List<T> parents = parentFactory.apply(pIdValuesMap);
if (false == parents.isEmpty()) {
recursiveRef.set(recursive);
recursiveRef.get().accept(parents);
}
return parents;
EasyStream.of(pIdValuesMap.values(), isParallel).flat(Function.identity())
.forEach(value -> {
final List<T> children = pIdValuesMap.get(idGetter.apply(value));
if (children != null) {
childrenSetter.accept(value, children);
}
});
return parentFactory.apply(pIdValuesMap);
};
}