apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-16 16:51:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Go back to response thread

Browse source
This commit is contained in:
Cyber MacGeddon 2024-12-06 22:52:53 +00:00
parent f61635384e
commit b15f956ac7
1 changed files with 6 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

34
trustgraph-flow/trustgraph/gateway/command.py
View file

@ -27,16 +27,16 @@ class CommandEndpoint(SocketEndpoint):
async def start(self):
pass
async def ASDasync_thread(self, ws, running):
async def async_thread(self, ws, running):
while running.get():
try:
svc, request = await asyncio.wait_for(self.q.get(), 1)
id, svc, request = await asyncio.wait_for(self.q.get(), 1)
except TimeoutError:
continue
except Exception as e:
await ws.send_json({"error": str(e)})
await ws.send_json({"id": id, "error": str(e)})
try:
@ -46,7 +46,7 @@ class CommandEndpoint(SocketEndpoint):
resp = await requestor.process(request)
await ws.send_json({ "response": resp })
await ws.send_json({"id": id, "response": resp })
except Exception as e:
@ -54,26 +54,6 @@ class CommandEndpoint(SocketEndpoint):
running.stop()
async def processor(self, ws, id, service, request):
try:
print(id, service, request)
requestor = self.services[service]
resp = await requestor.process(request)
await ws.send_json({"id": id, "response": resp })
except Exception as e:
await ws.send_json({"id": id, "error": str(e)})
async def async_thread(self, ws, running):
while running.get():
await asyncio.sleep(1)
async def listener(self, ws, running):
async for msg in ws:
@ -96,10 +76,8 @@ class CommandEndpoint(SocketEndpoint):
if "id" not in data:
raise RuntimeError("Bad message")
asyncio.create_task(
self.processor(
ws, data["id"], data["service"], data["request"]
)
await self.q.put(
(data["id"], data["service"], data["request"])
)
except Exception as e: