Fix mangum db connection pool

NASA-IMPACT · Jan 13, 2023 · 0075d5d · 0075d5d
1 parent a45fbb4
commit 0075d5d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/raster_api/runtime/handler.py b/raster_api/runtime/handler.py
@@ -10,7 +10,11 @@
 logging.getLogger("mangum.http").setLevel(logging.ERROR)
 
 
-handler = Mangum(app, lifespan="auto")
+handler = Mangum(app, lifespan="off")
+
+if "AWS_EXECUTION_ENV" in os.environ:
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(app.router.startup())
 
 # Add tracing
 handler.__name__ = "handler"  # tracer requires __name__ to be set