Merge pull request #105 from Asmod4n/patch-3

Raise SocketError on all possible Socket Errors
celluloid · Sep 27, 2013 · 74c5aeb · 74c5aeb
2 parents ea7124b + db25be7
commit 74c5aeb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/reel/stream.rb b/lib/reel/stream.rb
@@ -12,7 +12,7 @@ def call(socket)
     def write(data)
       @socket << data
       data
-    rescue Errno::EPIPE
+    rescue IOError, Errno::ECONNRESET, Errno::EPIPE
       raise SocketError, "error writing to socket"
     end
     alias :<< :write

diff --git a/lib/reel/websocket.rb b/lib/reel/websocket.rb
@@ -79,7 +79,7 @@ def body
     def write(msg)
       @socket << ::WebSocket::Message.new(msg).to_data
       msg
-    rescue Errno::EPIPE
+    rescue IOError, Errno::ECONNRESET, Errno::EPIPE
       cancel_timer!
       raise SocketError, "error writing to socket"
     rescue