KamonAkkaHttp.scala

class T extends WordSpec with FridaDefaultJsonProtocol with Directives with SprayJsonSupport with Matchers with ScalatestRouteTest {
  import scala.concurrent.ExecutionContext.Implicits.global

  Kamon.addReporter(new InfluxDBReporter(Kamon.config()))

  private val log = LoggerFactory.getLogger(getClass.getName)

  val futureFunc: Future[Int] = Future(123)

  val eHandler = ExceptionHandler {
    case NonFatal(_) =>
      log.error("789")
      complete(GenericResponse(ok = false))
  }

  def r: Route =
    get {
      path("abc") {
        withTxKContext {
          log.info("123")
          handleExceptions(eHandler) {
            complete {
              futureFunc.map(_ * 2).map {
                _ =>
                  log.info("456")
                  throw new Exception("Test exception")
                  GenericResponse(ok = true)
              }
            }
          }
        }
      }
    }

  "R test" should {
    "propagate context" in {
      Get("/abc") ~> r ~> check {
        StatusCodes.OK shouldEqual StatusCodes.OK
      }
    }
  }

  private def withTxKContext[T](f: => T): T = {
    Kamon.withContext(Kamon.currentContext().withKey(Key.broadcast[String](name = "A", emptyValue = ""), "abc"))(f)
  }
}