client state disconnected일 때 send 금지

gamultong · Dec 30, 2024 · b5bd526 · b5bd526
1 parent caf8493
commit b5bd526
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/conn/internal/conn.py b/conn/internal/conn.py
@@ -1,4 +1,4 @@
-from fastapi.websockets import WebSocket
+from fastapi.websockets import WebSocket, WebSocketState
 from websockets.exceptions import ConnectionClosed
 from message import Message
 from dataclasses import dataclass
@@ -23,6 +23,9 @@ async def receive(self):
         return Message.from_str(await self.conn.receive_text())
 
     async def send(self, msg: Message):
+        if self.conn.client_state == WebSocketState.DISCONNECTED:
+            return
+
         try:
             await self.conn.send_text(msg.to_str())
         except ConnectionClosed: