• Home
• io.scalaland
• mdc4s-monix_2.13
• 0.1.0
• source code
• TaskMDC.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.scalaland.mdc.monix.TaskMDC.scala Maven / Gradle / Ivy

package io.scalaland.mdc.monix

import io.scalaland.mdc.MDC
import monix.execution.misc.Local
import monix.eval.Task

final class TaskMDC(local: Local[MDC.Ctx]) extends MDC[Task] {

  private def enable[A](fa: Task[A]): Task[A] = fa // .executeWithOptions(_.enableLocalContextPropagation)

  def get(key: String): Task[Option[String]] = enable(Task(local().get(key)))
  def set(key: String, value: String): Task[Unit] = enable(Task(local := local().updated(key, value)))
}
object TaskMDC {

  final private class TaskCtxManager(local: Local[MDC.Ctx]) extends MDC.CtxManager {

    def getMDC: MDC.Ctx = local()
    def setMDC(mdc: MDC.Ctx): Unit = local := mdc
  }

  /** Requires `task.executeWithOptions(_.enableLocalContextPropagation)`. */
  def configure[A: MDC.Initializer](
      onFork: MDC.Ctx => MDC.Ctx = identity,
      onJoin: (MDC.Ctx, MDC.Ctx) => MDC.Ctx = (a, _) => a
  ): Task[MDC[Task]] = for {
    local <- Task.now(Local[MDC.Ctx](Map.empty))
    _ <- TaskGlobal.addHandler(local, ForkJoinLocalHandler(onFork)(onJoin))
    _ <- Task(MDC.Initializer[A](new TaskCtxManager(local)))
  } yield new TaskMDC(local)
}