run server until websocket connection is not closed

xconnio · May 4, 2024 · 92759ba · 92759ba
1 parent b9bdd55
commit 92759ba
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/src/server.dart b/lib/src/server.dart
@@ -19,13 +19,13 @@ class Server {
       BaseSession baseSession = await a.accept(webSocket);
       router.attachClient(baseSession);
 
-      _handleWebSocket(baseSession);
+      _handleWebSocket(baseSession, webSocket);
     }
   }
 
-  void _handleWebSocket(BaseSession baseSession) {
+  void _handleWebSocket(BaseSession baseSession, WebSocket webSocket) {
     Future.microtask(() async {
-      while (true) {
+      while (webSocket.closeCode == null) {
         var message = await baseSession.receiveMessage();
         await router.receiveMessage(baseSession, message);
       }