基于ClassTag的模式匹配对于基元失败

时间:2013-05-30 00:10:41

标签: scala reflection type-erasure

我认为以下将是收集满足给定类型的集合元素的最简洁和正确的形式:

def typeOnly[A](seq: Seq[Any])(implicit tag: reflect.ClassTag[A]): Seq[A] = 
  seq.collect {
    case tag(t) => t
  }

但这仅适用于AnyRef类型,而不适用于原语:

typeOnly[String](List(1, 2.3, "foo"))  // ok. List(foo)
typeOnly[Double](List(1, 2.3, "foo"))  // fail. List()

显然直接形式有效:

List(1, 2.3, "foo") collect { case d: Double => d }  // ok. List(2.3)

所以必须有(简单的!)方法来修复上述方法。

1 个答案:

答案 0 :(得分:19)

这是示例中的框,对吧?

scala> typeOnly[java.lang.Double](vs)
res1: Seq[Double] = List(2.3)

更新:oracle非常神秘:"boxing is supposed to be invisible, plus or minus"。我不知道这种情况是正还是负。

我的感觉是它是一个错误,因为否则它就是一个空洞的游戏。

更多Delphic异议:“我不知道这个例子预计会做什么。”请注意,未指定,由谁预期

这是一个很有用的练习,可以询问谁知道盒装,盒子是什么?这就好像编译器是一个魔术师正在努力隐藏一条线,这使得一张扑克牌悬浮在半空中,尽管每个观看者都知道必须有一根电线。

scala> def f[A](s: Seq[Any])(implicit t: ClassTag[A]) = s collect {
     | case v if t.runtimeClass.isPrimitive &&
     |   ScalaRunTime.isAnyVal(v) &&
     |   v.getClass.getField("TYPE").get(null) == t.runtimeClass =>
     |   v.asInstanceOf[A]
     | case t(x) => x
     | }
f: [A](s: Seq[Any])(implicit t: scala.reflect.ClassTag[A])Seq[A]

scala> f[Double](List(1,'a',(),"hi",2.3,4,3.14,(),'b'))
res45: Seq[Double] = List(2.3, 3.14)

ScalaRunTime不支持API - 对isAnyVal的调用只是对类型的匹配;也可以检查“TYPE”字段是否存在或

Try(v.getClass.getField("TYPE").get(null)).map(_ == t.runtimeClass).getOrElse(false)

但是为了回到一个漂亮的单行,你可以自己滚动ClassTag来处理特殊的提取。

2.11的版本。这可能不是最前沿,但它是最近烧灼的边缘。

object Test extends App {

  implicit class Printable(val s: Any) extends AnyVal {
    def print = Console println s.toString
  }

  import scala.reflect.{ ClassTag, classTag }
  import scala.runtime.ScalaRunTime

  case class Foo(s: String)

  val vs = List(1,'a',(),"hi",2.3,4,Foo("big"),3.14,Foo("small"),(),null,'b',null)

  class MyTag[A](val t: ClassTag[A]) extends ClassTag[A] {
    override def runtimeClass = t.runtimeClass
    /*
    override def unapply(x: Any): Option[A] = (
      if (t.runtimeClass.isPrimitive && (ScalaRunTime isAnyVal x) &&
          x.getClass.getField("TYPE").get(null) == t.runtimeClass)
        Some(x.asInstanceOf[A])
      else super.unapply(x)
    )
    */
    override def unapply(x: Any): Option[A] = (
      if (t.runtimeClass.isPrimitive) {
        val ok = x match {
          case _: java.lang.Integer   => runtimeClass == java.lang.Integer.TYPE
          //case _: java.lang.Double    => runtimeClass == java.lang.Double.TYPE
          case _: java.lang.Double    => t == ClassTag.Double  // equivalent
          case _: java.lang.Long      => runtimeClass == java.lang.Long.TYPE
          case _: java.lang.Character => runtimeClass == java.lang.Character.TYPE
          case _: java.lang.Float     => runtimeClass == java.lang.Float.TYPE
          case _: java.lang.Byte      => runtimeClass == java.lang.Byte.TYPE
          case _: java.lang.Short     => runtimeClass == java.lang.Short.TYPE
          case _: java.lang.Boolean   => runtimeClass == java.lang.Boolean.TYPE
          case _: Unit                => runtimeClass == java.lang.Void.TYPE
          case _ => false // super.unapply(x).isDefined
        }
        if (ok) Some(x.asInstanceOf[A]) else None
      } else if (x == null) {  // let them collect nulls, for example
        if (t == ClassTag.Null) Some(null.asInstanceOf[A]) else None
      } else super.unapply(x)
    )
  }

  implicit def mytag[A](implicit t: ClassTag[A]): MyTag[A] = new MyTag(t)

  // the one-liner
  def g[A](s: Seq[Any])(implicit t: ClassTag[A]) = s collect { case t(x) => x }

  // this version loses the "null extraction", if that's a legitimate concept
  //def g[A](s: Seq[Any])(implicit t: ClassTag[A]) = s collect { case x: A => x }

  g[Double](vs).print
  g[Int](vs).print
  g[Unit](vs).print
  g[String](vs).print
  g[Foo](vs).print
  g[Null](vs).print
}

对于2.10.x,额外的样板线,因为隐式分辨率是 - 好吧,我们不会说它已经坏了,我们只是说它不起作用。

// simplified version for 2.10.x   
object Test extends App {
  implicit class Printable(val s: Any) extends AnyVal {
    def print = Console println s.toString
  }
  case class Foo(s: String)

  val vs = List(1,'a',(),"hi",2.3,4,Foo("big"),3.14,Foo("small"),(),null,'b',null)

  import scala.reflect.{ ClassTag, classTag }
  import scala.runtime.ScalaRunTime

  // is a ClassTag for implicit use in case x: A
  class MyTag[A](val t: ClassTag[A]) extends ClassTag[A] {
    override def runtimeClass = t.runtimeClass
    override def unapply(x: Any): Option[A] = (
      if (t.runtimeClass.isPrimitive && (ScalaRunTime isAnyVal x) &&
          (x.getClass getField "TYPE" get null) == t.runtimeClass)
        Some(x.asInstanceOf[A])
      else t unapply x
    )
  }

  // point of the exercise in implicits is the type pattern.
  // there is no need to neutralize the incoming implicit by shadowing.
  def g[A](s: Seq[Any])(implicit t: ClassTag[A]) = {
    implicit val u = new MyTag(t)  // preferred as more specific
    s collect { case x: A => x }
  }

  s"Doubles? ${g[Double](vs)}".print
  s"Ints?    ${g[Int](vs)}".print
  s"Units?   ${g[Unit](vs)}".print
  s"Strings? ${g[String](vs)}".print
  s"Foos?    ${g[Foo](vs)}".print
}

宣传评论:

@WilfredSpringer有人听到了你的话。 SI-6967