flatMap中的性能差异与for-comprehension相比

Question

我还是Scala的新手,我读过的其中一件事就是for-comprehension在某种程度上等同于flatMap.但是,在我的代码(下面)中,flatMap的计算时间至少是两倍.这可能是什么原因？

这是一个缓慢的:

facts.flatMap(f => factActionsMap(f)).filter(_.isValid(facts))

这是快速等效的:

for {
  f <- facts
  a <- factActionsMap(f)
  if a.isValid(facts)
} yield a

factActionsMap是Fact和之间的地图Set[Action]. facts只是一个Set[Fact].

Answer 1

我们来看看翻译:

scala> trait Fact
defined trait Fact

scala> trait Action { def isValid(s: Set[Fact]): Boolean }
defined trait Action

scala> def facts: Set[Fact] = ???
facts: Set[Fact]

scala> def factActionsMap: Map[Fact, Set[Action]] = ???
factActionsMap: Map[Fact,Set[Action]]

scala> import scala.reflect.runtime.{universe => u}
import scala.reflect.runtime.{universe=>u}

scala> val expr = u reify {
     | for {
     |   f <- facts
     |   a <- factActionsMap(f)
     |   if a.isValid(facts)
     | } yield a
     | }
expr: reflect.runtime.universe.Expr[scala.collection.immutable.Set[Action]] = Expr[scala.collection.immutable.Set[Action]]($read.f
acts.flatMap(((f) => $read.factActionsMap.apply(f).withFilter(((a) => a.isValid($read.facts))).map(((a) => a))(Set.canBuildFrom)))
(Set.canBuildFrom))

scala> u show expr.tree
res0: String = $read.facts.flatMap(((f) => $read.factActionsMap.apply(f).withFilter(((a) => a.isValid($read.facts))).map(((a) => a
))(Set.canBuildFrom)))(Set.canBuildFrom)

因此,删除REPL东西(一切都以$)开头,隐式参数加上重新格式化,我们得到:

facts.flatMap(f => factActionsMap(f).withFilter(a => a.isValid(facts)).map(a => a))

这与您提出的内容有两个主要区别.首先,withFilter应用于fastActionsMap(f)结果,而您应用于facts.flatMap结果.这意味着flatMap将对所有结果起作用,而不仅仅是接受的结果.

其次,它使用withFilter而不是filter,这避免了创建额外的集合.