simon/ocaml-opentelemetry
1
0
Fork 0
mirror of https://github.com/ocaml-tracing/ocaml-opentelemetry.git synced 2026-05-05 17:04:52 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

trace state in cohttp-lwt integration

Browse source
This commit is contained in:
Simon Cruanes 2026-04-04 03:11:04 +00:00
parent dbf459996a
commit 3a7e753348
1 changed files with 28 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
src/integrations/cohttp/opentelemetry_cohttp_lwt.ml
View file

@ -87,11 +87,17 @@ end = struct
let set_trace_context (span : Otel.Span.t) req = let set_trace_context (span : Otel.Span.t) req =
let module Traceparent = Otel.Trace_context.Traceparent in let module Traceparent = Otel.Trace_context.Traceparent in
let module Tracestate = Otel.Trace_context.Tracestate in
let headers = let headers =
Header.add (Request.headers req) header_x_ocaml_otel_traceparent Header.add (Request.headers req) header_x_ocaml_otel_traceparent
(Traceparent.to_value ~trace_id:(Otel.Span.trace_id span) (Traceparent.to_value ~trace_id:(Otel.Span.trace_id span)
~parent_id:(Otel.Span.id span) ()) ~parent_id:(Otel.Span.id span) ())
in in
let headers =
match Otel.Span.trace_state span with
| [] -> headers
| ts -> Header.add headers Tracestate.name (Tracestate.to_w3c_string ts)
in
{ req with headers } { req with headers }
let get_trace_context ?(from = `Internal) req : Otel.Span.t option = let get_trace_context ?(from = `Internal) req : Otel.Span.t option =
@ -106,9 +112,20 @@ end = struct
| Some v -> | Some v ->
(match Traceparent.of_value v with (match Traceparent.of_value v with
| Ok (trace_id, parent_id) -> | Ok (trace_id, parent_id) ->
(* TODO: we need a span_ctx here actually *) let trace_state =
match from with
| `External ->
(match
Header.get (Request.headers req)
Otel.Trace_context.Tracestate.name
with
| None -> ""
| Some ts -> ts)
| `Internal -> ""
in
Some Some
(Otel.Span.make ~trace_id ~id:parent_id ~start_time:0L ~end_time:0L "") (Otel.Span.make ~trace_id ~id:parent_id ~trace_state ~start_time:0L
~end_time:0L "")
| Error _ -> None) | Error _ -> None)
let remove_trace_context req = let remove_trace_context req =
@ -169,14 +186,20 @@ let client ?(tracer = Otel.Tracer.default) ?(span : Otel.Span.t option)
let add_traceparent (span : Otel.Span.t) headers = let add_traceparent (span : Otel.Span.t) headers =
let module Traceparent = Otel.Trace_context.Traceparent in let module Traceparent = Otel.Trace_context.Traceparent in
let module Tracestate = Otel.Trace_context.Tracestate in
let headers = let headers =
match headers with match headers with
| None -> Header.init () | None -> Header.init ()
| Some headers -> headers | Some headers -> headers
in in
let headers =
Header.add headers Traceparent.name Header.add headers Traceparent.name
(Traceparent.to_value ~trace_id:(Otel.Span.trace_id span) (Traceparent.to_value ~trace_id:(Otel.Span.trace_id span)
~parent_id:(Otel.Span.id span) ()) ~parent_id:(Otel.Span.id span) ())
in
match Otel.Span.trace_state span with
| [] -> headers
| ts -> Header.add headers Tracestate.name (Tracestate.to_w3c_string ts)
let call ?ctx ?headers ?body ?chunked meth (uri : Uri.t) : let call ?ctx ?headers ?body ?chunked meth (uri : Uri.t) :
(Response.t * Cohttp_lwt.Body.t) Lwt.t = (Response.t * Cohttp_lwt.Body.t) Lwt.t =