Update PHPMem.php

Author: RobiNN1

src/Dashboards/Memcached/PHPMem.php

@@ -14,7 +14,12 @@
 use function str_ends_with;
 
 class PHPMem {
-    public const VERSION = '2.0.0';
+    public const VERSION = '2.0.1';
+
+    /**
+     * @var resource|null
+     */
+    private $stream;
 
     /**
      * @param array<string, int|string> $server
@@ -279,6 +284,36 @@ public function version(): string {
         return str_replace('VERSION ', '', $this->runCommand('version'));
     }
 
+    /**
+     * @throws MemcachedException
+     */
+    private function connect(): void {
+        if (is_resource($this->stream)) {
+            return; // Already connected
+        }
+
+        $address = isset($this->server['path']) ? 'unix://'.$this->server['path'] : 'tcp://'.$this->server['host'].':'.$this->server['port'];
+        $stream = @stream_socket_client($address, $error_code, $error_message, 0.5);
+
+        if ($stream === false) {
+            throw new MemcachedException('Could not connect: '.$error_code.' - '.$error_message);
+        }
+
+        stream_set_timeout($stream, 1);
+        $this->stream = $stream;
+    }
+
+    public function __destruct() {
+        $this->disconnect();
+    }
+
+    public function disconnect(): void {
+        if (is_resource($this->stream)) {
+            fclose($this->stream);
+            $this->stream = null;
+        }
+    }
+
     /**
      * Run command.
      *
@@ -335,20 +370,19 @@ public function runCommand(string $command): string {
      * @throws MemcachedException
      */
     private function streamConnection(string $command, string $command_name): string {
-        $address = isset($this->server['path']) ? 'unix://'.$this->server['path'] : 'tcp://'.$this->server['host'].':'.$this->server['port'];
-        $stream = @stream_socket_client($address, $error_code, $error_message, 0.5);
+        $this->connect();
 
-        if ($stream === false) {
-            throw new MemcachedException('Command: "'.$command.'": '.$error_code.' - '.$error_message);
+        if (!is_resource($this->stream) || feof($this->stream)) {
+            $this->disconnect();
+            $this->connect();
         }
 
-        stream_set_timeout($stream, 1);
-        fwrite($stream, $command);
+        fwrite($this->stream, $command);
 
         $buffer = '';
 
-        while (!feof($stream)) {
-            $line = fgets($stream, 4096);
+        while (!feof($this->stream)) {
+            $line = fgets($this->stream, 4096);
 
             if ($line === false) {
                 break;
@@ -370,8 +404,6 @@ private function streamConnection(string $command, string $command_name): string
             }
         }
 
-        fclose($stream);
-
         return $buffer;
     }