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

improve notifier_lwt

Browse source
This commit is contained in:
Simon Cruanes 2025-12-05 23:30:44 -05:00
parent 311776344a
commit 9e839befa4
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/client/lwt/notifier_lwt.ml
View file

@ -6,6 +6,7 @@ type t = {
notified: bool Atomic.t; notified: bool Atomic.t;
cond: unit Lwt_condition.t; cond: unit Lwt_condition.t;
notification: int; notification: int;
lwt_tid: int; (** thread ID where lwt runs *)
deleted: bool Atomic.t; deleted: bool Atomic.t;
} }
@ -17,14 +18,20 @@ let create () : t =
Atomic.set notified false; Atomic.set notified false;
Lwt_condition.broadcast cond ()) Lwt_condition.broadcast cond ())
in in
{ notified; notification; cond; deleted = Atomic.make false } let lwt_tid = Thread.id @@ Thread.self () in
{ notified; notification; cond; lwt_tid; deleted = Atomic.make false }
let delete self : unit = let delete self : unit =
if not (Atomic.exchange self.deleted true) then if not (Atomic.exchange self.deleted true) then
Lwt_unix.stop_notification self.notification Lwt_unix.stop_notification self.notification
let trigger (self : t) : unit = let trigger (self : t) : unit =
if not (Atomic.exchange self.notified true) then let tid = Thread.id @@ Thread.self () in
if tid = self.lwt_tid then
(* in lwt thread, directly use the condition *)
Lwt_condition.broadcast self.cond ()
else if not (Atomic.exchange self.notified true) then
Lwt_unix.send_notification self.notification Lwt_unix.send_notification self.notification
let wait (self : t) : unit Lwt.t = Lwt_condition.wait self.cond let wait (self : t) : unit Lwt.t = Lwt_condition.wait self.cond