diff options
Diffstat (limited to 'actionpack/lib/action_dispatch/middleware/body_proxy.rb')
-rw-r--r-- | actionpack/lib/action_dispatch/middleware/body_proxy.rb | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/actionpack/lib/action_dispatch/middleware/body_proxy.rb b/actionpack/lib/action_dispatch/middleware/body_proxy.rb new file mode 100644 index 0000000000..867afe3b48 --- /dev/null +++ b/actionpack/lib/action_dispatch/middleware/body_proxy.rb @@ -0,0 +1,28 @@ +# Keep this file meanwhile https://github.com/rack/rack/pull/313 is not released +module ActionDispatch + class BodyProxy + def initialize(body, &block) + @body, @block, @closed = body, block, false + end + + def respond_to?(*args) + super or @body.respond_to?(*args) + end + + def close + return if @closed + @closed = true + @body.close if @body.respond_to? :close + ensure + @block.call + end + + def closed? + @closed + end + + def method_missing(*args, &block) + @body.__send__(*args, &block) + end + end +end |