Ananto30/zero

from aiohttp import web

from zero import AsyncZeroClient, ZeroClient, ZeroPublisher

# TODO: why we can't use uvloop?
# try:
#     import uvloop

#     uvloop.install()
# except ImportError:
#     logging.warning("Cannot use uvloop")
#     pass

client = ZeroClient("server", 5559)
async_client = AsyncZeroClient("server", 5559)
zero_publisher = ZeroPublisher("server", 5558)


async def hello(request):
    resp = client.call("hello_world", None)
    return web.Response(text=resp)


async def async_hello(request):
    resp = await async_client.call("hello_world", None)
    return web.Response(text=resp)


async def order(request):
    resp = client.call("save_order", {"user_id": "1", "items": ["apple", "python"]})
    return web.json_response(resp)


async def async_order(request):
    resp = await async_client.call(
        "save_order", {"user_id": "1", "items": ["apple", "python"]}
    )
    return web.json_response(resp)


async def pubs(request):
    await zero_publisher.publish_async("hello_world", {"1": {"2": {"3": {"4": "5"}}}})
    return web.Response()


async def enc_dec_jwt(request):
    resp = await async_client.call("decode_jwt", {"user_id": "a1b2c3"})
    return web.json_response(resp)


async def echo(request):
    big_list = ["hello world" for i in range(100_000)]
    resp = await async_client.call("echo", big_list)
    return web.Response(text=str(resp))


app = web.Application()
app.router.add_get("/order", order)
app.router.add_get("/async_order", async_order)
app.router.add_get("/hello", hello)
app.router.add_get("/async_hello", async_hello)
app.router.add_get("/pub", pubs)
app.router.add_get("/echo", echo)
app.router.add_get("/jwt", enc_dec_jwt)