有什么干净的方法可以在 Scala 中结合 find 和 instanceof 吗？

Question

我想在一些 Iterable 中找到一些符合某种给定类型的元素，并验证将该类型作为参数的谓词。

我用命令式编程的方式写了这个方法，看起来符合我的预期。有没有办法以更“scalaesque”的方式写这个？

def findMatch[T](it: Iterable[_], clazz: Class[T], pred: T => Boolean): Option[T] = {
  val itr = it.iterator
  var res: Option[T] = None
  while (res.isEmpty && itr.hasNext) {
    val e = itr.next()
    if (clazz.isInstance(e) && pred(clazz.cast(e))) {
      res = Some(clazz.cast(e))
    }
  }
  res
}

Answer 1

如果您想查找，然后映射，您可以使用collect。

scala> val it: Iterable[Any] = List(1,2,3,"4")            
it: Iterable[Any] = List(1, 2, 3, 4)

scala> it.view.collect{case s: String => s}.headOption
res1: Option[String] = Some(4)

Answer 2

您可以使用存在类型 X forSome{typeX} 而不是使用 _ 作为类型参数。然后，您可以使用提到的 find 方法编写它，并在 Option 类型上使用 map 方法：

def findMatch[T](it: Iterable[X forSome {type X}], clazz: Class[T], pred: T => Boolean): Option[T] = {
    it.find{ e => clazz.isInstance(e) && pred(clazz.cast(e))}.map{clazz.cast(_)}
}

Answer 3

如果您将问题划分为子问题，则很容易找到更惯用的版本。
您希望

1. 在 Iterable[Any] 中查找 T 的所有实例，
2. 将它们强制转换为 T 以使编译器很乐意
3. 找到第一个匹配元素

For第一点，您可以轻松地在 Iterator 上使用 filter 方法。因此，您将

it.iterator.filter(x => clazz.isInstance(x))

返回一个仅包含 T 的 Iterator[Any] 。现在让我们说服编译器：

it.iterator.filter(x => clazz.isInstance(x)).map(x => x.asInstanceOf[T])

好的，现在你有了一个 Iterator[T] ，所以你只需要找到满足你的谓词的第一个元素：

def findMatch[T](it: Iterable[Any], clazz: Class[T], pred: T => Boolean): Option[T] = 
  it.iterator.filter(x => clazz.isInstance(x))
             .map(x => x.asInstanceOf[T])
             .find(pred)

Answer 4

您可以使用 Iterable 的 find 方法和带有守卫的模式匹配：

scala> val it: Iterable[Any] = List(1,2,3,"4")
it: Iterable[Any] = List(1, 2, 3, 4)

scala> it.find { _ match {
  case s: String if s == "4" => true
  case _ => false
}}.asInstanceOf[Option[String]]
res0: Option[String] = Some(4)

有关模式匹配的介绍，请查看：
http://programming-scala.labs.oreilly.com/ch03.html