Idiomatic Scala Bản đồ upsert

Question

Tôi đang làm việc với một bản đồ trong Scala và làm bình thường "nếu không có giá trị gắn liền với một chìa khóa, tạo ra nó, đặt nó trong bản đồ và gửi lại":

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

Đối với tôi, điều này không giống như mã Scala thành ngữ. Nó giống như Java. Có cách viết ngắn gọn hơn không? Có vẻ như tôi có thể ghi đè lên Map.default() để chèn giá trị mới và trả lại nó. Không chắc chắn mặc dù.

Cảm ơn!

Answer 1

mutable.Map có getOrElseUpdate thực hiện chính xác những gì bạn muốn, không cần thành ngữ.