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

perf: make with_ inlinable in case there is no collector

Browse source
This commit is contained in:
Simon Cruanes 2023-06-08 23:17:39 -04:00
parent ab8e8bd9c2
commit 0707d932bc
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
1 changed files with 18 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
src/trace.ml
View file

@ -22,19 +22,25 @@ let[@inline] exit_span span : unit =
| None -> ()
| Some (module C) -> C.exit_span span
let with_ ?__FUNCTION__ ~__FILE__ ~__LINE__ name f =
let with_collector_ (module C : Collector.S) ?__FUNCTION__ ~__FILE__ ~__LINE__
name f =
let sp = C.create_span ?__FUNCTION__ ~__FILE__ ~__LINE__ name in
match f sp with
| x ->
C.exit_span sp;
x
| exception exn ->
let bt = Printexc.get_raw_backtrace () in
C.exit_span sp;
Printexc.raise_with_backtrace exn bt
let[@inline] with_ ?__FUNCTION__ ~__FILE__ ~__LINE__ name f =
match A.get collector with
| None -> f Collector.dummy_span
| Some (module C) ->
let sp = C.create_span ?__FUNCTION__ ~__FILE__ ~__LINE__ name in
(match f sp with
| x ->
C.exit_span sp;
x
| exception exn ->
let bt = Printexc.get_raw_backtrace () in
C.exit_span sp;
Printexc.raise_with_backtrace exn bt)
| None ->
(* fast path: no collector, no span *)
f Collector.dummy_span
| Some collector ->
with_collector_ collector ?__FUNCTION__ ~__FILE__ ~__LINE__ name
let[@inline] message ?__FUNCTION__ ~__FILE__ ~__LINE__ msg : unit =
match A.get collector with