devkit/possibly_otel.real.ml at master · level-a/devkit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
open Opentelemetry

let (let*) o f = Option.map f o

module Traceparent = struct
  let name = Trace_context.Traceparent.name

  let get_ambient () =
    let* Scope.{ trace_id; span_id; _ } = Scope.get_ambient_scope () in
    Trace_context.Traceparent.to_value ~trace_id ~parent_id:span_id ()
end

let enter_manual_span ~__FUNCTION__ ~__FILE__ ~__LINE__ ?data name =
    match Scope.get_ambient_scope () with
    | None ->
      Trace_core.enter_manual_toplevel_span ~__FUNCTION__ ~__FILE__ ~__LINE__ ?data name
    | Some Scope.{ span_id; _ } ->
      let otrace_espan = Trace_core.{
        span = Opentelemetry_trace.Internal.otrace_of_otel span_id;
        meta = Trace_core.Meta_map.empty
      } in
      Trace_core.enter_manual_sub_span ~parent:otrace_espan ~__FUNCTION__ ~__FILE__ ~__LINE__ ?data name