在流上使用Collections.toMap()时,如何保持List的迭代顺序?


82

我创建一个MapList如下:

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()方法创建一个?


1
请在下面检查我的答案。它使用的是自定义的只有4行代码SupplierAccumulatorCombinercollect你的方法stream:)
hzitoun

1
这个问题已经被回答了,我只想为找到这个问题的答案铺平道路。要求一张地图。因此,在需要Map的地方使用LinkedHashMap。
Satyendra Kumar,

Answers:


116

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
        );
    }
}

2
为什么这么复杂?您可以轻松地做到这一点,请在下面检查我的答案
hzitoun

1
mergeFunction在4参数版本中,Collectors.toMap()它不知道要合并的键,值uv值。因此,有关IllegalStateException的消息并不正确。
MoonFruit,

1
@hzitoun,因为如果您仅使用Map::put,则最终会(可能)为同一键使用不同的值。通过使用Map::put,您间接选择遇到的第一个值不正确。那是最终用户想要的吗?你确定吗?如果是,则确定使用Map::put。否则,您将不确定并不能决定:让用户知道他们的流已映射到具有不同值的两个相同键。我会对您自己的答案发表评论,但目前已被锁定。
奥利维尔·格雷戈尔

68

提供您自己的SupplierAccumulator以及Combiner

    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}

1
@Sushil接受的答案允许重用逻辑
cylinder.y19年

1
流依赖于诸如map.put(..)错误之类的副作用。
Nikolas Charalambidis

你知道什么更快吗?使用的是您的版本还是接受的答案?
nimo23

@Nikolas您能解释一下副作用的意思吗?事实上,我必须决定哪一个选择问题:stackoverflow.com/questions/61479650/...
nimo23

0

在Kotlin,toMap()是保留订单的。

fun <K, V> Iterable<Pair<K, V>>.toMap(): Map<K, V>

返回一个新映射,其中包含给定对中的所有键值对。

返回的映射保留原始集合的输入迭代顺序。如果两对中的任何一个具有相同的密钥,则最后一个将添加到映射中。

这是它的实现:

public fun <K, V> Iterable<Pair<K, V>>.toMap(): Map<K, V> {
    if (this is Collection) {
        return when (size) {
            0 -> emptyMap()
            1 -> mapOf(if (this is List) this[0] else iterator().next())
            else -> toMap(LinkedHashMap<K, V>(mapCapacity(size)))
        }
    }
    return toMap(LinkedHashMap<K, V>()).optimizeReadOnlyMap()
}

用法很简单:

val strings = listOf("a", "bb", "ccc")
val map = strings.map { it to it.length }.toMap()

的基础集合mapLinkedHashMap(按插入顺序排列)。


0

通过某个字段映射对象数组的简单函数:

public static <T, E> Map<E, T> toLinkedHashMap(List<T> list, Function<T, E> someFunction) {
    return list.stream()
               .collect(Collectors.toMap(
                   someFunction, 
                   myObject -> myObject, 
                   (key1, key2) -> key1, 
                   LinkedHashMap::new)
               );
}


Map<String, MyObject> myObjectsByIdMap1 = toLinkedHashMap(
                listOfMyObjects, 
                MyObject::getSomeStringField()
);

Map<Integer, MyObject> myObjectsByIdMap2 = toLinkedHashMap(
                listOfMyObjects, 
                MyObject::getSomeIntegerField()
);
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.