From 85b8578c75130c43d3469005d7e3f7e28da38175 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Burak=20Kabakc=C4=B1?= Date: Sat, 12 Oct 2024 01:25:41 +0100 Subject: [PATCH] Don't do the lifespan during request time --- universql/protocol/lambda.py | 2 +- universql/protocol/snowflake.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/universql/protocol/lambda.py b/universql/protocol/lambda.py index b80c2b2..74d553e 100644 --- a/universql/protocol/lambda.py +++ b/universql/protocol/lambda.py @@ -2,4 +2,4 @@ from universql.protocol.snowflake import app as snowflake_app -snowflake = Mangum(snowflake_app) +snowflake = Mangum(snowflake_app, lifespan="off") diff --git a/universql/protocol/snowflake.py b/universql/protocol/snowflake.py index dba38bd..c771b35 100644 --- a/universql/protocol/snowflake.py +++ b/universql/protocol/snowflake.py @@ -40,6 +40,8 @@ @snowflake_server_opts def test(**kwargs): value[0] = kwargs + + with click.Context(snowflake) as ctx: ctx.params = os.environ ctx.invoke(test) @@ -57,7 +59,7 @@ async def add_process_time_header(request: Request, call_next): start_time = time.perf_counter() response = await call_next(request) # if request.url.path in ["/queries/v1/query-request"]: - print(f"Time took to process {request.url.path} is {time.perf_counter() - start_time} sec") + # print(f"Time took to process {request.url.path} is {time.perf_counter() - start_time} sec") return response