class SupplyDrop::ThreadPool

Public Class Methods

new(size) click to toggle source
# File lib/supply_drop/thread_pool.rb, line 5
def initialize(size)
  @size = size
  @jobs = Queue.new
  @retvals = []

  @pool = Array.new(@size) do |i|
    Thread.new do
      Thread.current[:id] = i

      catch(:exit) do
        loop do
          job, args = @jobs.pop
          @retvals << job.call(*args)
        end
      end
    end
  end
end

Public Instance Methods

schedule(*args, &block) click to toggle source
# File lib/supply_drop/thread_pool.rb, line 25
def schedule(*args, &block)
  @jobs << [block, args]
end
shutdown() click to toggle source
# File lib/supply_drop/thread_pool.rb, line 30
def shutdown
  @size.times do
    schedule { throw :exit }
  end

  @pool.map(&:join)
  @retvals
end