simon/ocaml-trace
1
0
Fork 0
mirror of https://github.com/ocaml-tracing/ocaml-trace.git synced 2026-03-07 18:37:56 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

share code for implicit spans

Browse source
This commit is contained in:
Simon Cruanes 2023-12-22 20:15:25 -05:00
parent ef15941936
commit b387729081
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
1 changed files with 16 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
src/tef/trace_tef.ml
View file

@ -386,8 +386,7 @@ let collector ~out () : collector =
else
Thread.id (Thread.self ())
let enter_span ~__FUNCTION__:fun_name ~__FILE__:_ ~__LINE__:_ ~data name :
span =
let[@inline] enter_span_ ~fun_name ~data name : span =
let span = Int64.of_int (A.fetch_and_add span_id_gen_ 1) in
let tid = get_tid_ () in
let time_us = now_us () in
@ -395,23 +394,26 @@ let collector ~out () : collector =
(E_define_span { tid; name; time_us; id = span; fun_name; data });
span
let enter_span ~__FUNCTION__:fun_name ~__FILE__:_ ~__LINE__:_ ~data name :
span =
enter_span_ ~fun_name ~data name
let exit_span span : unit =
let time_us = now_us () in
B_queue.push events (E_exit_span { id = span; time_us })
(* re-raise exception with its backtrace *)
external reraise : exn -> 'a = "%reraise"
let with_span ~__FUNCTION__:fun_name ~__FILE__:_ ~__LINE__:_ ~data name f =
let span = Int64.of_int (A.fetch_and_add span_id_gen_ 1) in
let tid = get_tid_ () in
let time_us = now_us () in
B_queue.push events
(E_define_span { tid; name; time_us; id = span; fun_name; data });
let finally () =
let time_us = now_us () in
B_queue.push events (E_exit_span { id = span; time_us })
in
Fun.protect ~finally (fun () -> f span)
let span = enter_span_ ~fun_name ~data name in
try
let x = f span in
exit_span span;
x
with exn ->
exit_span span;
reraise exn
let add_data_to_span span data =
if data <> [] then B_queue.push events (E_add_data { id = span; data })