Add internal Q.add helper

polytypic · polytypic · commit 26a190b9a9b5 · 2024-08-21T19:28:50.000+03:00
diff --git a/lib/picos_sync/condition.ml b/lib/picos_sync/condition.ml
@@ -40,11 +40,7 @@ let rec cleanup backoff trigger (t : t) =
 
 let rec wait (t : t) mutex trigger fiber backoff =
   let before = Atomic.get t in
-  let after =
-    match before with
-    | T Zero -> Q.singleton trigger
-    | T (One _ as q) -> Q.snoc q trigger
-  in
+  let after = Q.add before trigger in
   if Atomic.compare_and_set t before after then begin
     Mutex.unlock_as (Fiber.Maybe.of_fiber fiber) mutex Backoff.default;
     let result = Trigger.await trigger in
diff --git a/lib/picos_sync/mutex.ml b/lib/picos_sync/mutex.ml
@@ -73,11 +73,7 @@ let rec lock_as fiber t entry backoff =
               Entry { trigger; fiber }
           | Entry _ as entry -> entry
         in
-        let waiters =
-          match r.waiters with
-          | T Zero -> Q.singleton entry
-          | T (One _ as q) -> Q.snoc q entry
-        in
+        let waiters = Q.add r.waiters entry in
         let after = Locked { fiber = r.fiber; waiters } in
         if Atomic.compare_and_set t before after then begin
           match Trigger.await entry_r.trigger with
diff --git a/lib/picos_sync/q.ml b/lib/picos_sync/q.ml
@@ -34,6 +34,9 @@ let[@inline] snoc (One o as t : (_, _) one) value =
   let cons = Cons { value; next = S Nil } in
   T (One { head = o.head; tail = o.cons; cons })
 
+let[@inline] add t value =
+  match t with T Zero -> singleton value | T (One _ as o) -> snoc o value
+
 let[@inline] head (One { head = Cons hd; _ } : (_, _) one) = hd.value
 
 let[@inline] tail (One o as t : (_, _) one) =
