从流中获取最后 n 个元素
2022-09-01 02:03:42
在丢弃流的前 n 个元素后,返回由此流的其余元素组成的流。如果此流包含的元素少于 n 个,则将返回一个空流。
all.stream().skip(Math.max(0, all.size() - n)).forEach(doSomething);
自定义收集器可以这样编写:
public static <T> Collector<T, ?, List<T>> lastN(int n) {
return Collector.<T, Deque<T>, List<T>>of(ArrayDeque::new, (acc, t) -> {
if(acc.size() == n)
acc.pollFirst();
acc.add(t);
}, (acc1, acc2) -> {
while(acc2.size() < n && !acc1.isEmpty()) {
acc2.addFirst(acc1.pollLast());
}
return acc2;
}, ArrayList::new);
}
并像这样使用它:
List<String> lastTen = input.stream().collect(lastN(10));