scala - Scala中同时打印和返回值的惯用方式

Question

这个问题在这里已经有了答案:

how to keep return value when logging in scala (6 个回答)

关闭 7年前。

在 Scala 中打印(或做任何我需要做的事情)和返回值的惯用方式是什么？例如，

Seq(1,2,3)
  .map(_ * 2)
  .xxx(println) // Here I want to print the intermediate sequence
  .foldLeft(0)(_ + _)

我能想到的一种方法是使用隐式，但我不太喜欢自己修补标准库。

注意

在 Ruby 中我们可以使用 Object#tap

[1,2,3]
  .map { |i| i * 2 }
  .tap { |i| puts i }
  .reduce(0) { |x, i| x += i }

Answer 1

Object#tap在 Ruby 中基本上是 K 组合器的变体。我不相信 Scala 标准库中有实现，但很容易添加自己的:

implicit class TapExtension[T](o: => T) {
  def tap(f: T => Unit) = { f(o); o }
}

注意:这是一个隐式转换，它不是猴子补丁。

然后，你可以这样使用它:

Seq(1,2,3)
  .map(_ * 2)
  .tap(println)
  .foldLeft(0)(_ + _)