Close on disconnect

PlaidCloud · Oct 31, 2023 · ccc7ba0 · ccc7ba0
1 parent c626d5b
commit ccc7ba0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/redis_stomp/pubsub/topic.py b/redis_stomp/pubsub/topic.py
@@ -34,6 +34,10 @@ def __init__(self, connection: StompConnection, redis_url: str):
         self._closed = False
         self._redis = connect(redis_url, decode_responses=True).pubsub(ignore_subscribe_messages=True)
 
+    async def disconnect(self, connection: StompConnection):
+        await super().disconnect(connection)
+        await self.close()
+
     async def close(self):
         self._closing = True
         while not self._closed: