basyx commited on
Commit
77fb0a2
·
verified ·
1 Parent(s): a3a768d

fix: disable asyncpg statement cache for social poolers

Browse files
Files changed (1) hide show
  1. 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
- self.engine: AsyncEngine = create_async_engine(self.database_url, pool_pre_ping=True)
 
 
 
 
 
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
- self.worker_engine = create_async_engine(self.worker_database_url, pool_pre_ping=True)
 
 
 
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(