Rescue EOFError when reading on socket.

httprb · Mar 4, 2014 · 78036f3 · 78036f3
1 parent 50cf6e0
commit 78036f3
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/http/client.rb b/lib/http/client.rb
@@ -68,8 +68,12 @@ def readpartial(size = BUFFER_SIZE) # rubocop:disable CyclomaticComplexity
 
       chunk = @parser.chunk
       unless chunk
-        @parser << @socket.readpartial(BUFFER_SIZE)
-        chunk = @parser.chunk
+        begin
+          @parser << @socket.readpartial(BUFFER_SIZE)
+          chunk = @parser.chunk
+        rescue EOFError
+          chunk = nil
+        end
 
         # TODO: consult @body_remaining here and raise if appropriate
         return unless chunk