Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How to map elements of the list to their indices using Java 8 streams?

Having a list of strings, I need to construct a list of objects which are effectively pairs (string, its position in the list). Currently I have such code using google collections:

public Robots(List<String> names) {
    ImmutableList.Builder<Robot> builder = ImmutableList.builder();
    for (int i = 0; i < names.size(); i++) {
        builder.add(new Robot(i, names.get(i)));
    }
    this.list = builder.build();
}

I would like to do this using Java 8 streams. If there was no index, I could just do:

public Robots(List<String> names) {
    this.list = names.stream()
            .map(Robot::new) // no index here
            .collect(collectingAndThen(
                    Collectors.toList(),
                    Collections::unmodifiableList
            ));
}

To get the index, I would have to do something like this:

public Robots(List<String> names) {
    AtomicInteger integer = new AtomicInteger(0);
    this.list = names.stream()
            .map(string -> new Robot(integer.getAndIncrement(), string))
            .collect(collectingAndThen(
                    Collectors.toList(),
                    Collections::unmodifiableList
            ));
}

However, the documentation says that mapping function should be stateless, but the AtomicInteger is effectively its state.

Is there a way to map elements of the sequential stream to their positions in the stream?

like image 287
Jaroslaw Pawlak Avatar asked Mar 11 '15 14:03

Jaroslaw Pawlak


2 Answers

You could do something like this:

public Robots(List<String> names) {
    this.list = IntStream.range(0, names.size())
                         .mapToObj(i -> new Robot(i, names.get(i)))
                         .collect(collectingAndThen(toList(), Collections::unmodifiableList));
}

However it may not be as efficient depending on the underlying implementation of the list. You could grab an iterator from the IntStream; then calling next() in the mapToObj.

As an alternative, the proton-pack library defines the zipWithIndex functionality for streams:

 this.list = StreamUtils.zipWithIndex(names.stream())
                        .map(i -> new Robot(i.getIndex(), i.getValue()))
                        .collect(collectingAndThen(toList(), Collections::unmodifiableList));
like image 101
Alexis C. Avatar answered Nov 15 '22 21:11

Alexis C.


The easiest way is to stream indices:

List<Robot> robots = IntStream.range(0, names.size())
                              .mapToObj(i -> new Robot(i, names.get(i))
                              .collect(toList());
like image 37
assylias Avatar answered Nov 15 '22 23:11

assylias