blob: 45ee3ae4e37e9203da2ea7e6bb2532c7f967a27a (
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
|
# frozen-string-literal: true
module Plum
module Rack
class ThreadPool
def initialize(size)
@workers = Set.new
@jobs = Queue.new
size.times { |i|
spawn_worker
}
end
# returns cancel token
def acquire(tag = nil, err = nil, &blk)
@jobs << [blk, err]
end
private
def spawn_worker
t = Thread.new {
while true
job, err = @jobs.pop
begin
job.call
rescue => e
err << e if err
end
end
}
@workers << t
end
end
end
end
|