从超类列表中提取给定子类的元素

Question

鉴于此代码

class Animal{}
class Dog extends Animal{}
class Cat extends Animal{}
class Pitbull extends Dog{}

object MyClass {
    def main(args: Array[String]) {
        val animals: List[Animal] = List(new Dog(), new Cat(), new Pitbull(), new Dog(), new Cat())
        getElementOftype(animals, PitBull)
    }
    
    def getElementOftype(list: List[Animal], givenType: ???): Animal = {
        
    }
}

我想从这个 List 中提取 Pitbull 类型的第一个元素，我应该如何处理？

我试过但感觉不对

trait Identity{
    def who: String
}

class Animal extends Identity{
    override def who = "Animal"
}


class Dog extends Animal with Identity{
    override def who = "Dog"
}


class Cat extends Animal with Identity{
    override def who = "Cat"
}

class Pitbull extends Dog with Identity{
    override def who = "Pitbull"
}


object MyClass {
    def main(args: Array[String]) {
        val animals: List[Animal] = List(new Dog(), new Cat(), new Pitbull(), new Dog(), new Cat())
        println(getElementOftype(animals, new Pitbull()))
    }
    
    def getElementOftype(list: List[Animal], givenType: Animal): Option[Animal] = {
        list.collectFirst{case s if s.who == givenType.who => s}
    }
}

我想简单地将Pitbull 作为参数传递，而不是实例化一个空对象。

这个方法可行，但我认为这不是最好的方法。 必须有一种更大规模的方式来做到这一点。

Answer 1

你可以这样做（未经测试，可能有一些错别字）：

def getElementOftype[A <: Animal](list: List[Animal])(implicit ct: ClassTag[A]): Option[A] = {
  list.collectFirst { case s: A => s }
}

ClassTag 是匹配collectFirst 中A 类型所必需的。

Answer 2

当您写 Animal extends Identity 时，您的意思是 Animal IS an Identity，这是错误的，最好使用这种方法：

class Animal { self: Identity => 
}

这意味着当你想实例化一个 Animal 时，你需要给它一个 Identity，但你为什么要这样做呢？ 您可以像这样简单地使用模式匹配来过滤所需的类型：

import scala.reflect.ClassTag
def getElementsOfType[A <: Animal : ClassTag](animals: List[Animal]): Option[A] = {
  animals.filter {
    case a: A => true
    case _ => false
  }.map(_.asInstanceOf[A]).headOption
}

headOption 是因为你的 api 返回 Option[A] 作为结果，你可以让它返回 List[A] 并去掉 headOption，以防列表中有多个 Pitbull :)