support for domain-local-await when installed

dune-project

@@ -26,6 +26,8 @@
    (and
     (>= 1.9.0)
     :with-test)))
+ (depopts
+   (domain-local-await (>= 0.2)))
  (tags
   (thread pool domain)))
 

moonpool.opam

@@ -17,6 +17,9 @@ depends: [
   "odoc" {with-doc}
   "mdx" {>= "1.9.0" & with-test}
 ]
+depopts: [
+  "domain-local-await" {>= "0.2"}
+]
 build: [
   ["dune" "subst"] {dev}
   [

src/d_pool_.ml

@@ -1,6 +1,7 @@
 type domain = Domain_.t
 
 let work_ _i q : unit =
+  Dla_.setup_domain ();
   while true do
     let f = Bb_queue.pop q in
     try f () with _ -> ()

src/dla_.dummy.ml

@@ -0,0 +1,13 @@
+(** Interface to Domain-local-await.
+
+    This is used to handle the presence or absence of DLA. *)
+
+type t = {
+  release: unit -> unit;
+  await: unit -> unit;
+}
+
+let using : prepare_for_await:(unit -> t) -> while_running:(unit -> 'a) -> 'a =
+ fun ~prepare_for_await:_ ~while_running -> while_running ()
+
+let setup_domain () = ()

src/dla_.real.ml

@@ -0,0 +1,9 @@
+type t = Domain_local_await.t = {
+  release: unit -> unit;
+  await: unit -> unit;
+}
+
+let using : prepare_for_await:(unit -> t) -> while_running:(unit -> 'a) -> 'a =
+  Domain_local_await.using
+
+let setup_domain () = Domain_local_await.per_thread (module Thread)

src/dune

@@ -2,7 +2,10 @@
  (public_name moonpool)
  (name moonpool)
  (private_modules d_pool_)
- (libraries threads either))
+ (libraries threads either
+            (select dla_.ml from
+             (domain-local-await -> dla_.real.ml)
+             ( -> dla_.dummy.ml))))
 
 (rule
   (targets atomic_.ml)