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

pool: do some spinning in run, not in workers

Browse source
This commit is contained in:
Simon Cruanes 2023-06-08 14:49:28 -04:00
parent 939c6e117c
commit 82af5497f8
No known key found for this signature in database
GPG key ID: EBFFF6F283F3A2B4
1 changed files with 6 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/pool.ml
View file

@ -39,7 +39,7 @@ let run (self : t) (f : task) : unit =
try try
(* try each queue with a round-robin initial offset *) (* try each queue with a round-robin initial offset *)
for _retry = 1 to 3 do for _retry = 1 to 10 do
for i = 0 to n_qs - 1 do for i = 0 to n_qs - 1 do
let q_idx = (i + offset) mod Array.length self.qs in let q_idx = (i + offset) mod Array.length self.qs in
let q = self.qs.(q_idx) in let q = self.qs.(q_idx) in
@ -69,13 +69,11 @@ let worker_thread_ ~on_exn (active : bool A.t) (qs : task Bb_queue.t array)
let task = let task =
try try
for _retry = 1 to 3 do for i = 0 to num_qs - 1 do
for i = 0 to num_qs - 1 do let q = qs.((offset + i) mod num_qs) in
let q = qs.((offset + i) mod num_qs) in match Bb_queue.try_pop ~force_lock:false q with
match Bb_queue.try_pop ~force_lock:false q with | Some f -> raise_notrace (Got_task f)
| Some f -> raise_notrace (Got_task f) | None -> ()
| None -> ()
done
done; done;
pop_blocking () pop_blocking ()
with Got_task f -> f with Got_task f -> f