From 91df761990fa8a59e7ac96ce14a17695b4fbdb67 Mon Sep 17 00:00:00 2001
From: James Coglan <jcoglan@gmail.com>
Date: Tue, 7 Jul 2015 20:33:22 +0100
Subject: [PATCH] Hybi#shutdown doesn't really need to take a block, it can
 just take a flag to say whether to emit an error.

---
 lib/websocket/driver/hybi.rb | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/lib/websocket/driver/hybi.rb b/lib/websocket/driver/hybi.rb
index e2ab43e..91b0404 100644
--- a/lib/websocket/driver/hybi.rb
+++ b/lib/websocket/driver/hybi.rb
@@ -255,7 +255,7 @@ module WebSocket
         headers.join("\r\n")
       end
 
-      def shutdown(code, reason, &error)
+      def shutdown(code, reason, error = false)
         @frame = @message = nil
         @stage = 5
         @extensions.close
@@ -263,13 +263,13 @@ module WebSocket
         frame(reason, :close, code) if @ready_state < 2
         @ready_state = 3
 
-        error.call if error
+        emit(:error, ProtocolError.new(reason)) if error
         emit(:close, CloseEvent.new(code, reason))
       end
 
       def fail(type, message)
         return if @ready_state > 1
-        shutdown(ERRORS[type], message) { emit(:error, ProtocolError.new(message)) }
+        shutdown(ERRORS[type], message, true)
       end
 
       def parse_opcode(data)