繁体   English   中英

如何使用Java Streams为HashMap的多个键插入相同的值

[英]How to insert the same value for multiple keys of an HashMap using Java Streams

假设我有一个HashMap,我想将相同的值插入到一个键列表中。 如何在不迭代所有键并插入值的情况下使用Java 8执行此操作? 这更像是一个Java Streams问题。

这是直截了当的做法。 这是我编写的示例代码,用于演示我想要实现的目标。

public void foo(List<String> keys, Integer value) {
    Map<String, Integer> myMap = new HashMap<>(); 
    for (String key : keys) {
        myMap.put(key, value);
    }
}

有没有更简单的方法使用Java 8流进行上述操作? 如何避免使用Java 8流的for循环。 谢谢!

[Edit-1]下面是一个更好的代码片段。

public void foo() {
    Map<String, Integer> myMap = new HashMap<>(); 
    List<String> keys = getKeysFromAnotherFunction();
    Integer value = getValueToBeInserted(); // Difficult to show my actual use case. Imagine that some value is getting computed which has to be inserted for the keys.
    for (String key : keys) {
        myMap.put(key, value);
    }  

    List<String> keys2 = getNextSetOfKeys();
    Integer newValue = getValueToBeInserted(); 
    for (String key : keys2) {
        myMap.put(key, newValue);
    } 
}

使用收集器,如:

Map<String, Integer> myMap = keys.stream()
            .collect(Collectors.toMap(key -> key,
                    val -> value, (a, b) -> b));

我认为你的问题是将一些代码分解出来而不是将传统的for循环转换为流构造。

假设您具有以下通用实用程序方法:

public static <K, V, M extends Map<K, V>> M fillMap(
        Supplier<List<K>> keysFactory,
        Supplier<V> singleValueFactory,
        Supplier<M> mapFactory) {

    M map = mapFactory.get();
    List<K> keys = keysFactory.get();
    V singleValue = singleValueFactory.get();

    keys.forEach(k -> map.put(k, singleValue));

    return map;
}

然后,您可以使用以下方法:

Map<String, Integer> myMap = fillMap(() -> getKeysFromAnotherFunction(),
                                     () -> getValueToBeInserted(),
                                     HashMap::new); // create HashMap

myMap = fillMap(() -> getNextSetOfKeys(),
                () -> getValueToBeInserted(),
                () -> myMap); // use previously created map

上面的代码有变种,即该方法可以接收Map<K, V>实例而不是Supplier<Map<K, V>> ,或者甚至可能超载以支持两种变体。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM