идиоматический перечисление потока объектов в Java 8

Question

Как можно идиоматический перечислить Stream<T>, которая отображает каждый T экземпляр уникального целым число с помощью Java-8 методов потока (например, для массива T[] values, создавая Map<T,Integer>, где Map.get(values[i]) == i вычисляется в true)?

В настоящее время, я определяю анонимный класс, который приращение int поля для использования с Collectors.toMap(..) методом:

private static <T> Map<T, Integer> createIdMap(final Stream<T> values) { 
    return values.collect(Collectors.toMap(Function.identity(), new Function<T, Integer>() { 

     private int nextId = 0; 

     @Override 
     public Integer apply(final T t) { 
      return nextId++; 
     } 

    })); 
} 

Однако, есть не более лаконичный/элегантный способ сделать это с помощью Java- поток API? — бонусные баллы, если их можно безопасно распараллелить.

Answer 1

Ваш подход потерпит неудачу, если есть дубликат элемента.

Кроме того, ваша задача требует изменчивого состояния, поэтому ее можно решить с помощью Mutable reduction. Когда мы заполняем карту, мы можем просто использовать размер карты для получения неиспользуемого id.

Более сложная часть - операция слияния. Следующая операция просто повторяет задания для правильной карты, которые будут обрабатывать потенциальные дубликаты.

private static <T> Map<T, Integer> createIdMap(Stream<T> values) { 
    return values.collect(HashMap::new, (m,t) -> m.putIfAbsent(t,m.size()), 
     (m1,m2) -> { 
      if(m1.isEmpty()) m1.putAll(m2); 
      else m2.keySet().forEach(t -> m1.putIfAbsent(t, m1.size())); 
     }); 
} 

Если опираться на уникальные элементы, или вставить явное distinct(), мы можем использовать

private static <T> Map<T, Integer> createIdMap(Stream<T> values) { 
    return values.distinct().collect(HashMap::new, (m,t) -> m.put(t,m.size()), 
     (m1,m2) -> { int leftSize=m1.size(); 
      if(leftSize==0) m1.putAll(m2); 
      else m2.forEach((t,id) -> m1.put(t, leftSize+id)); 
     }); 

} 

Answer 2

Я хотел бы сделать это таким образом:

private static <T> Map<T, Integer> createIdMap2(final Stream<T> values) { 
    List<T> list = values.collect(Collectors.toList()); 
    return IntStream.range(0, list.size()).boxed() 
      .collect(Collectors.toMap(list::get, Function.identity())); 
} 

Ради или параллельности, он может быть изменен на

return IntStream.range(0, list.size()).parallel().boxed(). 
       (...) 

Answer 3

Сравнивая для преобразования входного потока в списке первым в растворе, предоставленной Andremoniy. Я бы предпочел сделать это по-другому, потому что мы не знаем стоимость «toList()» и «list.get (i)», и нет необходимости создавать дополнительный список, который может быть небольшим или большим.

private static <T> Map<T, Integer> createIdMap2(final Stream<T> values) { 
    final MutableInt idx = MutableInt.of(0); // Or: final AtomicInteger idx = new AtomicInteger(0);   
    return values.collect(Collectors.toMap(Function.identity(), e -> idx.getAndIncrement())); 
} 

Независимо от вопроса, я думаю, что это плохая конструкция для передачи потоков в качестве параметров в методе.