putIfAbsent 用于不可变映射？

Question

我经常发现自己在编写这样的代码：

import scala.collection.immutable._

var foos = Map.empty[Int, Foo]

def fooOf(n: Int): Foo = {
  foos.get(n) match {
    case Some(foo) => foo
    case None =>
      val foo = new Foo(n)
      foos = foos.updated(n, foo)
      foo
  }
}

使用并发映射，代码看起来更漂亮：

import scala.collection.concurrent._

var foos = TrieMap.empty[Int, Foo]

def fooOf(n: Int): Foo = {
  val foo = new Foo(n)
  foos.putIfAbsent(n, foo) getOrElse foo
}

但我真的想要一个非并发、不可变的映射。有类似putIfAbsent 的东西吗？

（顺便说一句，在并发示例中，是否只有在确实不存在时才能创建Foo？）

Answer 1

如果您对可变变量没问题（并且您在两个示例中都使用var foos 并且只指定了一个不可变集合），这应该符合您的需求：

var foos = Map.empty[Int, Foo]

def fooOf(n: Int): Foo = {
  foos.getOrElse(n, {
    foos = foos.updated(n, new Foo(n))
    foos(n)
  })
}