Java8 Stream.mapでindexをつける（タプル版）

前置き

i02302さんのJava8以降のStream.mapでindexをつけるでStreamの各要素にindexを付加する方法が書かれていました。

最初に見たときはindexのカウンターに配列を使っていて、並列Streamも考慮してAtomicIntegerにした方がいいんじゃないかな…と思っていましたが、さっき見たらそう変わっていました(笑)

あとは、intのindex部分がBiFuntionを使うとIntegerになってしまうので、関数型インターフェースも独自に用意した方がよいのではないかと思います。

public interface BiIntFunction<T, R> {
    public R apply(T t, int n);
}

public class WithIndex {
    public static <T, R> Function<T, R> apply(int start, BiIntFunction<T, R> function) {
        AtomicInteger counter = new AtomicInteger(start);
        return object -> function.apply(object, counter.getAndAdd(1));
    }
}

stream.map(WithIndex.apply(1, (obj, i) -> obj + i))

タプルを用意する方法

ScalaのzipWithIndexだと要素とインデックスのペア（タプル）を返すので、それと同様にする方法もあると思います。

public class ObjectIndexPair<T> {
    private final T object;
    private final int index;

    private ObjectIndexPair(T object, int index) {
        this.object = object;
        this.index = index;
    }

    public T getObject() {
        return object;
    }

    public int getIndex() {
        return index;
    }

    public static <T> Function<T, ObjectIndexPair<T>> start(int start) {
        AtomicInteger counter = new AtomicInteger(start);
        return object -> new ObjectIndexPair<>(object, counter.getAndAdd(1));
    }
}

stream.map(ObjectIndexPair.start(1)).map(pair -> pair.getObject() + pair.getIndex())