wip: various fixes

This commit is contained in:
Simon Cruanes 2025-12-04 10:18:18 -05:00
parent 3fdb0eebd5
commit ebed5d7ce8
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
4 changed files with 12 additions and 62 deletions

View file

@ -1,7 +1,7 @@
type t = Opentelemetry_client.Config.t
type t = Opentelemetry_client.Client_config.t
module Env = Opentelemetry_client.Config.Env ()
module Env = Opentelemetry_client.Client_config.Env ()
let pp = Opentelemetry_client.Config.pp
let pp = Opentelemetry_client.Client_config.pp
let make = Env.make (fun common () -> common)

View file

@ -1,4 +1,4 @@
type t = Opentelemetry_client.Config.t
type t = Opentelemetry_client.Client_config.t
(** Configuration.
To build one, use {!make} below. This might be extended with more fields in
@ -6,7 +6,7 @@ type t = Opentelemetry_client.Config.t
val pp : Format.formatter -> t -> unit
val make : (unit -> t) Opentelemetry_client.Config.make
val make : (unit -> t) Opentelemetry_client.Client_config.make
(** Make a configuration {!t}. *)
module Env : Opentelemetry_client.Config.ENV
module Env : Opentelemetry_client.Client_config.ENV

View file

@ -25,48 +25,6 @@ let last_gc_metrics = Atomic.make (Mtime_clock.now ())
let timeout_gc_metrics = Mtime.Span.(20 * s)
(* Cross-domain, thread-safe storage for GC metrics gathered from different fibres. *)
module GC_metrics : sig
val add : Proto.Metrics.resource_metrics -> unit
val drain : unit -> Proto.Metrics.resource_metrics list
end = struct
(* Used to prevent data races across domains *)
let mutex = Eio.Mutex.create ()
let gc_metrics = ref []
let add m =
Eio.Mutex.use_rw ~protect:true mutex (fun () ->
gc_metrics := m :: !gc_metrics)
let drain () =
Eio.Mutex.use_rw ~protect:true mutex (fun () ->
let metrics = !gc_metrics in
gc_metrics := [];
metrics)
end
(* capture current GC metrics if {!needs_gc_metrics} is true,
or it has been a long time since the last GC metrics collection,
and push them into {!gc_metrics} for later collection *)
let sample_gc_metrics_if_needed () =
let now = Mtime_clock.now () in
let alarm = Atomic.compare_and_set needs_gc_metrics true false in
let timeout () =
let elapsed = Mtime.span now (Atomic.get last_gc_metrics) in
Mtime.Span.compare elapsed timeout_gc_metrics > 0
in
if alarm || timeout () then (
Atomic.set last_gc_metrics now;
let l =
OT.Metrics.make_resource_metrics
~attrs:(Opentelemetry.GC_metrics.get_runtime_attributes ())
@@ Opentelemetry.GC_metrics.get_metrics ()
in
GC_metrics.add l
)
type error =
[ `Status of int * Opentelemetry.Proto.Status.status
| `Failure of string
@ -282,7 +240,6 @@ let mk_emitter ~stop ~net (config : Config.t) : (module EMITTER) =
let push_metrics x =
let@ () = guard_exn_ "push metrics" in
sample_gc_metrics_if_needed ();
push_to_batch batch_metrics x
let push_logs x =
@ -299,8 +256,7 @@ let mk_emitter ~stop ~net (config : Config.t) : (module EMITTER) =
let emit_metrics_maybe =
maybe_emit batch_metrics config.url_metrics (fun collected_metrics ->
let gc_metrics = GC_metrics.drain () in
gc_metrics @ collected_metrics |> Signal.Encode.metrics)
collected_metrics |> Signal.Encode.metrics)
let emit_logs_maybe =
maybe_emit batch_logs config.url_logs Signal.Encode.logs
@ -336,7 +292,6 @@ let mk_emitter ~stop ~net (config : Config.t) : (module EMITTER) =
if Config.Env.get_debug () then
Printf.eprintf "tick (from domain %d)\n%!" (Domain.self () :> int);
run_tick_callbacks ();
sample_gc_metrics_if_needed ();
emit_all ~force:false
let cleanup ~on_done () =
@ -344,13 +299,12 @@ let mk_emitter ~stop ~net (config : Config.t) : (module EMITTER) =
Printf.eprintf "opentelemetry: exiting…\n%!";
Atomic.set stop true;
run_tick_callbacks ();
sample_gc_metrics_if_needed ();
emit_all ~force:true;
on_done ()
end in
(module M : EMITTER)
module Backend (Emitter : EMITTER) : Opentelemetry.Collector.BACKEND = struct
module Backend (Emitter : EMITTER) : Opentelemetry.Exporter.t = struct
open Opentelemetry.Proto
open Opentelemetry.Collector
open Emitter

View file

@ -34,15 +34,11 @@ let emit_telemetry do_emit = Logs.Tag.(empty |> add emit_telemetry_tag do_emit)
(*****************************************************************************)
(* Log a message to otel with some attrs *)
let log ?service_name ?(attrs = []) ?(scope = Otel.Scope.get_ambient_scope ())
~level msg =
let log ?service_name ?(attrs = []) ?(scope = Otel.Ambient_span.get ()) ~level
msg =
let log_level = Logs.level_to_string (Some level) in
let span_id =
Option.map (fun (scope : Otel.Scope.t) -> scope.span_id) scope
in
let trace_id =
Option.map (fun (scope : Otel.Scope.t) -> scope.trace_id) scope
in
let span_id = Option.map Otel.Span.id scope in
let trace_id = Option.map Otel.Span.trace_id scope in
let severity = log_level_to_severity level in
let log =
Otel.Log_record.make_str ~severity ~log_level ?trace_id ?span_id msg