1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

(* SPDX-License-Identifier: AGPL-3.0-or-later *)
(* Copyright © 2021-2024 OCamlPro *)
(* Written by the Owi programmers *)

module Stack = Prelude.Stack

type 'a t = ('a, Prio.t * 'a) Synchronizer.t

let pop s pledge = Synchronizer.get s pledge

let make_pledge = Synchronizer.make_pledge

let end_pledge = Synchronizer.end_pledge

let rec read_as_seq (s : 'a t) ~finalizer : 'a Seq.t =
 fun () ->
  match pop s false with
  | None ->
    finalizer ();
    Nil
  | Some v -> Cons (v, read_as_seq s ~finalizer)

let push v prio s = Synchronizer.write (prio, v) s

let work_while f s = Synchronizer.work_while f s

let fail = Synchronizer.fail

let make () =
  let s = Stack.create () in
  let writter prio_v condvar =
    Stack.push (snd prio_v) s;
    Condition.signal condvar
  in
  Synchronizer.init (fun () -> Stack.pop_opt s) writter