about summary refs log tree commit homepage
path: root/lib/rainbows/process_client.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/rainbows/process_client.rb')
-rw-r--r--lib/rainbows/process_client.rb9
1 files changed, 3 insertions, 6 deletions
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))