50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
import asyncio
|
|
import base64
|
|
from uvicorn import Config, Server
|
|
import fastapi
|
|
|
|
from uvicorn.config import LOG_LEVELS
|
|
import pickle as pkl
|
|
|
|
from callSpec import CallPacket, ClientPacket
|
|
|
|
|
|
class Broker:
|
|
def __init__(self) -> None:
|
|
self.runnerQueue = []
|
|
self.router = fastapi.APIRouter()
|
|
self.router.add_api_websocket_route("/reg", self.registerRunner)
|
|
self.router.add_api_route("/cliReq", self.clientRequest, methods=["POST"])
|
|
self.taskQueue = asyncio.Queue()
|
|
|
|
|
|
async def registerRunner(self, wsConnection: fastapi.WebSocket):
|
|
|
|
await wsConnection.accept()
|
|
|
|
print("started")
|
|
while True:
|
|
data:CallPacket = await self.taskQueue.get()
|
|
# await asyncio.sleep(1)
|
|
await wsConnection.send_bytes(pkl.dumps(data))
|
|
|
|
print(await wsConnection.receive())
|
|
print(f"left: {self.taskQueue.qsize()}")
|
|
|
|
async def clientRequest(self, data:ClientPacket):
|
|
print(data)
|
|
callPacket = pkl.loads(base64.b64decode(data.data))
|
|
await self.taskQueue.put(callPacket)
|
|
print(self.taskQueue.qsize)
|
|
# await self.taskQueue.put(name)
|
|
|
|
def runBroker():
|
|
br = Broker()
|
|
app = fastapi.FastAPI()
|
|
app.include_router(br.router)
|
|
serverConf = Config(app = app, host="0.0.0.0", port=7732, log_level=LOG_LEVELS["debug"], ws_ping_interval=10, ws_ping_timeout=None)
|
|
server = Server(config=serverConf)
|
|
server.run()
|
|
|
|
runBroker()
|