Idiomatic Scala Karte Upsert

Question

Ich arbeite mit einer Karte in der Scala und dabei die üblichen „wenn kein Wert mit einem Schlüssel zugeordnet ist, erstellen Sie es, es in der Karte setzen und sie zurückbringen“:

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

Das fühlt sich für mich nicht wie idiomatischer Scala-Code an. Es fühlt sich an wie Java. Gibt es eine prägnantere Art, dies zu schreiben? Es sah so aus, als könnte ich Map.default() überschreiben, um den neuen Wert einzufügen und zurückzugeben. Nicht sicher, obwohl.

Danke!

Answer 1

mutable.Map hat getOrElseUpdate was genau das tut, was Sie wollen, kein Idiom notwendig.