Java 8 lambda：将集合转换为元素的映射，迭代位置

Question

如何将 ["a", "b", "c"] 之类的集合转换为 {"a": 0, "b":1, "c":2} 之类的映射，其值为迭代的顺序。 JDK8中是否有一个带有流和收集器的衬垫？ 老式的方式是这样的：

    Collection<String> col = apiCall();
    Map<String, Integer> map = new HashMap<>();
    int pos = 0;
    for (String s : collection) {
        map.put(s, pos++);
    }

Answer 1

如果您不需要并行流，您可以使用地图的长度作为索引计数器：

collection.stream().forEach(i -> map.put(i, map.size() + 1));

Answer 2

这是一种方法：

List<String> list = Arrays.asList("a", "b", "c");

Map<String, Integer> map =
    IntStream.range(0, list.size())
        .boxed()
        .collect(toMap(i -> list.get(i), i -> i));

不一定是单行或比直接循环更短，但如果将toMap 更改为toConcurrentMap，它确实可以使用并行流。

另外请注意，这假设您有一个随机访问列表，而不是一般的Collection。如果您有一个 Collection ，否则您无法做出任何假设，那么除了按顺序迭代它并增加一个计数器之外，您无能为力。

更新

OP 已澄清输入是Collection 而不是List，因此上述内容不适用。似乎我们可以对输入Collection 做出很少的假设。 OP 已指定迭代顺序。使用顺序迭代器，元素将以 some 顺序出现，尽管不能保证它。它可能会从一个运行到另一个运行，甚至从一个迭代到下一个迭代（尽管这在实践中是不寻常的——除非底层集合被修改）。

如果需要保留确切的迭代顺序，我认为没有办法将其保留到结果 Map 中而不依次迭代输入 Collection。

但是，如果确切的迭代顺序并不重要，并且要求输出 Map 对每个输入元素具有唯一值，则可以并行执行类似的操作：

Collection<String> col = apiCall();
Iterator<String> iter = col.iterator();

Map<String, Integer> map =
    IntStream.range(0, col.size())
        .parallel()
        .boxed()
        .collect(toConcurrentMap(i -> { synchronized (iter) { return iter.next(); }},
                                 i -> i));

现在这远单线。我也不清楚它有多大用处。 :-) 但它确实表明可以并行执行类似的操作。请注意，我们必须同步对输入集合的迭代器的访问，因为它将从多个线程中调用。另请注意，这是迭代器的不寻常用法，因为我们从不调用hasNext，并且我们假设调用next 的次数与输入集合的size() 返回的次数完全相同。

Answer 3

基于maba’s answer，一般解决方案是：

collection.stream().forEachOrdered(i -> map.put(i, map.size()));

来自documentation of void forEachOrdered(Consumer<? super T> action)：

此操作一次处理一个元素，如果存在则按遇到的顺序。

这里的重要方面是如果有订单，它会保留订单，例如如果Collection 是SortedSet 或List。这样的流称为有序流（不要与排序流混淆）。它可能由不同的线程调用消费者方法，但始终确保“一次一个”和线程安全保证。

当然，如果流是并行的，它不会从并行执行中受益。

---

为了完整起见，如果它们仍然是有序，那么即使在使用并行处理的并行流上也可以使用以下解决方案：

stream.collect(HashMap::new, (m, i) -> m.put(i, m.size()),
  (a, b) -> {int offset = a.size(); b.forEach((k, v) -> a.put(k, v + offset));});

Answer 4

如果您不介意使用 3rd 方库，我的 cyclops-react 库具有所有 JDK Collection 类型的扩展，并附有大量强大的操作，因此您可以像这样实现：-

    CollectionX<String> col = CollectionX.fromCollection(orgCol);
    col.zipWithIndex()
       .toMap(k->k.v1, v->v.v2);

cyclops-react Collection 扩展是急切的，所以你可以使用我们的 Stream 扩展获得更好的性能，ReactiveSeq（它扩展了 jOOλ 的 Seq，它又是 JDK 的 java.util.stream.Stream 的扩展，它还实现了反应流 api)。

    ReactiveSeq.fromCollection(col)
               .zipWithIndex()
               .toMap(k->k.v1, v->v.v2);

Answer 5

您可以使用 AtomicInteger 作为流中的索引：

    Collection col = Arrays.asList("a", "b", "c");
    AtomicInteger index = new AtomicInteger(0);
    Map collectedMap =  col.stream().collect(Collectors.toMap(Function.identity(), el -> index.getAndIncrement()));
    System.out.println("collectedMap = " + collectedMap);

Answer 6

试试

    int[] pos = { 0 };
    list.forEach( a -> map.put(a, pos[0]++));