CP-52039: Drop Semaphore from Xapi_stdext_threads

ocaml/libs/http-lib/http_svr.ml

@@ -648,7 +648,7 @@ let start ?header_read_timeout ?header_total_timeout ?max_header_length
     ; body=
         handle_connection ~header_read_timeout ~header_total_timeout
           ~max_header_length x
-    ; lock= Xapi_stdext_threads.Semaphore.create conn_limit
+    ; lock= Semaphore.Counting.make conn_limit
     }
   in
   let server = Server_io.server handler socket in

ocaml/libs/http-lib/server_io.ml

@@ -23,15 +23,15 @@ type handler = {
     name: string
   ; (* body should close the provided fd *)
     body: Unix.sockaddr -> Unix.file_descr -> unit
-  ; lock: Xapi_stdext_threads.Semaphore.t
+  ; lock: Semaphore.Counting.t
 }
 
 let handler_by_thread (h : handler) (s : Unix.file_descr)
     (caller : Unix.sockaddr) =
   Thread.create
     (fun () ->
       Fun.protect
-        ~finally:(fun () -> Xapi_stdext_threads.Semaphore.release h.lock 1)
+        ~finally:(fun () -> Semaphore.Counting.release h.lock)
         (Debug.with_thread_named h.name (fun () -> h.body caller s))
     )
     ()
@@ -49,7 +49,7 @@ let establish_server ?(signal_fds = []) forker handler sock =
       @@ Polly.wait epoll 2 (-1) (fun _ fd _ ->
              (* If any of the signal_fd is active then bail out *)
              if List.mem fd signal_fds then raise PleaseClose ;
-             Xapi_stdext_threads.Semaphore.acquire handler.lock 1 ;
+             Semaphore.Counting.acquire handler.lock ;
              let s, caller = Unix.accept ~cloexec:true sock in
              try ignore (forker handler s caller)
              with exc ->

ocaml/libs/http-lib/server_io.mli

@@ -16,7 +16,7 @@ type handler = {
     name: string  (** used for naming the thread *)
   ; body: Unix.sockaddr -> Unix.file_descr -> unit
         (** function called in a thread for each connection*)
-  ; lock: Xapi_stdext_threads.Semaphore.t
+  ; lock: Semaphore.Counting.t
 }
 
 type server = {

ocaml/libs/xapi-stdext/lib/xapi-stdext-threads/threadext.ml

@@ -14,11 +14,20 @@
 
 module M = Mutex
 
+let finally = Xapi_stdext_pervasives.Pervasiveext.finally
+
 module Mutex = struct
   (** execute the function f with the mutex hold *)
   let execute lock f =
     Mutex.lock lock ;
-    Xapi_stdext_pervasives.Pervasiveext.finally f (fun () -> Mutex.unlock lock)
+    finally f (fun () -> Mutex.unlock lock)
+end
+
+module Semaphore = struct
+  let execute s f =
+    let module Semaphore = Semaphore.Counting in
+    Semaphore.acquire s ;
+    finally f (fun () -> Semaphore.release s)
 end
 
 (** Parallel List.iter. Remembers all exceptions and returns an association list mapping input x to an exception.
@@ -60,7 +69,6 @@ module Delay = struct
   exception Pre_signalled
 
   let wait (x : t) (seconds : float) =
-    let finally = Xapi_stdext_pervasives.Pervasiveext.finally in
     let to_close = ref [] in
     let close' fd =
       if List.mem fd !to_close then Unix.close fd ;

ocaml/libs/xapi-stdext/lib/xapi-stdext-threads/threadext.mli

@@ -15,6 +15,10 @@ module Mutex : sig
   val execute : Mutex.t -> (unit -> 'a) -> 'a
 end
 
+module Semaphore : sig
+  val execute : Semaphore.Counting.t -> (unit -> 'a) -> 'a
+end
+
 val thread_iter_all_exns : ('a -> unit) -> 'a list -> ('a * exn) list
 
 val thread_iter : ('a -> unit) -> 'a list -> unit

ocaml/networkd/lib/network_utils.ml

@@ -1197,12 +1197,13 @@ module Ovs = struct
 
     val appctl : ?log:bool -> string list -> string
   end = struct
-    module Semaphore = Xapi_stdext_threads.Semaphore
+    module Semaphore = Semaphore.Counting
 
-    let s = Semaphore.create 5
+    let s = Semaphore.make 5
 
     let vsctl ?log args =
-      Semaphore.execute s (fun () ->
+      let execute = Xapi_stdext_threads.Threadext.Semaphore.execute in
+      execute s (fun () ->
           call_script ~on_error:error_handler ?log ovs_vsctl
             ("--timeout=20" :: args)
       )

ocaml/xapi-aux/throttle.ml

@@ -17,10 +17,12 @@ module type SIZE = sig
 end
 
 module Make (Size : SIZE) = struct
-  module Semaphore = Xapi_stdext_threads.Semaphore
+  module Semaphore = Semaphore.Counting
 
   let with_lock = Xapi_stdext_threads.Threadext.Mutex.execute
 
+  let execute = Xapi_stdext_threads.Threadext.Semaphore.execute
+
   let semaphore = ref None
 
   let m = Mutex.create ()
@@ -29,11 +31,11 @@ module Make (Size : SIZE) = struct
     with_lock m @@ fun () ->
     match !semaphore with
     | None ->
-        let result = Semaphore.create (Size.n ()) in
+        let result = Semaphore.make (Size.n ()) in
         semaphore := Some result ;
         result
     | Some s ->
         s
 
-  let execute f = Semaphore.execute (get_semaphore ()) f
+  let execute f = execute (get_semaphore ()) f
 end

ocaml/xapi/xapi_sr.ml

@@ -20,7 +20,6 @@ module Rrdd = Rrd_client.Client
 let with_lock = Xapi_stdext_threads.Threadext.Mutex.execute
 
 module Listext = Xapi_stdext_std.Listext
-module Semaphore = Xapi_stdext_threads.Semaphore
 module Unixext = Xapi_stdext_unix.Unixext
 
 let finally = Xapi_stdext_pervasives.Pervasiveext.finally