mirror of
https://gitee.com/chinabugotech/hutool.git
synced 2025-05-09 23:51:34 +08:00
:trollface:
This commit is contained in:
parent
62c80a2184
commit
0f3c658746
@ -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());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -66,4 +66,16 @@ public class CollectorUtilTest {
|
|||||||
Assert.assertEquals((Integer) 5, map.get("5"));
|
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);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user