program tip

스트림에서 Collections.toMap ()을 사용할 때 List의 반복 순서를 어떻게 유지합니까?

radiobox 2020. 11. 20. 08:50
반응형

스트림에서 Collections.toMap ()을 사용할 때 List의 반복 순서를 어떻게 유지합니까?


다음과 같이 Map에서 생성하고 List있습니다.

List<String> strings = Arrays.asList("a", "bb", "ccc");

Map<String, Integer> map = strings.stream()
    .collect(Collectors.toMap(Function.identity(), String::length));

에서와 동일한 반복 순서를 유지하고 싶습니다 List. 방법을 LinkedHashMap사용하여 어떻게 만들 수 Collectors.toMap()있습니까?


2 개 매개 변수 버전은 다음을Collectors.toMap() 사용합니다 HashMap.

public static <T, K, U> Collector<T, ?, Map<K,U>> toMap(
    Function<? super T, ? extends K> keyMapper, 
    Function<? super T, ? extends U> valueMapper) 
{
    return toMap(keyMapper, valueMapper, throwingMerger(), HashMap::new);
}

4 개 매개 변수 버전 을 사용하려면 다음을 대체 할 수 있습니다.

Collectors.toMap(Function.identity(), String::length)

와:

Collectors.toMap(
    Function.identity(), 
    String::length, 
    (u, v) -> {
        throw new IllegalStateException(String.format("Duplicate key %s", u));
    }, 
    LinkedHashMap::new
)

또는 좀 더 깔끔하게 만들려면 새 toLinkedMap()메서드를 작성하고 다음을 사용하십시오.

public class MoreCollectors
{
    public static <T, K, U> Collector<T, ?, Map<K,U>> toLinkedMap(
        Function<? super T, ? extends K> keyMapper,
        Function<? super T, ? extends U> valueMapper)
    {
        return Collectors.toMap(
            keyMapper,
            valueMapper, 
            (u, v) -> {
                throw new IllegalStateException(String.format("Duplicate key %s", u));
            },
            LinkedHashMap::new
        );
    }
}

자신의 Supplier, AccumulatorCombiner:

List<String> myList = Arrays.asList("a", "bb", "ccc"); 
// or since java 9 List.of("a", "bb", "ccc");

LinkedHashMap<String, Integer> mapInOrder = myList
    .stream()
    .collect(
        LinkedHashMap::new,                                   // Supplier
        (map, item) -> map.put(item, item.length()),          // Accumulator
        Map::putAll);                                         // Combiner

System.out.println(mapInOrder);  //{a=1, bb=2, ccc=3}

참고 URL : https://stackoverflow.com/questions/29090277/how-do-i-keep-the-iteration-order-of-a-list-when-using-collections-tomap-on-a

반응형