sij/sijapi
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Auto-update: Tue Aug 6 21:48:47 PDT 2024

Browse source
This commit is contained in:
sanj 2024-08-06 21:48:47 -07:00
parent d268805253
commit 72172dbc98
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
sijapi/classes.py
View file

@ -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: