我可以在Scala匹配语句中使用类变量吗?

时间:2010-11-05 11:11:02

标签: class scala

说我有这样的事情:

obj match {
    case objTypeOne : TypeOne => Some(objTypeOne)
    case objTypeTwo : TypeTwo => Some(objTypeTwo)
    case _ => None
}

现在我想概括一下,传入其中一种类型来匹配:

obj match {
    case objTypeOne : clazz => Some(objTypeOne)
    case objTypeTwo : TypeTwo => Some(objTypeTwo)
    case _ => None
}

但这是不允许的,我认为是出于句法而非语义的原因(虽然我猜也是即使clazz是一个Class [C],类型也会被删除,因此选项的类型将会丢失)。

我最终得到了:

if(clazzOne.isAssignableFrom(obj.getClass)) Some(clazz.cast(obj))
if(obj.isInstanceOf[TypeTwo]) Some(obj.asInstanceOf[TypeTwo])
None

我只是想知道是否有更好的方式。

3 个答案:

答案 0 :(得分:11)

您可以定义一个与您的对象匹配的提取器:

class IsClass[T: Manifest] {
  def unapply(any: Any): Option[T] = {
    if (implicitly[Manifest[T]].erasure.isInstance(any)) {
       Some(any.asInstanceOf[T])
    } else {
       None
    }
  }
}

让我们测试一下:

class Base { def baseMethod = () }
class Derived extends Base

val IsBase = new IsClass[Base]

def test(a:Any) = a match {
    case IsBase(b) => 
      println("base")
      b.baseMethod
    case _ => println("?")
  }

test(new Base)
test(1)

您必须为提取器定义val,例如,您不能内联IsBase。否则它将被解释为提取器。

答案 1 :(得分:5)

您可以使用图案防护来实现这一目标。尝试这样的事情:

obj match {
    case objTypeTwo : TypeTwo => Some(objTypeTwo)
    case objTypeOne if clazz.isAssignableFrom(objTypeOne.getClass) => Some(clazz.cast(objTypeOne))
    case _ => None
}

答案 2 :(得分:2)

您可以使用本地类型别名:

def matcher[T](obj: Any)(implicit man: Manifest[T]) = {
   val instance = man.erasure.newInstance.asInstanceOf[AnyRef]
   type T = instance.type // type alias
   obj match { 
      case objTypeOne : T => "a"
      case objTypeTwo : TypeTwo => "b"
      case _ => "c"
   }
}

scala> matcher[TypeOne](TypeOne())
res108: java.lang.String = a

scala> matcher[TypeTwo](TypeOne())
res109: java.lang.String = c

<击> 更新:Aaron Novstrup指出,单身人士类型只有man.erasure.newInstance==obj才有效(参见规范§3.2.1)