server.py

import logging
import sys
import time

from api.utils import func_invoker

# config
FUNCTIONS = {
    "add": lambda x, y: x + y,
    "sleep": lambda x: time.sleep(x) or x,
}

# FastAPI
if len(sys.argv) > 0 and sys.argv[0].split("/")[-1] == "uvicorn":
    import fastapi

    SERVER = func_invoker.FastapiServer(functions=FUNCTIONS, loginfo=logging.getLogger("uvicorn").info)

    # uvicorn server:app
    app = fastapi.FastAPI(
        title="Functions API",
        description="author: [TongZJ](https://github.com/Instinct323)\n\n" + SERVER.func_docs(),
        version="1.0.0"
    )


    @app.post("/invoke")
    async def invoke(request):
        return await SERVER.get_response(request)

# Zmq
else:
    import zmq
    from api.utils.zmq_socket import ZmqSocket

    SERVER = func_invoker.ZmqSocketServer(
        functions=FUNCTIONS, loginfo=print,
        sock=ZmqSocket(zmq.REP, addr="/tmp/myzmq", queue_size=10)
    )
    SERVER.main()