simon/moonpool
1
0
Fork 0
mirror of https://github.com/c-cube/moonpool.git synced 2025-12-06 03:05:30 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

perf: make run simple again

Browse source
This commit is contained in:
Simon Cruanes 2023-06-07 21:19:04 -04:00
parent 327045f5c5
commit 1e81588c9a
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
1 changed files with 4 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
src/pool.ml
View file

@ -26,30 +26,10 @@ let add_global_thread_loop_wrapper f : unit =
exception Shutdown exception Shutdown
let run (self : t) (f : task) : unit = let run self f : unit =
let n_qs = Array.length self.qs in let i = A.fetch_and_add self.cur_q 1 in
let offset = A.fetch_and_add self.cur_q 1 in let q = self.qs.(i mod Array.length self.qs) in
try Bb_queue.push q f with Bb_queue.Closed -> raise Shutdown
(* blocking push, last resort *)
let push_wait () =
let q_idx = offset mod Array.length self.qs in
let q = self.qs.(q_idx) in
Bb_queue.push q f
in
try
(* try each queue with a round-robin initial offset *)
for _retry = 1 to 3 do
for i = 0 to n_qs - 1 do
let q_idx = (i + offset) mod Array.length self.qs in
let q = self.qs.(q_idx) in
if Bb_queue.try_push q f then raise_notrace Exit
done
done;
push_wait ()
with
| Exit -> ()
| Bb_queue.Closed -> raise Shutdown
let size self = Array.length self.threads let size self = Array.length self.threads