scala - Scala中有没有类似Java Stream的 "peek"操作的东西？

Question

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

它当然很容易实现:

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

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

Answer 1

解决这个问题的一种方法是使用隐式:

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)

然后可以在 native 使用:

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

还有其他答案吗？