Class: HQ::Tools::ThreadPool

Inherits:
Object
  • Object
show all
Defined in:
lib/hq/tools/thread-pool.rb

Instance Method Summary collapse

Constructor Details

#initialize ⇒ ThreadPool

Returns a new instance of ThreadPool.



6
7
8
9
 
# File 'lib/hq/tools/thread-pool.rb', line 6

def initialize
	require "thread"
	@tasks = Queue.new
end

Instance Method Details

#deinit_hook(&block) ⇒ Object 



42
43
44
 
# File 'lib/hq/tools/thread-pool.rb', line 42

def deinit_hook &block
	@deinit_hook = block
end

#init_hook(&block) ⇒ Object 



38
39
40
 
# File 'lib/hq/tools/thread-pool.rb', line 38

def init_hook &block
	@init_hook = block
end

#schedule(*args, &block) ⇒ Object 



34
35
36
 
# File 'lib/hq/tools/thread-pool.rb', line 34

def schedule *args, &block
	@tasks << [ block, args ]
end

#start(threads) ⇒ Object 



46
47
48
49
50
51
52
53
54
55
56
57
58
59
 
# File 'lib/hq/tools/thread-pool.rb', line 46

def start threads

	@threads =
		Array.new threads do

			Thread.new do
				worker_init
				worker_run
				worker_deinit
			end

		end

end

#stop ⇒ Object 



61
62
63
64
65
66
67
68
69
70
71
72
 
# File 'lib/hq/tools/thread-pool.rb', line 61

def stop

	@threads.size.times do
		schedule { throw :exit }
	end

	@threads.each do
		|thread|
		thread.join
	end

end

#worker_deinit ⇒ Object 



29
30
31
32
 
# File 'lib/hq/tools/thread-pool.rb', line 29

def worker_deinit
	return unless @deinit_hook
	@deinit_hook.call
end

#worker_init ⇒ Object 



11
12
13
14
 
# File 'lib/hq/tools/thread-pool.rb', line 11

def worker_init
	return unless @init_hook
	@init_hook.call
end

#worker_run ⇒ Object 



16
17
18
19
20
21
22
23
24
25
26
27
 
# File 'lib/hq/tools/thread-pool.rb', line 16

def worker_run

	catch :exit do

		loop do
			task, args = @tasks.pop
			task.call *args
		end

	end

end