有没有像Java Stream" peek"在Scala中运作?

时间:2016-07-08 00:43:43

标签: scala java-stream monads

在Java中,您可以在Stream上调用peek(x -> println(x)),它将为每个元素执行操作并返回原始流,这与foreach是Unit不同。 Scala中是否有类似的东西,理想情况下适用于所有Monady类型,允许您通过"原来的Monad在进行副作用时会怎样? (记录,例如)

当然很容易实现:

def tap[A, U](a: A)(action: (A) => U): A = {
  action(a)
  a
}

但我希望有一些更优雅或惯用的东西。

1 个答案:

答案 0 :(得分:1)

解决此问题的一种方法是使用implicits:

class Tappable[A](a: A) {
  def tap[U](action: (A) => U): A = {
    action(a)
    a
  }
}

implicit def any2Tappable[A](a: A): Tappable[A] = new Tappable[A](a)

然后可以原生使用:

connection.tap(_.connect()) match {
  case c if c.getResponseCode == 304 => None
  ...

还有其他答案吗?