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

fix otel-lwt: logic for tracer.with_ was invalid

Browse source 
sometimes spans would be dropped!
This commit is contained in:
Simon Cruanes 2025-12-25 01:21:12 -05:00
parent 3b6e239c17
commit b4b864a0b6
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
1 changed files with 16 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
src/lwt/opentelemetry_lwt.ml
View file

@ -25,23 +25,29 @@ module Tracer = struct
let with_ (type a) ?(tracer = dynamic_main) ?force_new_trace_id ?trace_state
?attrs ?kind ?trace_id ?parent ?links name (cb : Span.t -> a Lwt.t) :
a Lwt.t =
let open Lwt.Syntax in
let thunk, finally =
with_thunk_and_finally tracer ?force_new_trace_id ?trace_state ?attrs
?kind ?trace_id ?parent ?links name cb
in
match thunk () with
| exception exn ->
let bt = Printexc.get_raw_backtrace () in
finally (Error (exn, bt));
Printexc.raise_with_backtrace exn bt
| promise ->
Lwt.on_any promise
(fun _ -> finally (Ok ()))
let* r =
Lwt.catch
(fun () ->
let+ res = thunk () in
Ok res)
(fun exn ->
let bt = Printexc.get_raw_backtrace () in
finally (Error (exn, bt)));
promise
Lwt.return (Error (exn, bt)))
in
match r with
| Ok r ->
finally (Ok ());
Lwt.return r
| Error (exn, bt) ->
finally (Error (exn, bt));
Lwt.fail exn
end
module Trace = Tracer [@@deprecated "use Tracer"]