首页 文章

在Scala中创建自定义Seq集合

提问于
浏览 次
2

我想从现有的Seq集合中创建一个新的自定义Scala集合 . 我有一个名为Ref的特征，其中包含如下数据

trait Ref[A] {
   def get: A
   def getAsOption: Option[A]
   def dataType: Class[A]
   // other methods
}

我的自定义集合名为Vec，它是Ref [A]的序列（即Vec [A]等同于Seq [Ref [A]]），我想创建这样的自定义集合的原因是因为我想保留类型参数Ref中的集合在自定义方法中处理 . 我的代码如下

trait VecLike[A, +Repr <: Seq[Ref[A]]]
  extends SeqLike[Ref[A], Repr]
  with TraversableLike[Ref[A], Repr] {
// custom methods
}

trait GenVec[A]
  extends VecLike[A, Seq[Ref[A]]]
  with Seq[Ref[A]]
  with PartialFunction[Int, Ref[A]]

abstract class AbstractVec[A](vec: Ref[A]*)
  extends AbstractSeq[Ref[A]]
  with GenVec[A] {...}

class Vec(vec: Ref[A]*)
  extends AbstractVec[A](vec:_*)
  with VecLike[A, Vec[A]]
  with GenericTraversableTemplate[Ref[A], Seq]

但是当我调用map（）函数时

Vec(1,2,3,4).map(intToString)

它返回一个Seq [String]，预期的结果是Vec [String] . 我还尝试使用SeqFactory [Seq]在伴随对象中创建自定义CanBuildFrom但它失败了 . :( :(

任何人都可以给我一些建议，以及我如何实现这一目标？

scala scala-collections

2 回答

  • 3

    您需要实现自定义 Builder 并在 VecLike 中使 newBuilder 方法返回它 . 查看此页面以获得有关实现自定义集合的非常好的教程：http://docs.scala-lang.org/overviews/core/architecture-of-scala-collections.html

    回复于
  • 0

    我完成了，这是我的错误，我创建了错误的CanBuildFrom . 以下是CanBuildFrom的完整实现（CBF灵感来自Scala GenericCanBuildFrom）

    trait VecLike[A, +Repr]
  extends SeqLike[Ref[A], Repr]

trait GenVec[A]
  extends AnyRef
  with VecLike[A, GenVec[A]]

trait Vec[A]
  extends GenVec[A]
  with VecLike[A, Vec[A]]

abstract class AbstractVec[A](vec: Seq[Ref[A]])
  extends AnyRef
  with Vec[A]

final class VecImpl[A](vec: Seq[Ref[A]])
  extends AbstractVec[A](vec)

object Vec {


  private[this] val VecCBFInstance: VecCanBuildFrom[Nothing] = new VecCanBuildFrom[Nothing] {
    override def apply() = newBuilder[Nothing]
  }
  private def VecCBF: VecCanBuildFrom[Nothing] = VecCBFInstance
  private class VecCanBuildFrom[A] extends CanBuildFrom[Vec[_], A, Vec[A]] {
    def apply(from: Vec[_]) = newBuilder[A]
    def apply() = newBuilder[A]
  }
  implicit def canBuildFrom[A]: CanBuildFrom[Vec[_], A, Vec[A]] = VecCBF.asInstanceOf[VecCanBuildFrom[A]]

}

    现在我可以在调用map（）函数后接收Vec结果而不是Seq，如下所示

    val vec = Vec(1,2,3,4)
var vecStr : Vec[String] = vec.map(intToString)
    回复于

相关问题