在Scala集中覆盖toString

时间:2013-03-09 04:38:25

标签: scala tostring scala-collections

我想创建一组名为IntSet的整数。 IntSet在各方面都与Set[Int]完全相同,只是它的toString函数以逗号分隔的方式打印元素(就像你调用mkString(",")一样),它有一个采用Traversable整数的构造函数。最简单的方法是什么?

> IntSet((1 to 3)).toString
1,2,3

我认为会有一些单行方式来做到这一点,但我一直在摆弄隐式函数并扩展HashSet而我无法弄明白。


诀窍是使用代理对象。 Eastsun的答案如下。这是一个稍微不同的版本,它定义了一个命名的IntSet类型并使其成为不可变的。

import collection.immutable.{HashSet, SetProxy}

class IntSet(values: Traversable[Int]) extends SetProxy[Int] {
  override val self: Set[Int] = HashSet(values.toSeq:_*)
  override def toString() = mkString(",")
}

1 个答案:

答案 0 :(得分:8)

scala> import scala.collection.mutable
import scala.collection.mutable

scala> def IntSet(c: Traversable[Int]): mutable.Set[Int] = new mutable.SetProxy[Int] {
     |   override val self: mutable.Set[Int] = mutable.HashSet(c.toSeq :_*)
     |   override def toString = mkString(",")
     | }
IntSet: (c: Traversable[Int])scala.collection.mutable.Set[Int]

scala> IntSet(1 to 3)
res0: scala.collection.mutable.Set[Int] = 1,2,3