使用def方法和val函数展平Vs flatMap

时间:2017-04-23 07:45:15

标签: scala flatten flatmap

使用def方法和val函数展平Vs flatMap:

我定义了一个名为 toInt 的def方法:

  def toInt(s: String): Option[Int] = {
    try {
      Some(Integer.parseInt(s.trim))
    } catch {
      case e: Exception => None
    }
  }

这个方法适用于flatten和flatMap,如下所示:

//using toInt method
val x = 1.to(5).toList
val y = List("a")
val z = x ++ y
val q = z.map(_.toString)

//using map and flatten
println(q.map(toInt).flatten)
//using flatMap
println(q.flatMap(toInt))

现在我在函数“tooInt”中使用val定义了相同的toInt功能(如在def方法中):

val tooInt: String => Option[Int] = s => {
  try {
    Some(Integer.parseInt(s.trim))
  } catch {
    case c: Exception => None
  }
}

这适用于flatten,但 NOT 使用flatMap,如下所示:

//using map and flatten
 println(q.map(tooInt).flatten)
 //using flatMap // this has error
 **println(q.flatMap(tooInt))**

你能帮我理解一下吗?

最诚挚的问候, 基兰

1 个答案:

答案 0 :(得分:9)

你需要通过扩展来帮助编译器做一些工作:

q.flatMap(s => tooInt(s))

这一切归结为我们在option2Iterable中定义了隐式Option.scala这一事实。当您首先map然后flatten时,Option[Int]已经在范围内,并且可以应用隐式。但是当你flatMap时,编译器必须首先将tooInt扩展为s => tooInt(s),然后应用隐式解决方案,但这不起作用。为什么它不起作用?因为编译器会查找隐式类型:

pt=(=> String => Option[Int]) => (String => scala.collection.GenTraversableOnce[?])

哪个不存在。相反,toInt方法首先扩展为函数类型,然后搜索隐式Option[Int]

-- toInt : pt=String => scala.collection.GenTraversableOnce[?] BYVALmode-EXPRmode-POLYmode (site: value r in X)
|    |    |    |    |    |    |-- { ((s: String) => toInt(s)) } : pt=String => scala.collection.GenTraversableOnce[?] BYVALmode-EXPRmode-POLYmode (site: value r in X)
|    |    |    |    |    |    |    |-- ((s: String) => toInt(s)) : pt=String => scala.collection.GenTraversableOnce[?] BYVALmode-EXPRmode-POLYmode (site: value r in X)
|    |    |    |    |    |    |    |    |-- (s: String)Option[Int] : pt=scala.collection.GenTraversableOnce[?] EXPRmode (site: value $anonfun in X)
|    |    |    |    |    |    |    |    |    |-- s : pt=String BYVALmode-EXPRmode (site: value $anonfun in X)
|    |    |    |    |    |    |    |    |    |    \-> String
|    |    |    |    |    |    |    |    |    [search #3] start `(s: String)Option[Int]`, searching for adaptation to pt=Option[Int] => scala.collection.GenTraversableOnce[?] (silent: value $anonfun in X) implicits disabled
|    |    |    |    |    |    |    |    |    [search #3] considering scala.this.Option.option2Iterable

我们也可以在反编译代码中看到它:

val r: scala.collection.immutable.IndexedSeq[Int] = q.flatMap[Int, scala.collection.immutable.IndexedSeq[Int]]({
    {
      final <artifact> def $anonfun$main(s: String): Iterable[Int] = scala.this.Option.option2Iterable[Int](toInt(s));
      ((s: String) => $anonfun$main(s))
    }
  }, immutable.this.IndexedSeq.canBuildFrom[Int]());