middleware: split out proxy class

It's easier to find this way.

1 files changed, 26 insertions, 0 deletions

diff --git a/lib/raindrops/middleware/proxy.rb b/lib/raindrops/middleware/proxy.rb
new file mode 100644
index 0000000..8b2c0c8
--- /dev/null
+++ b/lib/raindrops/middleware/proxy.rb
@@ -0,0 +1,26 @@
+# -*- encoding: binary -*-
+class Raindrops::Middleware::Proxy
+  def initialize(body, stats)
+    @body, @stats = body, stats
+  end
+
+  # yield to the Rack server here for writing
+  def each
+    @body.each { |x| yield x }
+  end
+
+  # the Rack server should call this after #each (usually ensure-d)
+  def close
+    @stats.decr_writing
+    @body.close if @body.respond_to?(:close)
+  end
+
+  def to_path
+    @body.to_path
+  end
+
+  def respond_to?(m)
+    m = m.to_sym
+    :close == m || @body.respond_to?(m)
+  end
+end