Java map.get(key) - 自动执行 put(key) 并在 key 不存在时返回？

Question

我厌倦了以下模式：

value = map.get(key);
if (value == null) {
    value = new Object();
    map.put(key, value);
}

这个示例仅涉及当您使用嵌套映射来表示多维结构时要编写的额外代码的表面。

我确信某个地方存在可以避免这种情况的东西，但我的谷歌搜索努力没有产生任何相关结果。有什么建议吗？

Answer 1

现有

java.util.concurrent.ConcurrentMap 

Java 8 中的 and

Java.util.Map

返回

putIfAbsent(K key, V value) 

值，如果为 null，则插入给定值。因此，如果键不存在值，则返回 null 并插入给定值，否则返回现有值

如果您需要对值进行惰性评估，则可以使用

computeIfAbsent(K key, Function<? super K,? extends V> mappingFunction)

Answer 2

Java 8 向 Map 添加了很好的方法： 计算, computeIfPresent< /a>, computeIfAbsent

要实现您所需要的：

Object existingOrCreated = map.computeIfAbsent(key, (k) -> new Object());

Answer 3

这种模式的问题在于，您必须以某种方式定义在 get() 返回 null 时应使用的值。

当然有图书馆，IIRC 也有一些较新的馆藏可以做到这一点，但不幸的是我不记得那些是哪些。

但是，您可以自己编写实用程序方法，前提是您有创建新值的标准方法。像这样的事情可能会起作用：

public static <K, V> V safeGet(K key, Map<K,V> map, Class<V> valueClass) throws /*add exceptions*/ {
  V value = map.get(key);
  if( value == null ) {
    value = valueClass.newInstance();
    map.put( key, value );
  }   

  return value;
} 

请注意，您要么必须抛出反射异常，要么在方法中处理它们。此外，这需要 valueClass 提供无参构造函数。或者，您可以简单地传递应使用的默认值。

Java 8 更新

在其他答案中已经提到过，但为了完整起见，我也会在此处添加信息。

从 Java 8 开始，有默认方法computeIfAbsent(key,mappingFunction)，它基本上做同样的事情，例如，如果值类是BigDecimal，它可能看起来像这样

BigDecimal value = map.computeIfAbsent(key, k -> new BigDecimal("123.456"));

：该方法类似于上面定义的 safeGet(...) 但更灵活，可以直接在地图实例上使用并且经过更好的测试。因此，如果可能的话，我建议改用computeIfAbsent()。

Answer 4

您可以使用 MutableMap 和 Eclipse Collections 中的 getIfAbsentPut() ，它返回映射到键或插入件给定值，如果没有值映射到键则返回给定值。

您可以使用方法引用来创建一个新的对象：

MutableMap<String, Object> map = Maps.mutable.empty();
Object value = map.getIfAbsentPut("key", Object::new);

或者您可以直接创建一个新的对象：

MutableMap<String, Object> map = Maps.mutable.empty();    
Object value = map.getIfAbsentPut("key", new Object());

在第一个示例中，仅当存在时才会创建该对象没有值映射到键。

在第二个示例中，无论如何都会创建对象。

注意：我是 Eclipse Collections 的贡献者。

Answer 5

如果在任何情况下您需要在地图中获取默认数据（如果它不存在）

map.getOrDefault(key, defaultValue);

javadocs

Answer 6

编辑：请注意，下面提到的功能已被长期弃用，并且 CacheBuilder。

Guava 库有一个“计算图”，请参阅MapMaker.makeComputingMap（函数）。

Map<String, Object> map = new MapMaker().makeComputingMap(
    new Function<String, Object>() {
      public String apply(Stringt) {
        return new Object();
      }
  });

如果您多次需要该 Function，请将其提取到实用程序类中，然后像这样创建 Map（其中 MyFunctions.NEW_OBJECT 是静态 Function 实例）：

Map<String, Object> map = new MapMaker()
    .makeComputingMap(MyFunctions.NEW_OBJECT);

Answer 7

也许我没有看到整个问题，但是使用继承或组合将此行为添加到 Map 对象怎么样？