From 4f05fb1a3b44f8eab1a9dda26d5b115f33a149cd Mon Sep 17 00:00:00 2001
From: Eric Wong <normalperson@yhbt.net>
Date: Mon, 29 Jun 2009 05:27:18 -0700
Subject: configurator: provide stream_input (true|false) option

The default is false because some applications were not
written to handle partial reads (even though IO#read allows
it, not just IO#readpartial).
---
 lib/unicorn/configurator.rb | 29 ++++++++++++++++++++++++++++-
 1 file changed, 28 insertions(+), 1 deletion(-)

(limited to 'lib/unicorn/configurator.rb')

diff --git a/lib/unicorn/configurator.rb b/lib/unicorn/configurator.rb
index a432f64..99a3c04 100644
--- a/lib/unicorn/configurator.rb
+++ b/lib/unicorn/configurator.rb
@@ -44,6 +44,7 @@ module Unicorn
       :preload_app => false,
       :stderr_path => nil,
       :stdout_path => nil,
+      :stream_input => false,
     }
 
     attr_reader :config_file #:nodoc:
@@ -64,8 +65,12 @@ module Unicorn
 
     def commit!(server, options = {}) #:nodoc:
       skip = options[:skip] || []
+      stream_input = @set.delete(:stream_input)
+      unless stream_input.nil?
+        Unicorn::HttpRequest::DEFAULTS[Const::STREAM_INPUT] = stream_input
+      end
       @set.each do |key, value|
-        (Symbol === value && value == :unset) and next
+        value == :unset and next
         skip.include?(key) and next
         setter = "#{key}="
         if server.respond_to?(setter)
@@ -249,6 +254,28 @@ module Unicorn
       end
     end
 
+    # Allow applications to stream input as it is being read from the
+    # network directly to the application.  Enabling this can allow
+    # real-time processing of request bodies as it is being sent by
+    # the client, useful for things like upload progress notification
+    # and tunneling arbitrary stream protocols via bidirectional chunked
+    # transfer encoding.
+    # This may not work with all applications because some broken
+    # applications assume env['rack.input'].read(size) always returns
+    # the requested amount of data.  This causes env['rack.input']#read
+    # to provide IO#readpartial semantics instead.  Some applications
+    # may also fully receive an input and never attempt to process it,
+    # causing clients confusion when they receive a response after
+    # only a partial request has been sent.
+    def stream_input(bool)
+      case bool
+      when TrueClass, FalseClass
+        @set[:stream_input] = bool
+      else
+        raise ArgumentError, "stream_input=#{bool.inspect} not a boolean"
+      end
+    end
+
     # Allow redirecting $stderr to a given path.  Unlike doing this from
     # the shell, this allows the unicorn process to know the path its
     # writing to and rotate the file if it is used for logging.  The
-- 
cgit v1.2.3-24-ge0c7