scala - 惯用的 Scala Map upsert

Question

我正在 Scala 中使用 map 并执行通常的“如果没有与键关联的值，则创建它，将其放入 map 并返回它”:

def alphaMemory(key : AlphaMemoryKey) = {
    var am = map.getOrElse(key, null)
    if(am == null) {
        am = new AlphaMemory(key)
        map.put(key, am)
    }
    am
}

对我来说，这不像是惯用的 Scala 代码。感觉像Java。有没有更简洁的写法？看起来也许我可以覆盖 Map.default() 以插入新值并返回它。虽然不确定。

谢谢!

Answer 1

mutable.Map 有 getOrElseUpdate 这正是你想要的，不需要成语。