如果你:
>不喜欢创建包含流中所有字符串的列表的想法
>不想使用外部库
>喜欢弄脏你的手
然后,您可以使用Java 8低级流构建器StreamSupport和Spliterator创建一种从流中对元素进行分组的方法:
class StreamUtils {
public static Stream> sliding(int size, Stream stream) {
return sliding(size, 1, stream);
}
public static Stream> sliding(int size, int step, Stream stream) {
Spliterator spliterator = stream.spliterator();
long estimateSize;
if (!spliterator.hasCharacteristics(Spliterator.SIZED)) {
estimateSize = Long.MAX_VALUE;
} else if (size > spliterator.estimateSize()) {
estimateSize = 0;
} else {
estimateSize = (spliterator.estimateSize() - size) / step + 1;
}
return StreamSupport.stream(
new Spliterators.AbstractSpliterator>(estimateSize, spliterator.characteristics()) {
List buffer = new ArrayList<>(size);
@Override
public boolean tryAdvance(Consumer super List> consumer) {
while (buffer.size() < size && spliterator.tryAdvance(buffer::add)) {
// Nothing to do
}
if (buffer.size() == size) {
List keep = new ArrayList<>(buffer.subList(step, size));
consumer.accept(buffer);
buffer = keep;
return true;
}
return false;
}
}, stream.isParallel());
}
}
方法和参数命名的灵感来自他们的Scala对应物.
我们来测试一下:
Stream testing = Stream.of("A", "Apple", "B", "Banana", "C", "Carrot");
System.out.println(StreamUtils.sliding(2, testing).collect(Collectors.toList()));
[[A, Apple], [Apple, B], [B, Banana], [Banana, C], [C, Carrot]]
不重复元素怎么样:
Stream testing = Stream.of("A", "Apple", "B", "Banana", "C", "Carrot");
System.out.println(StreamUtils.sliding(2, 2, testing).collect(Collectors.toList()));
[[A, Apple], [B, Banana], [C, Carrot]]
现在有一个无限的流:
StreamUtils.sliding(5, Stream.iterate(0, n -> n + 1))
.limit(5)
.forEach(System.out::println);
[0, 1, 2, 3, 4] [1, 2, 3, 4, 5] [2, 3, 4, 5, 6] [3, 4, 5, 6, 7] [4, 5, 6, 7, 8]