natchez-lightstep_3.0.2.1.source-code.LightstepEntryPoint.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of natchez-lightstep_3 Show documentation

Lightstep support for Natchez.

There is a newer version: 0.3.7

// Copyright (c) 2019-2020 by Rob Norris and Contributors
// This software is licensed under the MIT License (MIT).
// For more information see LICENSE or https://opensource.org/licenses/MIT

package natchez
package lightstep

import cats.effect.{Resource, Sync}
import cats.syntax.all._
import io.opentracing.Tracer
import io.opentracing.propagation.{Format, TextMapAdapter}

import scala.jdk.CollectionConverters._

final class LightstepEntryPoint[F[_]: Sync](tracer: Tracer) extends EntryPoint[F] {
  override def root(name: String): Resource[F, Span[F]] =
    Resource
      .make(Sync[F].delay(tracer.buildSpan(name).start()))(s => Sync[F].delay(s.finish()))
      .map(LightstepSpan(tracer, _))

  override def continue(name: String, kernel: Kernel): Resource[F, Span[F]] =
    Resource
      .make(
        Sync[F].delay {
          val p =
            tracer.extract(Format.Builtin.HTTP_HEADERS, new TextMapAdapter(kernel.toHeaders.asJava))
          tracer.buildSpan(name).asChildOf(p).start()
        }
      )(s => Sync[F].delay(s.finish()))
      .map(LightstepSpan(tracer, _))

  override def continueOrElseRoot(name: String, kernel: Kernel): Resource[F, Span[F]] =
    continue(name, kernel).flatMap {
      case null => root(name)
      case a    => a.pure[Resource[F, *]]
    }
}