Class: Cosmo::Utils::ThreadPool

Inherits:
Object
  • Object
show all
Extended by:
Forwardable
Defined in:
lib/cosmo/utils/thread_pool.rb,
sig/cosmo/utils/thread_pool.rbs

Overview

A thread pool that reuses a fixed number of threads operating off a fixed size queue. At any point, at most num_threads will be active processing tasks. When all threads are busy new tasks posted to the thread pool are blocked until a thread becomes available. Should a thread crash for any reason the thread will immediately be removed from the pool and replaced.

Instance Method Summary collapse

Constructor Details

#initialize(concurrency) ⇒ ThreadPool

Returns a new instance of ThreadPool.

Parameters:

  • concurrency (Integer) 


17
18
19
20
21
22
 
# File 'lib/cosmo/utils/thread_pool.rb', line 17

def initialize(concurrency)
  @mutex = Thread::Mutex.new
  @available = concurrency
  @cond = ConditionVariable.new
  @pool = Concurrent::FixedThreadPool.new(concurrency)
end

Instance Method Details

#post { ... } ⇒ void

This method returns an undefined value.

Yields:

Yield Returns:

  • (void) 


24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/cosmo/utils/thread_pool.rb', line 24

def post
  @mutex.synchronize do
    @cond.wait(@mutex) while @available <= 0
    @available -= 1
  end

  @pool.post do
    yield
  ensure
    @mutex.synchronize do
      @available += 1
      @cond.signal
    end
  end
end

#shutdown ⇒ void

This method returns an undefined value.



13
 
# File 'sig/cosmo/utils/thread_pool.rbs', line 13

def shutdown: () -> void

#wait_for_termination ⇒ Boolean

Parameters:

  • timeout (Integer, nil)

Returns:

  • (Boolean) 


15
 
# File 'sig/cosmo/utils/thread_pool.rbs', line 15

def wait_for_termination: (Integer? timeout) -> bool