Sadeep Sachintha commited on
Commit ·
0116ebe
1
Parent(s): d24d9cd
fix: resolve create_engine type error and use dynamic prepared statement names
Browse files- db/session.py +3 -3
db/session.py
CHANGED
|
@@ -4,6 +4,7 @@ from core.config import settings
|
|
| 4 |
from db.models import Base
|
| 5 |
|
| 6 |
import ssl
|
|
|
|
| 7 |
|
| 8 |
logger = logging.getLogger(__name__)
|
| 9 |
|
|
@@ -14,7 +15,7 @@ if "supabase" in settings.database_url or "postgres" in settings.database_url or
|
|
| 14 |
ssl_context.verify_mode = ssl.CERT_NONE
|
| 15 |
connect_args["ssl"] = ssl_context
|
| 16 |
connect_args["statement_cache_size"] = 0
|
| 17 |
-
connect_args["
|
| 18 |
|
| 19 |
# Initialize the async engine
|
| 20 |
engine = create_async_engine(
|
|
@@ -22,8 +23,7 @@ engine = create_async_engine(
|
|
| 22 |
echo=(settings.log_level == "DEBUG"),
|
| 23 |
future=True,
|
| 24 |
pool_pre_ping=True,
|
| 25 |
-
connect_args=connect_args
|
| 26 |
-
prepared_statement_cache_size=0
|
| 27 |
)
|
| 28 |
|
| 29 |
# Create an async session factory
|
|
|
|
| 4 |
from db.models import Base
|
| 5 |
|
| 6 |
import ssl
|
| 7 |
+
from uuid import uuid4
|
| 8 |
|
| 9 |
logger = logging.getLogger(__name__)
|
| 10 |
|
|
|
|
| 15 |
ssl_context.verify_mode = ssl.CERT_NONE
|
| 16 |
connect_args["ssl"] = ssl_context
|
| 17 |
connect_args["statement_cache_size"] = 0
|
| 18 |
+
connect_args["prepared_statement_name_func"] = lambda: f"__asyncpg_{uuid4().hex}__"
|
| 19 |
|
| 20 |
# Initialize the async engine
|
| 21 |
engine = create_async_engine(
|
|
|
|
| 23 |
echo=(settings.log_level == "DEBUG"),
|
| 24 |
future=True,
|
| 25 |
pool_pre_ping=True,
|
| 26 |
+
connect_args=connect_args
|
|
|
|
| 27 |
)
|
| 28 |
|
| 29 |
# Create an async session factory
|