Replace traceparent header, rather than setting all headers

The bug manifested itself by missing authentication headers, as these
were cleared when the child trace header was set.

1 files changed, 6 insertions, 2 deletions

diff --git a/src/main/scala/xyz/driver/core/init/HttpApi.scala b/src/main/scala/xyz/driver/core/init/HttpApi.scala
index 6330fbe..81428bf 100644
--- a/src/main/scala/xyz/driver/core/init/HttpApi.scala
+++ b/src/main/scala/xyz/driver/core/init/HttpApi.scala
@@ -75,8 +75,12 @@ trait HttpApi extends CloudServices with Directives with SprayJsonSupport { self
     }
     reporter
       .traceWithOptionalParentAsync(s"http_handle_rpc", tags, parent) { spanContext =>
-        val header     = Traceparent(spanContext)
-        val withHeader = ctx.withRequest(ctx.request.withHeaders(header))
+        val header = Traceparent(spanContext)
+        val withHeader = ctx.withRequest(
+          ctx.request
+            .removeHeader(header.name)
+            .addHeader(header)
+        )
         inner(withHeader)
       }
   }