修复内存泄露

hyperf · Dec 5, 2024 · 631ea14 · 631ea14
1 parent 79ed1d4
commit 631ea14
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/roc/channel_manager.py b/roc/channel_manager.py
@@ -23,6 +23,7 @@ def close(self, key: int) -> bool:
         if chan is None:
             return True
 
+        del self.channels[key]
         return chan.close()
 
     def flush(self):

diff --git a/roc/socket.py b/roc/socket.py
@@ -98,13 +98,16 @@ async def request(self, request: Request) -> Response:
 
         await self.send(packet)
 
-        res = await chan.pop()
-        if res is False:
-            raise RequestException("request failed")
+        try:
+            res = await chan.pop()
+            if res is False:
+                raise RequestException("request failed")
 
-        data = json.loads(res)
+            data = json.loads(res)
 
-        return make_response(data)
+            return make_response(data)
+        finally:
+            self.channelManager.close(key)
 
     async def start(self):
         reader, writer = await asyncio.open_connection(self.host, self.port)