Question

我希望看到地图/平面图中传递的f函数，但没有运气。我抛出了一个例外，看到f的任何迹象都没有用。那是什么功能？它是如何在幕后产生的？

Exception in thread "main" java.lang.RuntimeException
   at x.x.Main$A.getInt(Empty.scala:8)
   at x.x.Main$A.flatMap(Empty.scala:10)

object Main extends App {

  class A {
    def getInt: Int = throw new RuntimeException
    def map(f: Int => Boolean): Boolean = f(getInt)
    def flatMap(f: Int => Boolean): Boolean = f(getInt)
  }

  for {
    x <- new A
    y <- new A
  } yield x == y
}

Answer 1

for-comprehension的等效代码

for {
  x <- new A
  y <- new A
} yield x == y

是这样的：

new A().flatMap{ x => new A().map{ y => x == y } }

您可以使用scalac -Xprint:parser main.scala来获取for-comprehension生成的代码。在这种情况下，你会得到这个：

new A().flatMap(((x) => new A().map(((y) => x.$eq$eq(y)))))

如果没有像这样的附加宏，你也可以在REPL中获取它：

import reflect.runtime.universe._

show{ reify{
  for {
    x <- new A
    y <- new A
  } yield x == y
}.tree }
// new $read.A().flatMap(((x) => new $read.A().map(((y) => x.$eq$eq(y)))))

Answer 2

如果您使用的是Scala 2.10或更高版本，则可以使用以下命令在repl中显示desugared scala代码：

import scala.reflect.macros.Context // use BlackboxContext if you're in 2.11
import scala.reflect.runtime.universe._
import scala.language.experimental.macros

def _desugar(c : Context)(expr : c.Expr[Any]): c.Expr[Unit] = {
  import c.universe._
  println(show(expr.tree))
  reify {}
}

def desugar(expr : Any): Unit = macro _desugar

这将允许您传入代码块，并查看它们转换成的内容。例如，在repl：

中

scala> class A {
     |   def getInt: Int = throw new RuntimeException
     |   def map(f: Int => Boolean): Boolean = f(getInt)
     |   def flatMap(f: Int => Boolean): Boolean = f(getInt)
     | }
defined class A

scala> desugar {
     |   for {
     |     x <- new A
     |     y <- new A
     |   } yield x == y
     | }
new $line15.$read.$iw.$iw.$iw.$iw.A().flatMap(((x: Int) => new $line15.$read.$iw.$iw.$iw.$iw.A().map(((y: Int) => x.==(y)))))

它有点乱，因为repl创建了几个中间临时变量，但你可以看到发生了什么的结构。

new A().flatMap { (x: Int) =>
  new A().map { (y: Int) =>
    x == y
  }
}

这适用于大多数表达式，并允许您检查实际代码在编译期间将转换为什么。

更新

我应该指出我的来源 - 我的desugar版本是github上Macrocosm repo中找到的函数的略微修改版本。

什么是用于理解的地图/平面地图功能？

2 个答案:

更新