Spaces:
Running
Running
fix: disable asyncpg statement cache for social poolers
Browse files- app/social/database.py +10 -2
app/social/database.py
CHANGED
|
@@ -71,7 +71,12 @@ class SocialDatabase:
|
|
| 71 |
def __init__(self, settings: Settings) -> None:
|
| 72 |
self.settings = settings
|
| 73 |
self.database_url = normalize_async_database_url(settings.resolved_social_database_url)
|
| 74 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 75 |
if self.database_url.startswith("sqlite"):
|
| 76 |
event.listen(self.engine.sync_engine, "connect", self._configure_sqlite)
|
| 77 |
self.session_factory = async_sessionmaker(
|
|
@@ -81,7 +86,10 @@ class SocialDatabase:
|
|
| 81 |
self.worker_engine: AsyncEngine | None = None
|
| 82 |
self.worker_session_factory: async_sessionmaker[AsyncSession] | None = None
|
| 83 |
if self.worker_database_url:
|
| 84 |
-
|
|
|
|
|
|
|
|
|
|
| 85 |
if self.worker_database_url.startswith("sqlite"):
|
| 86 |
event.listen(self.worker_engine.sync_engine, "connect", self._configure_sqlite)
|
| 87 |
self.worker_session_factory = async_sessionmaker(
|
|
|
|
| 71 |
def __init__(self, settings: Settings) -> None:
|
| 72 |
self.settings = settings
|
| 73 |
self.database_url = normalize_async_database_url(settings.resolved_social_database_url)
|
| 74 |
+
# Supabase/PgBouncer transaction pooling requires asyncpg's prepared
|
| 75 |
+
# statement cache to be disabled to avoid stale statement-name reuse.
|
| 76 |
+
connect_args = {"statement_cache_size": 0} if self.database_url.startswith(("postgresql", "postgres")) else {}
|
| 77 |
+
self.engine: AsyncEngine = create_async_engine(
|
| 78 |
+
self.database_url, pool_pre_ping=True, connect_args=connect_args
|
| 79 |
+
)
|
| 80 |
if self.database_url.startswith("sqlite"):
|
| 81 |
event.listen(self.engine.sync_engine, "connect", self._configure_sqlite)
|
| 82 |
self.session_factory = async_sessionmaker(
|
|
|
|
| 86 |
self.worker_engine: AsyncEngine | None = None
|
| 87 |
self.worker_session_factory: async_sessionmaker[AsyncSession] | None = None
|
| 88 |
if self.worker_database_url:
|
| 89 |
+
worker_connect_args = {"statement_cache_size": 0} if self.worker_database_url.startswith(("postgresql", "postgres")) else {}
|
| 90 |
+
self.worker_engine = create_async_engine(
|
| 91 |
+
self.worker_database_url, pool_pre_ping=True, connect_args=worker_connect_args
|
| 92 |
+
)
|
| 93 |
if self.worker_database_url.startswith("sqlite"):
|
| 94 |
event.listen(self.worker_engine.sync_engine, "connect", self._configure_sqlite)
|
| 95 |
self.worker_session_factory = async_sessionmaker(
|