Question

我想创建一个包含HashSets作为值的HashMap，并在找不到键时返回一个空的HashSet。

public class IsbnHashMap<K,V> extends HashMap<K,V> {
    protected V defaultValue;
    public IsbnHashMap(V defaultValue) {
        this.defaultValue = defaultValue;
    }
    @Override
    public V get(Object k) {
        return containsKey(k) ? super.get(k) : defaultValue;
    }
}

但是我的实施不起作用。

private static IsbnHashMap<String, HashSet<String>> isbnToId = new IsbnHashMap<String, HashSet<String>>();

返回“无法应用HashSet”。如果我尝试将K,V中的IsbnHashMap更改为<String, HashSet<String>>，我也会遇到一些时髦的错误。我该如何实现呢？

Answer 1

首先应该注意的是，在Java-8中你可以使用：

isbnToId.computeIfAbsent(isbn, k -> new HashSet<>()).add(_id);

其次，如果你真的想在以前的Java版本中做这样的事情，你最好为此目的创建单独的方法（例如，getOrDefault()），以免违反合同。第三，您需要为每个新密钥创建new HashSet<>()。如果返回相同的实例，它将在给定的键之间共享。如果您不希望用户修改它，最好使用不可修改的Collections.emptySet()作为默认值。这样用户可以安全地执行isbnToId.getOrDefault(isbn).contains(_id)，但尝试isbnToId.getOrDefault(isbn).add(_id)会导致异常。如果您想支持修改（在Java-8之前），您可以将元素类传递给构造函数：

public static class MyMap<K, V> extends HashMap<K, V> {
    private Class<?> clazz;

    public MyMap(Class<?> clazz) {
        this.clazz = clazz;
    }

    public V getOrCompute(K key) {
        V v = get(key);
        if(v == null) {
            try {
                v = (V) clazz.newInstance();
            } catch (InstantiationException | IllegalAccessException e) {
                throw new RuntimeException(e);
            }
            put(key, v);
        }
        return v;
    }
}

用法示例：

MyMap<String, Set<String>> map = new MyMap<>(HashSet.class);
map.getOrCompute("a").add("b");
map.getOrCompute("a").add("c");
map.getOrCompute("d").add("e");
System.out.println(map); // {a=[b, c], d=[e]}

这里我们假设使用默认构造函数实例化传递的类是可以的。另一种方法是传递能够产生默认值的工厂接口。

Answer 2

正如Jon Skeet所说......

private static IsbnHashMap<String, HashSet<String>> isbnToId = new IsbnHashMap<String, HashSet<String>>(new HashSet<String>());

...但是，这将返回与Dunni指出的相同的默认对象。

所以这样做：

private static HashMap<String, HashSet<String>> isbnToId = new HashMap<String, HashSet<String>>();

public static void coupleIsbnToId(String isbn, String _id) {
    if (!isbnToId.containsKey(isbn)) {
        isbnToId.put(isbn, new HashSet<String>());
    }
    isbnToId.get(isbn).add(_id);
}

扩展HashMap以返回未找到键的空HashSet

2 个答案: