Move try/catch from readFrame to read

src/Connection.php

@@ -7,9 +7,9 @@ namespace Nsq;
 use Nsq\Config\ClientConfig;
 use Nsq\Config\ConnectionConfig;
 use Nsq\Exception\AuthenticationRequired;
+use Nsq\Exception\BadResponse;
 use Nsq\Exception\ConnectionFail;
 use Nsq\Exception\NsqError;
-use Nsq\Exception\BadResponse;
 use Nsq\Exception\NsqException;
 use Nsq\Exception\NullReceived;
 use Nsq\Protocol\Error;
@@ -187,20 +187,7 @@ abstract class Connection
             return null;
         }
 
-        try {
-            $buffer = $this->read(Bytes::BYTES_SIZE);
-
-            if ('' === $buffer->bytes()) {
-                $this->disconnect();
-
-                throw new ConnectionFail('Probably connection lost');
-            }
-
-            $size = $buffer->consumeSize();
-
-            do {
-                $this->read($size);
-            } while ($buffer->size() < $size);
+        $buffer = $this->read();
 
         $this->logger->debug('Received buffer: '.addcslashes($buffer->bytes(), PHP_EOL));
 
@@ -224,14 +211,6 @@ abstract class Connection
                 ($currentTime = microtime(true)) > $deadline ? 0 : $deadline - $currentTime
             );
         }
-        }
-        // @codeCoverageIgnoreStart
-        catch (Exception $e) {
-            $this->disconnect();
-
-            throw ConnectionFail::fromThrowable($e);
-        }
-        // @codeCoverageIgnoreEnd
 
         return $frame;
     }
@@ -264,11 +243,38 @@ abstract class Connection
         throw new NsqException('Unreachable statement.');
     }
 
-    private function read(int $size): Buffer
+    private function read(): Buffer
     {
-        return $this->input->append(
-            $this->socket()->read($size),
+        try {
+            $socket = $this->socket();
+
+            $buffer = $this->input->append(
+                $socket->read(Bytes::BYTES_SIZE),
             );
+
+            if ('' === $buffer->bytes()) {
+                $this->disconnect();
+
+                throw new ConnectionFail('Probably connection lost');
+            }
+
+            $size = $buffer->consumeSize();
+
+            do {
+                $buffer->append(
+                    $socket->read($size),
+                );
+            } while ($buffer->size() < $size);
+
+            return $buffer;
+        }
+        // @codeCoverageIgnoreStart
+        catch (Exception $e) {
+            $this->disconnect();
+
+            throw ConnectionFail::fromThrowable($e);
+        }
+        // @codeCoverageIgnoreEnd
     }
 
     private function flush(): void