Java で Sliding Window処理

動機

（時）系列処理とかでしばしば利用するSliding Window処理をJavaでやりたい。
SQLだと標準で（大半の実装で？）Window関数があるので悩まなくていいのですが、Javaでは無いので本記事を記載します。

StackOverflowにコードはあったのですが、Listでデータを持ってしまっているか、ライブラリを使っているかのどちらかでした。

引用させて頂くと以下のような感じ。

public static <T> Stream<List<T>> sliding(List<T> list, int size) {
   if(size > list.size()) 
       return Stream.empty();
   return IntStream.range(0, list.size()-size+1)
                   .mapToObj(start -> list.subList(start, start+size));
}

このコードの気に入らない点は一つだけです。パラメータがListなところです。これだとメモリの消費が大きくなってしまう場合があります。

コード

特段、工夫せずに素直に書き直しました。

public static <T> void sliding(Stream<T> stream, int size, Consumer<List<T>> consumer) {
	Iterator<T> itr = stream.iterator();
	List<T> buffer = new LinkedList<T>();
	while (itr.hasNext()) {
		buffer.add(itr.next());
		if (buffer.size() > size)
			buffer.remove(0);
		consumer.accept(buffer);
	}
}

動作確認

呼び出しの例は以下のような感じです。

sliding(Stream.of(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10), 3, System.out::println);

実行結果

上記の実行結果は、以下のような感じです。

[0]
[0, 1]
[0, 1, 2]
[1, 2, 3]
[2, 3, 4]
[3, 4, 5]
[4, 5, 6]
[5, 6, 7]
[6, 7, 8]
[7, 8, 9]
[8, 9, 10]

参考資料

StackOverflow - How to transform a Java stream into a sliding window?