Add lock to internal socket fetch function

2026-07-07 18:29:55 +00:00 · 2024-07-07 20:33:45 +02:00
1 changed files with 6 additions and 4 deletions
@@ -19,6 +19,7 @@ class UnixSocket:
        self.reader = None
        self.writer = None
        self.server_writer = None
+        self.open_lock = asyncio.Lock()

    async def setup_server(self):
        try:
@@ -42,10 +43,11 @@ class UnixSocket:
            return True

    async def get_socket_connection(self):
-        if not await self._open_socket_if_not_exists():
-            return None, None
-        
-        return self.reader, self.writer
+        async with self.open_lock:
+            if not await self._open_socket_if_not_exists():
+                return None, None
+            
+            return self.reader, self.writer
    
    async def close_socket_connection(self):
        if self.writer != None: