ocaml-trace/test/t1.ml
2026-01-14 22:47:08 -05:00

49 lines
1.4 KiB
OCaml

let run () =
Trace.set_process_name "main";
Trace.set_thread_name "t1";
let n = ref 0 in
Trace.with_span ~__FILE__ ~__LINE__ "main" @@ fun _sp ->
for _i = 1 to 50 do
Trace.with_span ~__FILE__ ~__LINE__ "outer.loop" @@ fun _sp ->
let pseudo_async_sp =
Trace.enter_span ~parent:None ~flavor:`Async ~__FILE__ ~__LINE__
"fake_sleep"
in
for _j = 2 to 5 do
incr n;
Trace.with_span ~__FILE__ ~__LINE__ "inner.loop" @@ fun _sp ->
Trace.messagef (fun k -> k "hello %d %d" _i _j);
Trace.message "world";
Trace.counter_int "n" !n;
Trace.add_data_to_span _sp [ "i", `Int _i ];
if _j = 2 then (
Trace.add_data_to_span _sp [ "j", `Int _j ];
let _sp =
Trace.enter_span ~parent:(Some pseudo_async_sp)
~flavor:
(if _i mod 3 = 0 then
`Sync
else
`Async)
~__FILE__ ~__LINE__ "sub-sleep"
in
(* fake micro sleep *)
Thread.delay 0.005;
Trace.exit_span _sp
) else if _j = 3 then (
(* pretend some task finished. Note that this is not well scoped wrt other spans. *)
Trace.add_data_to_span pseudo_async_sp [ "slept", `Bool true ];
Trace.exit_span pseudo_async_sp
)
done
done
let () =
Trace_tef.Private_.mock_all_ ();
Trace_tef.with_setup ~out:`Stdout () @@ fun () -> run ()