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

Question

我想创建一个包含 HashSet 作为值的 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

正如乔恩·斯基特所说...

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