在第一次出现后打破 flatMap

Question

abstract class Node(id: Long, name: String) {
  def find(id: Long) = if (this.id == id) Some(this) else None
}
case class Contact(id: Long, name: String, phone: String) extends Node(id, name)
case class Group(id: Long, name: String, entries: Node*) extends Node(id, name) {
  override def find(id: Long): Option[Node] = {
    super.find(id) orElse entries.flatMap(e => e.find(id)).headOption
  }
}
val tree =
  Group(0, "Root"
      , Group(10, "Shop", Contact(11, "Alice", "312"))
      , Group(20, "Workshop"
          , Group(30, "Tyres", Contact(31, "Bob", "315"), Contact(32, "Greg", "319"))
          , Contact(33, "Mary", "302"))
      , Contact(1, "John", "317"))
println(tree.find(32))

树数据由 Contacts 和 Groups（包含子组和联系人）构建。我想找到具有特定 id 的节点。目前我遍历 Group 的成员使用：

entries.flatMap(e => e.find(id)).headOption

但这不是最佳选择，因为我检查所有子条目而不是在第一次发现时中断。

非常感谢你在魔术方面的帮助 Scala Wold。谢谢

Answer 1

您需要 collectFirst，这将 select 第一个匹配元素并将其包装在 Some 中，如果未找到则 None。你也可以把 entries 变成一个视图，让计算变得懒惰。

entries.view.map(_.find(id)).collectFirst { case Some(node) => node }

它也适用于您的原始代码：

entries.view.flatMap(_.find(id)).headOption

Answer 2

解决此问题的另一种方法是为数据结构提供遍历支持。总体而言，它是一棵树，因此可以轻松遍历。请检查以下代码：

sealed abstract class Node(val id: Long, val name: String) extends Traversable[Node] {
  def foreach[U](f:Node => U) = this match {
    case x: Contact => f(x)
    case x: Group => f(x); x.entries.foreach(_ foreach f)
  }
}
case class Contact(override val id: Long, override val name: String, phone: String) extends Node(id, name) {
  override def toString = (id, name, phone).toString
}
case class Group(override val id: Long, override val name: String, entries: Node*) extends Node(id, name) {
  override def toString = (id, name).toString + entries.map(_.toString).mkString
}

val tree =
  Group(0, "Root"
    , Group(10, "Shop", Contact(11, "Alice", "312"))
    , Group(20, "Workshop"
      , Group(30, "Tyres", Contact(31, "Bob", "315"), Contact(32, "Greg", "319"))
      , Contact(33, "Mary", "302"))
    , Contact(1, "John", "317"))

println(tree) // (0,Root)(10,Shop)(11,Alice,312)(20,Workshop)(30,Tyres)(31,Bob,315)(32,Greg,319)(33,Mary,302)(1,John,317)
println(tree find {_.id == 32}) //Some((32,Greg,319))
println(tree map {_.name }) //List(Root, Shop, Alice, Workshop, Tyres, Bob, Greg, Mary, John)

好处是现在您可以使用 Traversable[Node] 特性的所有好处。另一方面的问题是我不得不覆盖 toString 方法以防类。所以我觉得还有改进的空间。

在第一次出现后打破 flatMap

Break flatMap after first Some occurence

collections

scala

在第一次出现后打破 ​​flatMap

Break flatMap after first Some occurence

collections

scala

在第一次出现后打破 flatMap