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

!1307 增加将流分至两个和三个集合的收集器

Browse Source 
Merge pull request !1307 from tltwuyu/v6-dev
This commit is contained in:
Looly 2025-02-06 03:37:35 +00:00 committed by Gitee
commit 0542866c49
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 173 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

132
hutool-core/src/main/java/org/dromara/hutool/core/stream/CollectorUtil.java
View File

@ -17,6 +17,8 @@
package org.dromara.hutool.core.stream;
import org.dromara.hutool.core.array.ArrayUtil;
import org.dromara.hutool.core.lang.tuple.Pair;
import org.dromara.hutool.core.lang.tuple.Triple;
import org.dromara.hutool.core.text.StrUtil;
import java.util.*;
@ -468,4 +470,134 @@ public class CollectorUtil {
downstream.combiner(), downstream.finisher(),
downstream.characteristics());
}
/**
* 将一个Collection<T>两个属性分流至两个ArrayList,并使用Pair收集。
*
* @param lMapper 左属性收集方法
* @param rMapper 右属性收集方法
* @param <T> 元素类型
* @param <L> 左属性类型
* @param <R> 右属性类型
* @return Pair<List<L>,List<R>> Pair收集的两个List
* @author Tanglt
*/
public static <T, L, R> Collector<T, Pair<List<L>,List<R>>, Pair<List<L>,List<R>>> toPairList(Function<T, L> lMapper,
Function<T, R> rMapper) {
return toPairCollection(lMapper,rMapper,ArrayList::new,ArrayList::new);
}
/**
* 将一个Collection<T>两个属性分流至两个Collection,并使用Pair收集。需要指定Collection类型
*
* @param lMapper 左属性收集方法
* @param rMapper 右属性收集方法
* @param newCollectionL 左属性Collection创建方法
* @param newCollectionR 右属性Collection创建方法
* @param <T> 元素类型
* @param <L> 左属性类型
* @param <R> 右属性类型
* @param <LC> 左分流Collection类型
* @param <RC> 右分流Collection类型
* @return Pair<C<L>,C<R>> Pair收集的两个List
* @author Tanglt
*/
public static <T, L, R,LC extends Collection<L>,RC extends Collection<R>>
Collector<T, Pair<LC, RC>, Pair<LC, RC>> toPairCollection(Function<T, L> lMapper,
Function<T, R> rMapper,
Supplier<LC> newCollectionL,
Supplier<RC> newCollectionR) {
return new SimpleCollector<>(() -> Pair.of(newCollectionL.get(), newCollectionR.get()),
(listPair, element) -> {
L lValue = lMapper.apply(element);
if (lValue != null) {
listPair.getLeft().add(lValue);
}
R rValue = rMapper.apply(element);
if (rValue != null) {
listPair.getRight().add(rValue);
}
},
(listPair1, listPair2) -> {
listPair1.getLeft().addAll(listPair2.getLeft());
listPair1.getRight().addAll(listPair2.getRight());
return listPair1;
},
CH_ID);
}
/**
* 将一个Collection<T>三个属性分流至三个ArrayList,并使用Triple收集。
*
* @param lMapper 左属性收集方法
* @param mMapper 中属性收集方法
* @param rMapper 右属性收集方法
* @param <T> 元素类型
* @param <L> 左属性类型
* @param <M> 中属性类型
* @param <R> 右属性类型
* @return Triple<List<L>,List<M>,List<R>> Triple收集的三个List
* @author Tanglt
*/
public static <T, L, M, R>
Collector<T, Triple<List<L>, List<M>, List<R>>, Triple<List<L>, List<M>, List<R>>> toTripleList(Function<T, L> lMapper,
Function<T, M> mMapper,
Function<T, R> rMapper) {
return toTripleCollection(lMapper, mMapper, rMapper, ArrayList::new, ArrayList::new, ArrayList::new);
}
/**
* 将一个Collection<T>两个属性分流至两个Collection,并使用Pair收集。需要指定Collection类型
*
* @param lMapper 左属性收集方法
* @param mMapper 中属性收集方法
* @param rMapper 右属性收集方法
* @param newCollectionL 左属性Collection创建方法
* @param newCollectionM 中属性Collection创建方法
* @param newCollectionR 右属性Collection创建方法
* @param <T> 元素类型
* @param <L> 左属性类型
* @param <M> 中属性类型
* @param <R> 右属性类型
* @param <LC> 左分流Collection类型
* @param <MC> 中分流Collection类型
* @param <RC> 右分流Collection类型
* @return Triple<LC<L>,MC<M>,RC<R>> Triple收集的三个List
* @author Tanglt
*/
public static <T, L, M,R,
LC extends Collection<L>,
MC extends Collection<M>,
RC extends Collection<R>>
Collector<T, Triple<LC,MC,RC>,Triple<LC,MC,RC>> toTripleCollection(Function<T, L> lMapper,
Function<T, M> mMapper,
Function<T, R> rMapper,
Supplier<LC> newCollectionL,
Supplier<MC> newCollectionM,
Supplier<RC> newCollectionR){
return new SimpleCollector<>(
() -> Triple.of(newCollectionL.get(),newCollectionM.get(), newCollectionR.get()),
(listTriple, element) -> {
L lValue = lMapper.apply(element);
if (lValue != null) {
listTriple.getLeft().add(lValue);
}
M mValue = mMapper.apply(element);
if (mValue != null) {
listTriple.getMiddle().add(mValue);
}
R rValue = rMapper.apply(element);
if (rValue != null) {
listTriple.getRight().add(rValue);
}
},
(listTriple1, listTriple2) -> {
listTriple1.getLeft().addAll(listTriple2.getLeft());
listTriple1.getRight().addAll(listTriple2.getRight());
return listTriple1;
},
CH_ID);
}
}

