如何在 Scala ExecutionContext 之间传输堆栈跟踪?

How can I transfer a stacktrace betwen Scala ExecutionContexts?

我已经围绕 Scala ExecutionContext 编写了一个小型实用程序包装器,以实现跨 Future 个实例的 MDC 上下文传输。它按预期工作,但一个不受欢迎的副作用是我们现在似乎没有得到跨越 Futures 的堆栈跟踪。如何确保堆栈跟踪与 MDC 一起传播?

这是我的代码供参考:

import org.slf4j.MDC

import scala.concurrent.ExecutionContext
import scala.jdk.CollectionConverters._


object MdcOps {

  implicit class ExecutionContextExt(value: ExecutionContext) {
    def withMdc: ExecutionContext = new MdcExecutionContext(value)
  }

  def withMdc[A](mdc: Map[String, Any], replace: Boolean)(doIt: => A): A = {
    val currentMdcContext = getMdc
    val newMdc = if(replace) mdc else mdc ++ currentMdcContext
    try { setMdc(newMdc); doIt }
    finally { setMdc(currentMdcContext) }
  }

  def setMdc(mdc: Map[String, Any]): Unit = {
    if(mdc.isEmpty) {
      MDC.clear()
    } else
      MDC.setContextMap(mdc.view.mapValues(_.toString).toMap.asJava)
  }

  def getMdc: Map[String, String] = Option(MDC.getCopyOfContextMap).map(_.asScala.toMap).getOrElse(Map.empty)
}

class MdcExecutionContext(underlying: ExecutionContext, context: Map[String, String] = Map.empty) extends ExecutionContext {
  override def prepare(): ExecutionContext = new MdcExecutionContext(underlying, MdcOps.getMdc)
  override def execute(runnable: Runnable): Unit = underlying.execute { () =>
    MdcOps.withMdc(context, replace = true)(runnable.run())
  }
  override def reportFailure(t: Throwable): Unit = underlying.reportFailure(t)
}

总的来说。 ZIO 实现了它 - 参见:https://www.slideshare.net/jdegoes/error-management-future-vs-zio from slide 53 to see the result - but creation of stacks, and passing it around causes performance penalty x2.4.

因为您无法将 API 更改为例如在编译时使用宏生成堆栈跟踪(使用例如 sourcode 库),在每个 .map/.flatMap 上你必须创建一个堆栈跟踪,从中删除不相关的帧,也许将它组合起来使用来自先前上下文的帧并将其设置在 MDC 中。撇开技术细节不谈——这很难做到正确且重量级,更像是整个库的 material 而不是简单的实用程序。如果可以便宜地完成,这将是内置的。

如果您对它非常感兴趣,请选择 ZIO 或分析它是如何实现它的——据我所知,它需要大量的努力并将此功能构建到库中,即使那样它也会导致性能下降。我只能想象使用 JVM 堆栈跟踪并传递它们的惩罚会更大。