Java：添加到集合的地圖

Question

我正在嘗試編寫一個通用函數，它將一個元素添加到Collection of Map中。 這適用於一個Map的List S：

public static <TKey, TVal> void addToMapOfLists(Map<TKey, List<TVal>> map, TKey key, TVal val) {
    List<TVal> list = map.get(key);
    if (list == null) {
        list = new ArrayList<>();
        list.add(val);
        map.put(key, list);
    } else
        list.add(val);
}

我想讓這個函數在Map<TKey, Set<TVal>>以及Map<TKey, List<TVal>> 。 我希望這應該是可能的，因為它們都實現了具有我調用的add(TVal)成員的Collection 。

我的問題是，當我嘗試將參數Map<TKey, List<TVal>> map更改為Map<TKey, ? extends Collection<TVal>> map Map<TKey, ? extends Collection<TVal>> map - 我需要以某種方式替換new ArrayList<>(); 調用Collection的實現者的構造函數。

Answer 1

您必須將另一個參數傳遞給您的方法 - Collection實例Supplier 。

這是一個可能的實現：

public static <TKey, TVal, TCol extends Collection<TVal>> void addToMapOfCollections(Map<TKey, Collection<TVal>> map, TKey key, TVal val, Supplier<TCol> supplier) 
{
    Collection<TVal> col = map.get(key);
    if (col == null) {
        col = supplier.get ();
        col.add(val);
        map.put(key, col);
    } else {
        col.add(val);
    }
}

這是另一種選擇：

public static <TKey, TVal> void addToMapOfCollections(Map<TKey, Collection<TVal>> map, TKey key, TVal val, Supplier<Collection<TVal>> supplier)
{
    Collection<TVal> col = map.get(key);
    if (col == null) {
        col = supplier.get ();
        col.add(val);
        map.put(key, col);
    } else {
        col.add(val);
    }
}

並且代碼更少（Gerald建議）：

public static <TKey, TVal> void addToMapOfCollections(Map<TKey, Collection<TVal>> map, TKey key, TVal val, Supplier<Collection<TVal>> supplier)
{
    map.putIfAbsent(key, supplier.get());
    map.get(key).add(val);
}

我測試了第二個變種：

Map<String,Collection<Integer>> map = new HashMap<String, Collection<Integer>>();
addToMapOfCollections(map,"String1",5,HashSet::new);
addToMapOfCollections(map,"String2",67,ArrayList::new);
addToMapOfCollections(map,"String2",68,ArrayList::new);
System.out.println (map);
for (Collection<Integer> col : map.values ()) {
    System.out.println (col.getClass () + " : " + col);
}

輸出：

{String2=[67, 68], String1=[5]}
class java.util.ArrayList : [67, 68]
class java.util.HashSet : [5]