feat: use traceparent header from incoming shape requests to set pare…

…nt span (#2000)

Fix #1914

- open-telemetry/opentelemetry-erlang#518

---------

Co-authored-by: Ilia Borovitinov <[email protected]>

.changeset/odd-walls-walk.md

@@ -0,0 +1,5 @@
+---
+"@core/sync-service": patch
+---
+
+use traceparent header from incoming shape requests to set parent span

packages/sync-service/lib/electric/plug/router.ex

@@ -10,6 +10,7 @@ defmodule Electric.Plug.Router do
   plug Plug.Head
   plug :match
   plug Electric.Plug.LabelProcessPlug
+  plug Electric.Plug.TraceContextPlug
   plug Plug.Telemetry, event_prefix: [:electric, :routing]
   plug Plug.Logger
   plug :put_cors_headers

packages/sync-service/lib/electric/plug/trace_context_plug.ex

@@ -0,0 +1,28 @@
+defmodule Electric.Plug.TraceContextPlug do
+  @moduledoc """
+  A plug that extracts trace context from incoming HTTP headers and sets it as the parent span.
+  """
+  @behaviour Plug
+
+  require Logger
+
+  def init(opts), do: opts
+
+  def call(%Plug.Conn{req_headers: headers} = conn, _opts) do
+    # Extract function expects a list of headers as tuples and knows
+    # the expected header keys, so we don't have to prefilter.
+    ctx = :otel_propagator_text_map.extract_to(:otel_ctx.new(), headers)
+
+    # Get the span context from the extracted context
+    case :otel_tracer.current_span_ctx(ctx) do
+      :undefined ->
+        # No parent, continue as-is
+        conn
+
+      span_ctx ->
+        # Parent found, set as current span
+        :otel_tracer.set_current_span(span_ctx)
+        conn
+    end
+  end
+end