phanipavank/harmoney
1
0
Fork 0
Code Issues 5 Pull Requests Actions Packages Projects 1 Releases 1 Wiki Activity

probably fixed #5, need more testing

Browse Source
This commit is contained in:
Phani Pavan K
2026-02-22 12:09:53 +05:30
parent fc2e4179c4
commit 2d601d5ce1
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
harmoney/router/router.py
View File

@@ -2,7 +2,7 @@ import asyncio
import base64 import base64
import pickle as pkl import pickle as pkl
import uuid import uuid
from threading import Thread from threading import Lock, Thread
import fastapi import fastapi
from uvicorn import Config, Server 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. Method which queries an available task and sends the data to the attached runner.
""" """
l = Lock()
l.acquire()
await wsConnection.accept() await wsConnection.accept()
await wsConnection.send_text(str(self.runnerCount)) await wsConnection.send_text(str(self.runnerCount))
methods = await wsConnection.receive() methods = await wsConnection.receive()
@@ -41,6 +43,7 @@ class _Router:
runnerID = self.runnerCount runnerID = self.runnerCount
self.runnerCount += 1 self.runnerCount += 1
runnerCounter = 0 runnerCounter = 0
l.release()
try: try:
while True: while True:
# add this id back into available id pools using a shared list. # add this id back into available id pools using a shared list.