aboutsummaryrefslogtreecommitdiffhomepage
path: root/stm/workerPool.mli
blob: de396d85b010939cb7e6b0c38afed23158523aeb (plain)
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
36
37
38
39
40
41
42
43
44
45
46
(************************************************************************)
(*  v      *   The Coq Proof Assistant  /  The Coq Development Team     *)
(* <O___,, *   INRIA - CNRS - LIX - LRI - PPS - Copyright 1999-2017     *)
(*   \VV/  **************************************************************)
(*    //   *      This file is distributed under the terms of the       *)
(*         *       GNU Lesser General Public License Version 2.1        *)
(************************************************************************)

type worker_id = string

type 'a cpanel = {
  exit : unit -> unit; (* called by manager to exit instead of Thread.exit *)
  cancelled : unit -> bool; (* manager checks for a request of termination *)
  extra : 'a;                        (* extra stuff to pass to the manager *)
}

module type PoolModel = sig
  (* this shall come from a Spawn.* model *)
  type process
  val spawn : int -> worker_id * process * CThread.thread_ic * out_channel

  (* this defines the main loop of the manager *)
  type extra
  val manager :
    extra cpanel -> worker_id * process * CThread.thread_ic * out_channel -> unit
end

module Make(Model : PoolModel) : sig

  type pool
  
  val create : Model.extra -> size:int -> pool

  val is_empty : pool -> bool
  val n_workers : pool -> int

  (* cancel signal *)
  val cancel : worker_id -> pool -> unit
  val cancel_all : pool -> unit
  (* camcel signal + true removal, the pool is empty afterward *)
  val destroy : pool -> unit

  (* The worker should call this function *)
  val worker_handshake : CThread.thread_ic -> out_channel -> unit

end