diff options
Diffstat (limited to 'lib/rainbows/response/body.rb')
-rw-r--r-- | lib/rainbows/response/body.rb | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/lib/rainbows/response/body.rb b/lib/rainbows/response/body.rb index 0a2bb5d..9e36412 100644 --- a/lib/rainbows/response/body.rb +++ b/lib/rainbows/response/body.rb @@ -88,9 +88,11 @@ module Rainbows::Response::Body # :nodoc: ensure body.respond_to?(:close) && inp != body and body.close end - else + elsif method_defined?(:write_body_stream) def write_body_path(sock, body) - write_body_stream(sock, body_to_io(body)) + write_body_stream(sock, inp = body_to_io(body)) + ensure + body.respond_to?(:close) && inp != body and body.close end end |