feat!: update titiler version and fix lifespan (#61)

Author: vincentsarago

lib/stac-api/runtime/src/handler.py

@@ -2,8 +2,15 @@
 Handler for AWS Lambda.
 """
 
+import asyncio
+import os
 from mangum import Mangum
 
 from .app import app
 
-handler = Mangum(app)
+handler = Mangum(app, lifespan="off")
+
+
+if "AWS_EXECUTION_ENV" in os.environ:
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(app.router.startup())

lib/titiler-pgstac-api/runtime/requirements.txt

@@ -1,3 +1,3 @@
-titiler.pgstac==0.3.3
+titiler.pgstac==0.5.1
 boto3>=1.26.139
-psycopg[binary, pool]
+psycopg[binary, pool]

lib/titiler-pgstac-api/runtime/src/handler.py

@@ -2,10 +2,12 @@
 Handler for AWS Lambda.
 """
 
+import asyncio
 import os
 from mangum import Mangum
 from utils import get_secret_dict
 from titiler.pgstac.main import app
+from titiler.pgstac.db import connect_to_db
 
 pgstac_secret_arn = os.environ["PGSTAC_SECRET_ARN"]
 
@@ -20,4 +22,16 @@
     }
 )
 
-handler = Mangum(app)
+
+@app.on_event("startup")
+async def startup_event() -> None:
+    """Connect to database on startup."""
+    await connect_to_db(app)
+
+
+handler = Mangum(app, lifespan="off")
+
+
+if "AWS_EXECUTION_ENV" in os.environ:
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(app.router.startup())