扩展内置集合,内置方法的问题

时间:2010-08-12 22:09:19

标签: scala traits

我是Scala新手,请原谅我,如果这是一个愚蠢的问题,但这里有...

想象一下,我希望创建一个包含其他方法的扩展Map类型。我可以看到几种方法来做到这一点。第一个是组成:

class Path[V](val m: Map[V, Int]) {
  // Define my methods
}

另一种是通过继承,例如

class Path[V] extends Map[V, Int] {
// Define my methods
}

最后,我还考虑了“特质”路线,例如

trait Path[V] extends Map[V, Int] {
// Define my methods
}

组合有点尴尬,因为你经常不得不引用里面的东西。遗传是相当自然的,但我有一个皱纹(更多的是一秒)。 Traits似乎是一种非常优雅的方式来实现这一点并且使用“with”构造它非常好但是它对我来说也有问题。

我遇到的皱纹是用像++这样的方法。他们返回一张新地图。所以,让我们说上面提到的“我的方法”希望在地图上添加一些东西(只是一个例子,我知道地图已经有了这个),例如。

trait Path[V] extends Map[V,Int] {
    def addOne(v: V, i: Int): Path[V] = this + (v -> i)
}

这会产生错误,因为返回类型不是Path [V]。现在我知道我可以在新实例上使用“with”来添加Path [V]特征。但我不控制这里新地图的构造。有没有办法添加Path [V]特征?我考虑创建一个预先填充的新不可变映射,然后在“with Path [V]”上进行标记,但是没有这样的构造函数可以用来创建预先填充的映射。

我怀疑(虽然我还没有证实)我会使用继承类似的问题。我可以添加一个新方法来向地图添加一个新条目,但我不会得到一个“Path [V]”这就是我想要的。组合方法似乎是走到这里的唯一方法。

我希望这很清楚。评论

1 个答案:

答案 0 :(得分:6)

也许最简单的方法是使用MapProxy特征:

import collection._

class Path[V](val self: Map[V, Int]) extends MapProxy[V, Int] {
   // without the companion object below
   def addOne(v: V, i: Int): Path[V] = new Path(this + (v -> i))
   // with the companion object below
   def addOne(v: V, i: Int): Path[V] = this + (v -> i)
   // other methods
}

// this companion object is not strictly necessary, but is useful:
object Path {
   implicit def map2path[V](map: Map[V, Int]): Path[V] = new Path(map)
}

通过允许您将Path视为Map来消除组合方法的尴尬:

scala> new Path(Map("a" -> 1)) + ("b" -> 2)
res1: scala.collection.Map[java.lang.String,Int] = Map((a,1), (b,2))

scala> new Path(Map("a" -> 1)).get("a")
res2: Option[Int] = Some(1)

如果您包含从MapPath的隐式转化(在Path随播广告对象中定义),那么您也可以将Map视为Path

scala> Map("a" -> 1).addOne("b", 2)
res3: Path[java.lang.String] = Map((a,1), (b,2))

SeqProxy添加行为有类似的Seq特征。

在更一般的情况下,解决方案是使用pimp my library pattern。这个related question有一个例子。