Auto-update: Tue Aug 6 21:48:47 PDT 2024
This commit is contained in:
parent
d268805253
commit
72172dbc98
1 changed files with 8 additions and 2 deletions
|
@ -844,19 +844,25 @@ class APIConfig(BaseModel):
|
||||||
primary_key = await self.ensure_sync_columns(conn, table_name)
|
primary_key = await self.ensure_sync_columns(conn, table_name)
|
||||||
|
|
||||||
if query.strip().upper().startswith("INSERT") and "RETURNING" in query.upper():
|
if query.strip().upper().startswith("INSERT") and "RETURNING" in query.upper():
|
||||||
# For INSERT queries with RETURNING clause, use fetch to get the returned values
|
|
||||||
result = await conn.fetch(query, *args)
|
result = await conn.fetch(query, *args)
|
||||||
else:
|
else:
|
||||||
# For other queries, use execute
|
|
||||||
result = await conn.execute(query, *args)
|
result = await conn.execute(query, *args)
|
||||||
|
|
||||||
|
# Schedule the sync task
|
||||||
asyncio.create_task(self._sync_changes(table_name, primary_key))
|
asyncio.create_task(self._sync_changes(table_name, primary_key))
|
||||||
|
|
||||||
return result
|
return result
|
||||||
|
except Exception as e:
|
||||||
|
err(f"Error executing write query on table {table_name}: {str(e)}")
|
||||||
|
err(f"Query: {query}")
|
||||||
|
err(f"Args: {args}")
|
||||||
|
err(f"Traceback: {traceback.format_exc()}")
|
||||||
|
raise
|
||||||
finally:
|
finally:
|
||||||
await conn.close()
|
await conn.close()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
async def _run_sync_tasks(self, tasks):
|
async def _run_sync_tasks(self, tasks):
|
||||||
for task in tasks:
|
for task in tasks:
|
||||||
try:
|
try:
|
||||||
|
|
Loading…
Add table
Reference in a new issue