diff options
Diffstat (limited to 'metrics')
| -rw-r--r-- | metrics/brave-backend/src/main/scala/com/softwaremill/sttp/brave/BraveBackend.scala | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/metrics/brave-backend/src/main/scala/com/softwaremill/sttp/brave/BraveBackend.scala b/metrics/brave-backend/src/main/scala/com/softwaremill/sttp/brave/BraveBackend.scala index a530251..3e36616 100644 --- a/metrics/brave-backend/src/main/scala/com/softwaremill/sttp/brave/BraveBackend.scala +++ b/metrics/brave-backend/src/main/scala/com/softwaremill/sttp/brave/BraveBackend.scala @@ -1,7 +1,7 @@ package com.softwaremill.sttp.brave import brave.http.{HttpClientAdapter, HttpClientHandler, HttpTracing} -import brave.propagation.TraceContext +import brave.propagation.{Propagation, TraceContext} import brave.{Span, Tracing} import com.softwaremill.sttp.brave.BraveBackend._ import com.softwaremill.sttp.{FollowRedirectsBackend, MonadError, Request, Response, SttpBackend} @@ -69,8 +69,10 @@ class BraveBackend[R[_], S] private (delegate: SttpBackend[R, S], httpTracing: H httpTracing .tracing() .propagation() - .injector((_: Request[_, _], key: String, value: String) => { - tracedRequest = tracedRequest.header(key, value) + .injector(new Propagation.Setter[AnyRequest, String] { + override def put(carrier: AnyRequest, key: String, value: String): Unit = { + tracedRequest = tracedRequest.header(key, value) + } }) .inject(span.context(), request) |