idiomático Scala Mapa upsert

Question

estoy trabajando con un mapa en la Scala y haciendo el habitual "si no hay un valor asociado a una clave, cree él, lo puso en el mapa y devolverlo":

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

Para mí, esto no se siente como un código Scala idiomático. Se siente como Java. ¿Hay alguna forma más concisa de escribir esto? Parecía que tal vez podría anular Map.default() para insertar el nuevo valor y devolverlo. Aunque no estoy seguro

Gracias!

Answer 1

mutable.Map tiene getOrElseUpdate que hace exactamente lo que quiere, no es necesario modismo idiomático.