about summary refs log tree commit homepage
path: root/lib
diff options
context:
space:
mode:
authorEric Wong <normalperson@yhbt.net>2010-12-08 18:35:27 -0800
committerEric Wong <normalperson@yhbt.net>2010-12-08 18:35:27 -0800
commit6b750f5f952963009a2e6e8702fc8f3d8adc94ea (patch)
tree3debc215a179f416883aae3c02fd4ea4a4b5a322 /lib
parent10d96a76a1ea4431dd10ba181d747169c22c1cec (diff)
downloadrainbows-6b750f5f952963009a2e6e8702fc8f3d8adc94ea.tar.gz
This was completely overlooked for the Rainbows 2.0.x
releases.
Diffstat (limited to 'lib')
-rw-r--r--lib/rainbows/base.rb1
-rw-r--r--lib/rainbows/process_client.rb9
-rw-r--r--lib/rainbows/rack_input.rb17
-rw-r--r--lib/rainbows/revactor.rb3
4 files changed, 23 insertions, 7 deletions
diff --git a/lib/rainbows/base.rb b/lib/rainbows/base.rb
index 3027aba..bf9ef87 100644
--- a/lib/rainbows/base.rb
+++ b/lib/rainbows/base.rb
@@ -18,6 +18,7 @@ module Rainbows::Base
     super(worker)
     Rainbows::Response.setup(self.class)
     Rainbows::MaxBody.setup
+    Rainbows::RackInput.setup
     G.tmp = worker.tmp
 
     listeners = Rainbows::HttpServer::LISTENERS
diff --git a/lib/rainbows/process_client.rb b/lib/rainbows/process_client.rb
index 3055b9d..271185d 100644
--- a/lib/rainbows/process_client.rb
+++ b/lib/rainbows/process_client.rb
@@ -1,12 +1,11 @@
 # -*- encoding: binary -*-
 # :enddoc:
+require 'rainbows/rack_input'
 module Rainbows::ProcessClient
   G = Rainbows::G
   include Rainbows::Response
   HttpParser = Unicorn::HttpParser
-  NULL_IO = Unicorn::HttpRequest::NULL_IO
-  RACK_INPUT = Unicorn::HttpRequest::RACK_INPUT
-  TeeInput = Unicorn::TeeInput
+  include Rainbows::RackInput
   include Rainbows::Const
 
   # once a client is accepted, it is processed in its entirety here
@@ -25,9 +24,7 @@ module Rainbows::ProcessClient
         buf << buf2
       end
 
-      env[CLIENT_IO] = client
-      env[RACK_INPUT] = 0 == hp.content_length ?
-                        NULL_IO : TeeInput.new(client, hp)
+      set_input(env, hp, client)
       env[REMOTE_ADDR] = remote_addr
       status, headers, body = APP.call(env.update(RACK_DEFAULTS))
 
diff --git a/lib/rainbows/rack_input.rb b/lib/rainbows/rack_input.rb
new file mode 100644
index 0000000..df51ac1
--- /dev/null
+++ b/lib/rainbows/rack_input.rb
@@ -0,0 +1,17 @@
+# -*- encoding: binary -*-
+# :enddoc:
+# only used by synchronous interfaces
+module Rainbows::RackInput
+  NULL_IO = Unicorn::HttpRequest::NULL_IO
+  RACK_INPUT = Unicorn::HttpRequest::RACK_INPUT
+  CLIENT_IO = Rainbows::Const::CLIENT_IO
+
+  def self.setup
+    const_set(:IC, Unicorn::HttpRequest.input_class)
+  end
+
+  def set_input(env, hp, client)
+    env[RACK_INPUT] = 0 == hp.content_length ? NULL_IO : IC.new(client, hp)
+    env[CLIENT_IO] = client
+  end
+end
diff --git a/lib/rainbows/revactor.rb b/lib/rainbows/revactor.rb
index 51f6c1f..a181df1 100644
--- a/lib/rainbows/revactor.rb
+++ b/lib/rainbows/revactor.rb
@@ -53,7 +53,7 @@ module Rainbows::Revactor
 
       env[CLIENT_IO] = client
       env[RACK_INPUT] = 0 == hp.content_length ?
-               NULL_IO : Unicorn::TeeInput.new(ts = TeeSocket.new(client), hp)
+               NULL_IO : IC.new(ts = TeeSocket.new(client), hp)
       env[REMOTE_ADDR] = remote_addr
       status, headers, body = app.call(env.update(RACK_DEFAULTS))
 
@@ -87,6 +87,7 @@ module Rainbows::Revactor
     init_worker_process(worker)
     require 'rainbows/revactor/body'
     self.class.__send__(:include, Rainbows::Revactor::Body)
+    self.class.const_set(:IC, Unicorn::HttpRequest.input_class)
     RD_ARGS[:timeout] = G.kato if G.kato > 0
     nr = 0
     limit = worker_connections