🐛 I forgot some

ThatXliner · Dec 17, 2023 · 07394d2 · 07394d2
1 parent 34dbf05
commit 07394d2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/aioudp/server.py b/aioudp/server.py
@@ -37,7 +37,7 @@ def connection_made(
         self.transport = transport
 
     def datagram_received(self, data: bytes, addr: connection.AddrType) -> None:
-        if addr not in self.msg_handler:
+        if addr not in self.msg_queues:
             self.msg_queues[addr] = asyncio.Queue()
             assert self.transport is not None
 
@@ -71,8 +71,8 @@ def connection_lost(self, exc: None | Exception) -> None:
         # Haven't figured out why this can happen
         if exc is not None:
             raise exc
-        for key in self.msg_handler:
-            self.msg_handler[key].put_nowait(None)
+        for key in self.msg_queues:
+            self.msg_queues[key].put_nowait(None)
 
 
 @asynccontextmanager