繁体   English   中英

在 Scala 中创建自定义 Seq 集合

[英]Create custom Seq collection in Scala

我想从现有的Seq集合创建一个新的自定义 Scala 集合。 我有一个名为Ref的特征,它保存如下数据

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

我的自定义集合名为Vec ,它是一个Ref[A] Sequence (即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但它失败了。 :( :(

任何人都可以就此给我一些建议以及我如何实施以实现这一目标吗?

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

我完成了,那是我的错误我创建了错误的 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)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM