Don't do the lifespan during request time

buremba · Oct 12, 2024 · 85b8578 · 85b8578
1 parent c70b17f
commit 85b8578
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2,4 +2,4 @@

		from universql.protocol.snowflake import app as snowflake_app

		snowflake = Mangum(snowflake_app)
		snowflake = Mangum(snowflake_app, lifespan="off")