MiniThreadPool - A minimum viable thread pool implementation for Ruby

mini-thread-pool.rb

class MiniThreadPool
  def initialize(size: 8, abort_on_exception: true)
    @queue = Queue.new
    @join_all = false
    @threads = []
    size.times do
      thr = Thread.new do
        loop do
          break if @join_all && @queue.empty?
          begin
            job = @queue.pop(true) # non-blocking dequeue
          rescue ThreadError => e
            # retry if queue is empty
            sleep 0.1
            next
          end
          job.call
        end
        # end of the thread
      end
      thr.abort_on_exception = abort_on_exception
      @threads << thr
    end
  end

  def enqueue(&job)
    @queue.push(job)
  end

  def join_all
    @join_all = true
    @threads.each do |thr|
      thr.join
    end
  end
end