about summary refs log tree commit homepage
path: root/lib/rainbows/rev/client.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/rainbows/rev/client.rb')
-rw-r--r--lib/rainbows/rev/client.rb5
1 files changed, 4 insertions, 1 deletions
diff --git a/lib/rainbows/rev/client.rb b/lib/rainbows/rev/client.rb
index 2242c18..e2095bb 100644
--- a/lib/rainbows/rev/client.rb
+++ b/lib/rainbows/rev/client.rb
@@ -50,11 +50,14 @@ module Rainbows
       # here to prevent connections from being closed on us.
       def defer_body(io)
         @deferred_bodies << io
-        schedule_write
+        @_write_buffer.empty? ? on_write_complete : schedule_write
       end
 
       def next
         @deferred_bodies.shift
+        if :close == @state && @deferred_bodies.empty? && @_write_buffer.empty?
+          close
+        end
       end
 
       def timeout?