我想对一个集合进行分组,但我不希望聚合包含一些值。这样做的最佳解决方案是什么?
解决方案可能是收集器filtering(Predicate,Collector)
但没有这样的收集器。没有实现自己的收藏家,有没有办法做到这一点?
IntStream.range(0,100).collect(
groupingBy(
i -> i % 3,
HashMap::new,
filtering( i -> i % 2 == 0, toSet() )
)
);
我所知道的另一种方法是事先进行过滤,即
stream.filter(filterPredicate).collect(collector)
也就是说,目前还不清楚你实际上要对你的过滤函数做什么,它需要三个参数而不是两个。
我怀疑你试图将一个函数映射到与过滤器匹配的元素上,但这本身就是一个地图,你可以很容易地做到这一点:
Collectors.mapping(input -> condition(input) ? function(input) : input, collector)
制作自己的收藏家实际上并不困难:
static <T, A, R> Collector<T, A, R> filtering(
Predicate<? super T> filter, Collector<T, A, R> collector) {
return Collector.of(
collector.supplier(),
(accumulator, input) -> {
if (filter.test(input)) {
collector.accumulator().accept(accumulator, input);
}
},
collector.combiner(),
collector.finisher());
}
在Java 9中有一个新的filtering收集器:
Map<Department, Set<Employee>> wellPaidEmployeesByDepartment = employees.stream()
.collect(groupingBy(Employee::getDepartment,
filtering(e -> e.getSalary() > 2000, toSet())));