Auto-update: Mon Aug 12 18:02:22 PDT 2024
This commit is contained in:
parent
d3930ca85f
commit
9863965976
1 changed files with 9 additions and 0 deletions
|
@ -273,6 +273,15 @@ class Database:
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
l.error(f"Error calling /db/sync on {url}: {str(e)}")
|
l.error(f"Error calling /db/sync on {url}: {str(e)}")
|
||||||
|
|
||||||
|
async def ensure_query_tracking_table(self):
|
||||||
|
for ts_id, engine in self.engines.items():
|
||||||
|
try:
|
||||||
|
async with engine.begin() as conn:
|
||||||
|
await conn.run_sync(Base.metadata.create_all)
|
||||||
|
l.info(f"Ensured query_tracking table exists for {ts_id}")
|
||||||
|
except Exception as e:
|
||||||
|
l.error(f"Failed to create query_tracking table for {ts_id}: {str(e)}")
|
||||||
|
|
||||||
async def close(self):
|
async def close(self):
|
||||||
for engine in self.engines.values():
|
for engine in self.engines.values():
|
||||||
await engine.dispose()
|
await engine.dispose()
|
||||||
|
|
Loading…
Reference in a new issue