update connections after handling messages

laravel · Dec 29, 2023 · bd77a9f · bd77a9f
1 parent add4d46
commit bd77a9f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/Servers/ApiGateway/Server.php b/src/Servers/ApiGateway/Server.php
@@ -30,14 +30,14 @@ public function handle(Request $request): void
         try {
             match ($request->event()) {
                 'CONNECT' => $this->server->open(
-                    $this->connect($request)
+                    $connection = $this->connect($request)
                 ),
                 'DISCONNECT' => $this->server->close(
                     $this->connect($request),
                 ),
                 'MESSAGE' => $this->server->message(
                     $this->connect($request),
-                    $request->message()
+                    $connection = $request->message()
                 ),
             };
         } catch (InvalidApplication $e) {
@@ -51,6 +51,10 @@ public function handle(Request $request): void
                 $e
             );
         }
+
+        if (isset($connection)) {
+            app()->terminating(fn () => app(ConnectionManager::class)->update($connection));
+        }
     }
 
     /**