Handle network level disconnect

Author: appkit-framework

spec/generate.php

@@ -283,6 +283,13 @@ function amqpTypeToLength(string $type, string $e): array
 $connectionContent .= "        private readonly Configuration \$configuration,\n";
 $connectionContent .= "        private readonly Closure \$frameMax,\n";
 $connectionContent .= "    ) {\n";
+$connectionContent .= "        \$this->connection->on('close', function (): void {\n";
+$connectionContent .= "            if (!\$this->client->isConnected()) {\n";
+$connectionContent .= "                return;\n";
+$connectionContent .= "            }\n";
+$connectionContent .= "\n";
+$connectionContent .= "            \$this->client->disconnect(0, 'Connection lost', ClientInterface::RAW_CONNECTION_INACTIVE);\n";
+$connectionContent .= "        });\n";
 $connectionContent .= "        \$this->connection->on('data', function (string \$data): void {\n";
 $connectionContent .= "            \$this->readBuffer->append(\$data);\n";
 $connectionContent .= "\n";

src/Connection.php

@@ -67,6 +67,13 @@ public function __construct(
         private readonly Configuration $configuration,
         private readonly Closure $frameMax,
     ) {
+        $this->connection->on('close', function (): void {
+            if (!$this->client->isConnected()) {
+                return;
+            }
+
+            $this->client->disconnect(0, 'Connection lost', ClientInterface::RAW_CONNECTION_INACTIVE);
+        });
         $this->connection->on('data', function (string $data): void {
             $this->readBuffer->append($data);
 

test/ConnectionTest.php

@@ -189,4 +189,52 @@ static function (): int {
 
         self::assertSame($baseBuffer, $afterCloseBuffer);
     }
+
+    public function testConnectingEmittingCloseWillResultInFastClosure(): void
+    {
+        $client = new MockClientInterface(true);
+        $mockConnection = new MockConnectionInterface();
+        new Connection(
+            $client,
+            $mockConnection,
+            new Buffer(),
+            new Buffer(),
+            new ProtocolReader(),
+            new ProtocolWriter(),
+            new Channels(),
+            new Configuration(),
+            static function (): int {
+                return Constants::FRAME_MAX;
+            },
+        );
+
+        $mockConnection->emit('close');
+
+        self::assertSame(1, $client->getIsConnectedCount());
+        self::assertSame(1, $client->getDisconnectCount());
+    }
+
+    public function testConnectingEmittingCloseWillResultInFastClosureButNotWhenNotConnected(): void
+    {
+        $client = new MockClientInterface(false);
+        $mockConnection = new MockConnectionInterface();
+        new Connection(
+            $client,
+            $mockConnection,
+            new Buffer(),
+            new Buffer(),
+            new ProtocolReader(),
+            new ProtocolWriter(),
+            new Channels(),
+            new Configuration(),
+            static function (): int {
+                return Constants::FRAME_MAX;
+            },
+        );
+
+        $mockConnection->emit('close');
+
+        self::assertSame(1, $client->getIsConnectedCount());
+        self::assertSame(0, $client->getDisconnectCount());
+    }
 }

test/MockClientInterface.php

@@ -9,6 +9,7 @@
 
 final class MockClientInterface implements ClientInterface
 {
+    private int $isConnectedCount = 0;
     private int $disconnectCount = 0;
     private bool $isConnected;
 
@@ -29,9 +30,16 @@ public function disconnect(int $replyCode = 0, string $replyText = '', bool $con
 
     public function isConnected(): bool
     {
+        $this->isConnectedCount++;
+
         return $this->isConnected;
     }
 
+    public function getIsConnectedCount(): int
+    {
+        return $this->isConnectedCount;
+    }
+
     public function getDisconnectCount(): int
     {
         return $this->disconnectCount;