我发现自己使用了很多嵌套映射,例如 Map[Int、Map[String、Set[String]]],并且我希望在访问新密钥时自动创建新的 Map、Set 等。例如。像下面这样:
val m = ...
m(1992)("foo") += "bar"
请注意,如果不需要,我不想在此处使用 getOrElseUpdate,因为当您有嵌套映射并掩盖代码中实际发生的情况时,它会变得非常冗长:
m.getOrElseUpdate(1992, Map[String, Set[String]]()).getOrElseUpdate("foo", Set[String]()) ++= "bar"
所以我重写了 HashMap 的“默认”方法。我尝试了两种方法来做到这一点,但都不是完全令人满意。我的第一个解决方案是编写一个创建映射的方法,但似乎在声明变量时我仍然必须指定完整的嵌套映射类型,否则事情将不起作用:
scala> def defaultingMap[K, V](defaultValue: => V): Map[K, V] = new HashMap[K, V] { | override def default(key: K) = {
| val result = defaultValue
| this(key) = result
| result
| }
| }
defaultingMap: [K,V](defaultValue: => V)scala.collection.mutable.Map[K,V]
scala> val m: Map[Int, Map[String, Set[String]]] = defaultingMap(defaultingMap(Set[String]()))
m: scala.collection.mutable.Map[Int,scala.collection.mutable.Map[String,scala.collection.mutable.Set[String]]] = Map()
scala> m(1992)("foo") += "bar"; println(m)
Map(1992 -> Map(foo -> Set(bar)))
scala> val m = defaultingMap(defaultingMap(Set[String]()))
m: scala.collection.mutable.Map[Nothing,scala.collection.mutable.Map[Nothing,scala.collection.mutable.Set[String]]] = Map()
scala> m(1992)("foo") += "bar"; println(m)
<console>:11: error: type mismatch;
found : Int(1992)
required: Nothing
m(1992)("foo") += "bar"; println(m)
^
我的第二个解决方案是编写一个带有方法的工厂类,这样我只需声明每种类型一次。但是每次我想要一个新的默认值映射时,我都必须实例化工厂类,然后调用该方法,这看起来仍然有点冗长:
scala> class Factory[K] {
| def create[V](defaultValue: => V) = new HashMap[K, V] {
| override def default(key: K) = {
| val result = defaultValue
| this(key) = result
| result
| }
| }
| }
defined class Factory
scala> val m = new Factory[Int].create(new Factory[String].create(Set[String]()))
m: scala.collection.mutable.HashMap[Int,scala.collection.mutable.HashMap[String,scala.collection.mutable.Set[String]]] = Map()
scala> m(1992)("foo") += "bar"; println(m)
Map(1992 -> Map(foo -> Set(bar)))
我真的很想要像这样简单的东西:
val m = defaultingMap[Int](defaultingMap[String](Set[String]()))
有人看到有办法做到这一点吗?