danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Pre-ping the Postgres queue engine

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-06-04 11:29:31 +03:00
parent e37d764ab2
commit a04dc9445b
No known key found for this signature in database
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
haiku_rag_slim/haiku/rag/ingester/queue/migrations.py
View file

@ -17,7 +17,9 @@ def make_engine(config: QueueConfig) -> AsyncEngine:
"""Build the queue's AsyncEngine from config. Uses `dburi` when set, """Build the queue's AsyncEngine from config. Uses `dburi` when set,
otherwise a `sqlite+aiosqlite` URL pointing at the resolved `path` otherwise a `sqlite+aiosqlite` URL pointing at the resolved `path`
(creating the parent directory). SQLite is capped to a single pooled (creating the parent directory). SQLite is capped to a single pooled
connection so the two-statement claim stays atomic without row locks.""" connection so the claim stays atomic without row locks; Postgres uses
pool_pre_ping so a long-running ingester survives a DB restart or idle
connection drop."""
if config.dburi: if config.dburi:
url = make_url(config.dburi) url = make_url(config.dburi)
else: else:
@ -28,7 +30,7 @@ def make_engine(config: QueueConfig) -> AsyncEngine:
if url.get_backend_name() == "sqlite": if url.get_backend_name() == "sqlite":
engine = create_async_engine(url, pool_size=1, max_overflow=0) engine = create_async_engine(url, pool_size=1, max_overflow=0)
else: else:
engine = create_async_engine(url) engine = create_async_engine(url, pool_pre_ping=True)
install_sqlite_pragmas(engine) install_sqlite_pragmas(engine)
return engine return engine