allow manual close of client connection - closes igrigorik#11

snyff · May 8, 2011 · 2924095 · 2924095
1 parent a1b44cc
commit 2924095
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/lib/em-proxy/connection.rb b/lib/em-proxy/connection.rb
@@ -88,15 +88,15 @@ def unbind
       def unbind_backend(name)
         debug [:unbind_backend, name]
         @servers[name] = nil
-
-        # if all connections are terminated downstream, then notify client
-        close_connection_after_writing if @servers.values.compact.size.zero?
+        close = :close
 
         if @on_finish
-          @on_finish.call(name)
+          close = @on_finish.call(name)
+        end
 
-          # not sure if this is required
-          # @on_finish.call(:done) if @servers.values.compact.size.zero?
+        # if all connections are terminated downstream, then notify client
+        if @servers.values.compact.size.zero? and close != :keep
+          close_connection_after_writing
         end
       end