:trollface:

This commit is contained in:
TanShengyuan 2022-09-24 17:07:56 +08:00 committed by Rosie
parent 62c80a2184
commit 0f3c658746
2 changed files with 53 additions and 20 deletions

View File

@ -495,4 +495,25 @@ public class CollectorUtil {
};
}
/**
* <p>过滤</p >
*
* @param predicate 断言
* @param downstream 下游操作
* @param <T> 元素类型
* @param <A> 中间类型
* @param <R> 结束类型
* @return 一个用于过滤元素的 {@link java.util.stream.Collector}
* @author TanShengYuan
*/
public static <T, A, R>
Collector<T, ?, R> filtering(final Predicate<? super T> predicate,
final Collector<? super T, A, R> downstream) {
final BiConsumer<A, ? super T> downstreamAccumulator = downstream.accumulator();
return new SimpleCollector<>(downstream.supplier(),
(r, t) -> Opt.of(t).filter(predicate).ifPresent(e -> downstreamAccumulator.accept(r, e)),
downstream.combiner(), downstream.finisher(),
downstream.characteristics());
}
}

View File

@ -66,4 +66,16 @@ public class CollectorUtilTest {
Assert.assertEquals((Integer) 5, map.get("5"));
}
@Test
public void testFiltering() {
final Map<Integer, Long> map = Stream.of(1, 2, 3)
.collect(Collectors.groupingBy(Function.identity(),
CollectorUtil.filtering(i -> i > 1, Collectors.counting())
));
Assert.assertEquals(MapUtil.builder()
.put(1, 0L)
.put(2, 1L)
.put(3, 1L)
.build(), map);
}
}