for构造中的scala模式匹配

Question

我有下面的数据模型，稍后我将对其进行模式匹配：

abstract class A
case class C(s:String) extends A

abstract class B extends A
case class D(i:Int) extends B
case class E(s:Int, e:Int) extends B

A是层次结构的抽象超类型。C是a的具体子类。a的其他具体子类是B的子类，B又是a的子类。

现在如果我写这样的东西，它是有效的：

def match(a:A) a match {
   a:C => println("C")
   a:B => println("B")
}

然而，在for循环中，我无法与B匹配。我假设我需要一个构造函数模式，但由于B是抽象的，所以B没有构造函数模式。

val list:List[A] = List(C("a"), D(1), E(2,5), ...)

for (b:B <- list) println(b)  // Compile error
for (b@B <- list) println(b)  // Compile error

这里，我只想打印B个实例。这个案子有解决办法吗？

Answer 1

如果你问我，你不能在这里使用模式匹配的事实是scala的一个不幸的不一致性。事实上，scala确实可以让您在理解上进行模式匹配，如本例所示：

val list:List[A] = List(C("a"), D(1), E(2,5)
for ((b:B,_) <- list.map(_ -> null)) println(b)

在这里，我暂时将元素包装成对（其中有一个虚拟的和未使用的第二个值），然后为第一个元素类型为B的一对进行模式匹配。

 D(1)
 E(2,5)

这样看来，scala确实支持基于模式匹配的过滤（即使是按类型匹配），只是语法似乎不处理按类型匹配单个元素的模式。

显然，我不是建议使用这个技巧，这只是为了说明。使用collect当然更好。

如果出于某种原因，你真的更喜欢理解，那么还有一个更普遍的解决方案：

object Value {
  def unapply[T]( value: T ) = Some( value )
}
for ( Value(b:B) <- list ) println(b)

我们刚刚在Value对象中引入了一个虚拟提取器，它什么也不做，因此Value（b:b）与justb:b具有相同的含义，只是前者进行编译。与我之前的配对技巧不同，它相对可读，而且Value只需编写一次，然后你就可以随意使用它（特别是，不需要为你想要匹配的每种类型编写一个新的提取器，如@Faiz的回答。不过，我会让你找到一个比Value更好的名称。

最后，还有一个现成的解决方案（归功于丹尼尔·索布拉尔），但可读性稍差，需要一个虚拟标识符（这里是foo）：

for ( b @(foo:B) <- list ) println(b)
// or similarly:
for ( foo @(b:B) <- list ) println(b)

Answer 2

谢尔盖是对的；如果你想进行模式匹配并只过滤B实例，你就必须放弃。如果出于任何原因，你仍然想使用进行理解，我认为一种方法就是使用警卫：

for (b <- list if b.isInstanceOf[B]) println(b)

但最好选择模式匹配，而不是isInstanceOf。因此，我会使用收集建议（如果它在我的其他代码中有意义的话）。

另一个建议是定义一个同名的B伴生对象，并定义unapply方法：

abstract class A
case class C(s:String) extends A

abstract class B extends A
object B { def unapply(b: B) = Option(b)  } // Added a companion to B
case class D(i:Int) extends B
case class E(s:Int, e:Int) extends B

然后你可以这样做：

for (B(b) <- list) println(b)

这不是B的“构造函数”，而是它的方法。这很有效，这就是朋友的作用，对吧？

（见http://www.scala-lang.org/node/112 )

Answer 3

可以使用收集：

list.collect { case b: B => println(b) }

如果你想更好地理解这一点，我建议你阅读部分函数。这里举个例子。