41
hutool-core/src/test/java/org/dromara/hutool/core/stream/CollectorUtilTest.java
View File

@ -17,6 +17,8 @@
package org.dromara.hutool.core.stream;
import org.dromara.hutool.core.collection.ListUtil;
import org.dromara.hutool.core.lang.tuple.Pair;
import org.dromara.hutool.core.lang.tuple.Triple;
import org.dromara.hutool.core.map.MapUtil;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
@ -135,4 +137,43 @@ public class CollectorUtilTest {
}
@Test
public void testToPairList() {
final List<Pair<Integer,String>> list = Arrays.asList(Pair.of(1,"one"), Pair.of(2,"two"));
Pair<List<Integer>, List<String>> pairList = list.stream()
.collect(CollectorUtil.toPairList(Pair::getLeft, Pair::getRight));
Assertions.assertEquals(pairList.getLeft().size(),list.size());
Assertions.assertEquals(pairList.getRight().size(),list.size());
Pair<HashSet<Integer>, ArrayList<String>> pairMixed = list.stream()
.collect(CollectorUtil.toPairCollection(Pair::getLeft, Pair::getRight, HashSet::new, ArrayList::new));
Assertions.assertEquals(pairMixed.getLeft().size(),list.size());
Assertions.assertEquals(pairMixed.getRight().size(),list.size());
}
@Test
public void testToTripleList() {
final List<Triple<Integer,Long,String>> list
= Arrays.asList(Triple.of(1,1L,"one"), Triple.of(2,2L,"two"));
Triple<List<Integer>, List<Long>, List<String>> tripleList = list.stream()
.collect(CollectorUtil.toTripleList(Triple::getLeft, Triple::getMiddle, Triple::getRight));
Assertions.assertEquals(tripleList.getLeft().size(),list.size());
Assertions.assertEquals(tripleList.getMiddle().size(),list.size());
Assertions.assertEquals(tripleList.getRight().size(),list.size());
Triple<HashSet<Integer>, HashSet<Long>, ArrayList<String>> tripleMixed = list.stream()
.collect(CollectorUtil.toTripleCollection(Triple::getLeft, Triple::getMiddle, Triple::getRight, HashSet::new, HashSet::new, ArrayList::new));
Assertions.assertEquals(tripleMixed.getLeft().size(),list.size());
Assertions.assertEquals(tripleMixed.getMiddle().size(),list.size());
Assertions.assertEquals(tripleMixed.getRight().size(),list.size());
}
}