lsp-devtools: Apply ruff fixes

swyddfa · Nov 23, 2024 · d4bafff · d4bafff
1 parent 9228616
commit d4bafff
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/lib/lsp-devtools/lsp_devtools/agent/client.py b/lib/lsp-devtools/lsp_devtools/agent/client.py
@@ -25,6 +25,7 @@ def __init__(self):
         )
         self.connected = False
         self._buffer: list[bytes] = []
+        self._tasks: set[asyncio.Task[Any]] = set()
 
     def _report_server_error(self, error, source):
         # Bail on error
@@ -61,8 +62,12 @@ def forward_message(self, message: bytes):
         while len(self._buffer) > 0:
             res = self.protocol.writer.write(self._buffer.pop(0))
             if inspect.isawaitable(res):
-                asyncio.ensure_future(res)
+                task = asyncio.ensure_future(res)
+                task.add_done_callback(self._tasks.discard)
+                self._tasks.add(task)
 
         res = self.protocol.writer.write(message)
         if inspect.isawaitable(res):
-            asyncio.ensure_future(res)
+            task = asyncio.ensure_future(res)
+            task.add_done_callback(self._tasks.discard)
+            self._tasks.add(task)