Wrap Socket

src/NsqSocket.php

@@ -0,0 +1,73 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Nsq;
+
+use Nsq\Exception\ConnectionFail;
+use Nsq\Socket\Socket;
+use PHPinnacle\Buffer\ByteBuffer;
+use Throwable;
+use const PHP_EOL;
+
+final class NsqSocket
+{
+    private Buffer $input;
+
+    private ByteBuffer $output;
+
+    public function __construct(
+        private Socket $socket,
+    ) {
+        $this->input = new Buffer();
+        $this->output = new ByteBuffer();
+    }
+
+    public function write(string $command, string $data = null): void
+    {
+        $this->output->append($command.PHP_EOL);
+
+        if (null !== $data) {
+            $this->output->appendUint32(\strlen($data));
+            $this->output->append($data);
+        }
+
+        $this->socket->write($this->output->flush());
+    }
+
+    public function wait(float $timeout): bool
+    {
+        return $this->socket->selectRead($timeout);
+    }
+
+    public function read(): Buffer
+    {
+        $buffer = $this->input;
+
+        $buffer->append(
+            $this->socket->read(Bytes::BYTES_SIZE),
+        );
+
+        if ('' === $buffer->bytes()) {
+            throw new ConnectionFail('Probably connection closed.');
+        }
+
+        $size = $buffer->consumeSize();
+
+        do {
+            $buffer->append(
+                $this->socket->read($size - $buffer->size()),
+            );
+        } while ($buffer->size() < $size);
+
+        return $buffer;
+    }
+
+    public function close(): void
+    {
+        try {
+            $this->socket->close();
+        } catch (Throwable) {
+        }
+    }
+}