Threadpool

threadpool.rb

class ThreadPool
 
  def initialize(size)
    @size = size
    @queue = Queue.new
    @pool = Array.new(@size) do |i|
      Thread.new do
        Thread.current.abort_on_exception = true
        Thread.current[:id] = i
        catch(:exit) do
          loop do
            job, args = @queue.pop
            job.call(*args)
          end
        end
      end
    end
  end
 
  def schedule(*args, &block)
    @queue.push([block, args])
  end
 
  def inspect
    "#{self.class}<Queue size: #{@queue.size}>"
  end
 
  def shutdown
    @size.times do
      schedule { throw :exit }
    end
    wait
  end

  private
  def wait
    @pool.map(&:join)
  end
 
end