From d845692816079cf78ded919bc1175647576d8529 Mon Sep 17 00:00:00 2001 From: Keuin Date: Sat, 3 Jun 2023 15:57:56 +0800 Subject: Use non-blocking file/subprocess IO --- web.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) (limited to 'web.py') diff --git a/web.py b/web.py index e2d7fdf..82126b7 100644 --- a/web.py +++ b/web.py @@ -1,7 +1,11 @@ +import asyncio +import os import re +import uvicorn from fastapi import FastAPI, Response from fastapi.responses import FileResponse +from uvicorn.loops.auto import auto_loop_setup import htmlcache import texgen @@ -32,3 +36,28 @@ async def get_utaten_lyric_tex(item_id: str): return Response(content=tex, media_type='application/x-tex') except htmlcache.TexSourceGenerationError as e: return Response(content=str(e), status_code=503) + + +@app.on_event("startup") +async def startup_event(): + pass + + +def setup_loop(): + if os.name == 'nt': + # use ProactorEventLoop to support async subprocess on Windows + print('Driving event loop with IOCP.') + loop = asyncio.ProactorEventLoop() + asyncio.set_event_loop(loop) + auto_loop_setup() + + +if __name__ == '__main__': + setup_loop() + uvicorn.run( + 'web:app', + host='127.0.0.1', + port=8000, + log_level='info', + loop='none', # use custom loop initializer + ) -- cgit v1.2.3