Traced transactor for trace4cats

dependencies.scala

object V {   
  val Doobie = "1.0.0-RC2"
  val NatchezExtras = "7.0.0-M4"
  val Trace4Cats = "0.14.0"
}

val trace4catsNatchez = "io.janstenpickle" %% "trace4cats-natchez" % V.Trace4Cats
val natchezExtrasDoobie = "com.ovoenergy" %% "natchez-extras-doobie" % V.NatchezExtras
val doobie = "org.tpolecat" %% "doobie-core" % V.Doobie

libraryDependencies ++= Seq(trace4catsNatchez, natchezExtrasDoobie, doobie)

TracedTransactor.scala

package io.github.ccarlile

import cats.effect.Async
import cats.data.Kleisli
import cats.~>
import cats.arrow.FunctionK
import natchez.{Span => NatchezSpan}
import doobie.Transactor
import trace4cats.{Span}
import trace4cats.kernel.ToHeaders
import com.ovoenergy.natchez.extras.doobie.{TracedTransactor => NatchezTracedTransactor}
import trace4cats.natchez.Trace4CatsSpan

object TracedTransactor {
  
  type NatchezTraced[F[_], A] = Kleisli[F, NatchezSpan[F], A]
  type Traced[F[_], A] = Kleisli[F, Span[F], A]

  def apply[F[_]: Async](transactor: Transactor[F], appName: String): Transactor[Traced[F, *]] =
    traceTransactor(traceNatchez[F](transactor, appName))

  private def traceTransactor[F[_]: Async](
    transactor: Transactor[NatchezTraced[F, *]]
  ): Transactor[Traced[F, *]] = transactor.mapK(convert[F])

  private def convert[F[_]: Async]: NatchezTraced[F, *] ~> Traced[F, *] =
    new FunctionK[NatchezTraced[F, *], Traced[F, *]] {
      def apply[A](fa: Kleisli[F, NatchezSpan[F], A]): Kleisli[F, Span[F], A] = fa.local(convertSpan)
    }

  private def convertSpan[F[_]: Async]: Span[F] => NatchezSpan[F] = span => Trace4CatsSpan(span, ToHeaders.standard)

  private def traceNatchez[F[_]: Async](
    transactor: Transactor[F],
    appName: String
  ): Transactor[NatchezTraced[F, *]] =
    NatchezTracedTransactor[F](appName, transactor)

}