Merge pull request #155 from smohiudd/fix/db-connect-kwargs

Change db connection kwargs default

Author: vincentsarago

titiler/pgstac/db.py

@@ -1,6 +1,6 @@
 """Database connection handling."""
 
-from typing import Optional
+from typing import Any, Dict, Optional
 
 from fastapi import FastAPI
 from psycopg_pool import ConnectionPool
@@ -9,20 +9,28 @@
 
 
 async def connect_to_db(
-    app: FastAPI, settings: Optional[PostgresSettings] = None
+    app: FastAPI,
+    settings: Optional[PostgresSettings] = None,
+    pool_kwargs: Optional[Dict[str, Any]] = None,
 ) -> None:
     """Connect to Database."""
     if not settings:
         settings = PostgresSettings()
 
+    pool_kwargs = (
+        pool_kwargs
+        if pool_kwargs is not None
+        else {"options": "-c search_path=pgstac,public -c application_name=pgstac"}
+    )
+
     app.state.dbpool = ConnectionPool(
         conninfo=str(settings.database_url),
         min_size=settings.db_min_conn_size,
         max_size=settings.db_max_conn_size,
         max_waiting=settings.db_max_queries,
         max_idle=settings.db_max_idle,
         num_workers=settings.db_num_workers,
-        kwargs={"options": "-c search_path=pgstac,public -c application_name=pgstac"},
+        kwargs=pool_kwargs,
     )
 
     # Make sure the pool is ready