From 2d601d5ce1e00a39c21ad384bd265a0385da7896 Mon Sep 17 00:00:00 2001
From: Phani Pavan K <kphanipavan@gmail.com>
Date: Sun, 22 Feb 2026 12:09:53 +0530
Subject: [PATCH] probably fixed #5, need more testing

---
 harmoney/router/router.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/harmoney/router/router.py b/harmoney/router/router.py
index 947c2e4..1a847f8 100644
--- a/harmoney/router/router.py
+++ b/harmoney/router/router.py
@@ -2,7 +2,7 @@ import asyncio
 import base64
 import pickle as pkl
 import uuid
-from threading import Thread
+from threading import Lock, Thread
 
 import fastapi
 from uvicorn import Config, Server
@@ -31,6 +31,8 @@ class _Router:
         """
         Method which queries an available task and sends the data to the attached runner.
         """
+        l = Lock()
+        l.acquire()
         await wsConnection.accept()
         await wsConnection.send_text(str(self.runnerCount))
         methods = await wsConnection.receive()
@@ -41,6 +43,7 @@ class _Router:
         runnerID = self.runnerCount
         self.runnerCount += 1
         runnerCounter = 0
+        l.release()
         try:
             while True:
                 # add this id back into available id pools using a shared list.