lib/rainbows/coolio_thread_pool.rb

   1 # -*- encoding: binary -*-
   2
   3 # A combination of the Coolio and ThreadPool models.  This allows Ruby
   4 # Thread-based concurrency for application processing.  It DOES NOT
   5 # expose a streamable "rack.input" for upload processing within the
   6 # app.  DevFdResponse should be used with this class to proxy
   7 # asynchronous responses.  All network I/O between the client and
   8 # server are handled by the main thread and outside of the core
   9 # application dispatch.
  10 #
  11 # Unlike ThreadPool, Cool.io makes this model highly suitable for
  12 # slow clients and applications with medium-to-slow response times
  13 # (I/O bound), but less suitable for sleepy applications.
  14 #
  15 # This concurrency model is designed for Ruby 1.9, and Ruby 1.8
  16 # users are NOT advised to use this due to high CPU usage.
  17 module Rainbows::CoolioThreadPool
  18   # :stopdoc:
  19   DEFAULTS = {
  20     :pool_size => 20, # same default size as ThreadPool (w/o Coolio)
  21   }
  22   #:startdoc:
  23
  24   def self.setup # :nodoc:
  25     o = Rainbows::O
  26     DEFAULTS.each { |k,v| o[k] ||= v }
  27     Integer === o[:pool_size] && o[:pool_size] > 0 or
  28       raise ArgumentError, "pool_size must a be an Integer > 0"
  29   end
  30   include Rainbows::Coolio::Core
  31
  32   def init_worker_threads(master, queue) # :nodoc:
  33     Rainbows::O[:pool_size].times.map do
  34       Thread.new do
  35         begin
  36           client = queue.pop
  37           master << [ client, client.app_response ]
  38         rescue => e
  39           Rainbows::Error.listen_loop(e)
  40         end while true
  41       end
  42     end
  43   end
  44
  45   def init_worker_process(worker) # :nodoc:
  46     super
  47     cloop = Coolio::Loop.default
  48     master = Rainbows::Coolio::Master.new(Queue.new).attach(cloop)
  49     queue = Client.const_set(:QUEUE, Queue.new)
  50     threads = init_worker_threads(master, queue)
  51     Watcher.new(threads).attach(cloop)
  52     logger.info "CoolioThreadPool pool_size=#{Rainbows::O[:pool_size]}"
  53   end
  54 end
  55 # :enddoc:
  56 require 'rainbows/coolio_thread_pool/client'
  57 require 'rainbows/coolio_thread_pool/watcher'