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

fix trace-tef: cleanup

Browse source
This commit is contained in:
Simon Cruanes 2023-06-09 08:10:35 -04:00
parent c58810f4e8
commit 718cbf120f
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/tef/trace_tef.ml
View file

@ -182,8 +182,9 @@ let bg_thread ~out (events : event B_queue.t) : unit =
}
| E_exit_span { id; time_us = stop_us } ->
(match Span_tbl.find_opt spans id with
| None -> (* bug! *) ()
| None -> (* bug! TODO: emit warning *) ()
| Some { (* __FUNCTION__; __FILE__; __LINE__; *) tid; name; start_us } ->
Span_tbl.remove spans id;
Writer.emit_duration_event ~tid ~name ~start:start_us ~end_:stop_us
~args:[] writer)
in
@ -193,7 +194,12 @@ let bg_thread ~out (events : event B_queue.t) : unit =
let ev = B_queue.pop events in
handle_ev ev
done
with B_queue.Closed -> ()
with B_queue.Closed ->
(* warn if app didn't close all spans *)
if Span_tbl.length spans > 0 then
Printf.eprintf "trace-tef: warning: %d spans were not closed\n%!"
(Span_tbl.length spans);
()
type output =
[ `Stdout
@ -218,9 +224,7 @@ let collector ~out () : collector =
let shutdown () =
if A.exchange active false then (
Printf.eprintf "shutdown\n%!";
B_queue.close events;
Printf.eprintf "wait\n%!";
Thread.join t_write
)