Scala泛型类型匹配

时间:2014-12-11 19:36:42

标签: scala generics casting

我经常发现自己正在进行模式匹配,比如这个

val foo: Foo = ...
foo match {
  case bar: Bar => Some(bar)
  case _ => None
}

我想缩短它。

val foo: Foo = ...
optInstance[Foo, Bar](foo)

可能性1 - 直接翻译

def optInstance[A, B <: A](a: A) = a match {
  case b: B => Some(b)
  case _ => None
}
// warning: abstract type pattern B is unchecked since it is eliminated by erasure

可能性2 - 尝试抓住

def optInstance[A, B <: A](a: A) =
  try {
    Some(a.asInstanceOf[B])
  } catch {
    case e: ClassCastException => None
  }
// no warnings

Possiblility 3 - if-else

def optInstance[A, B <: A](a: A) =
  if(a.isInstanceOf[B]) {
    Some(a.asInstanceOf[B])
  } else {
    None
  }
// no warnings

它们都不起作用。 (Scala 2.11.2)


是否有通用的写作方式

foo match {
  case bar: Bar => Some(bar)
  case _ => None
}

(如果没有,是否至少有一个较短的方式?)

3 个答案:

答案 0 :(得分:4)

只需为首次实施添加隐式ClassTag

import scala.reflect.ClassTag

def optInstance[A, B <: A : ClassTag](a: A): Option[B] = a match {
  case b: B => Some(b)
  case _ => None
}

示例:

sealed trait Foo
case object Bar extends Foo
case object Baz extends Foo

scala> optInstance[Foo, Bar.type](Bar)
res4: Option[Bar.type] = Some(Bar)

scala> optInstance[Foo, Bar.type](Baz)
res5: Option[Bar.type] = None

答案 1 :(得分:2)

使用隐式值类的清洁方式:

 implicit class AsOpt[A](val a: A) extends AnyVal {
   def asOpt[B <: A : scala.reflect.ClassTag]: Option[B] = a match {
     case b: B => Some(b)
     case _    => None
   }
 }

示例:

 val seq: Seq[Int] = Seq.empty
 seq.asOpt[List[Int]]   // Option[List[Int]] = Some(List())
 seq.asOpt[Vector[Int]] // Option[Vector[Int]] = None

答案 2 :(得分:1)

如果没有通用功能,您可以使用Optioncollect

class Foo
class Bar extends Foo
class Baz extends Foo

scala> val foo: Foo = new Bar
scala> Some(foo).collect { case b: Bar => b }
res1: Option[Bar] = Some(Bar@483edb6b)

scala> val baz: Foo = new Baz
scala> Some(baz).collect { case b: Bar => b }
res3: Option[Bar] = None