Как сопоставить элементы списка с их индексами с использованием потоков Java 8?
Имея список строк, мне нужно построить список объектов, которые эффективно пары (string, its position in the list)
. В настоящее время у меня есть такой код с использованием коллекций google:
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();
}
Я хотел бы сделать это, используя потоки Java 8. Если бы не было индекса, я мог бы просто сделать:
public Robots(List<String> names) {
this.list = names.stream()
.map(Robot::new) // no index here
.collect(collectingAndThen(
Collectors.toList(),
Collections::unmodifiableList
));
}
Чтобы получить индекс, мне нужно было бы сделать что-то вроде этого:
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
));
}
Однако в документации указано, что функция сопоставления должна быть неактивной, но AtomicInteger
фактически является ее состоянием.
Есть ли способ сопоставить элементы последовательного потока с их позициями в потоке?
Ответы
Ответ 1
Вы можете сделать что-то вроде этого:
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));
}
Однако это может быть не так эффективно в зависимости от базовой реализации списка. Вы можете взять итератор из IntStream
; затем вызов next()
в mapToObj
.
В качестве альтернативы библиотека proton-pack определяет функциональность zipWithIndex
для потоков:
this.list = StreamUtils.zipWithIndex(names.stream())
.map(i -> new Robot(i.getIndex(), i.getValue()))
.collect(collectingAndThen(toList(), Collections::unmodifiableList));
Ответ 2
Самый простой способ - индексы потока:
List<Robot> robots = IntStream.range(0, names.size())
.mapToObj(i -> new Robot(i, names.get(i))
.collect(toList());