From 0bc936dc479769a9c198fecac9299a3243537b67 Mon Sep 17 00:00:00 2001 From: sanj <67624670+iodrift@users.noreply.github.com> Date: Mon, 12 Aug 2024 17:00:10 -0700 Subject: [PATCH] Auto-update: Mon Aug 12 17:00:10 PDT 2024 --- sijapi/database.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/sijapi/database.py b/sijapi/database.py index 5783260..227af6a 100644 --- a/sijapi/database.py +++ b/sijapi/database.py @@ -408,13 +408,18 @@ class Database: for server in self.config['POOL']: if server['ts_id'] in online_servers and server['ts_id'] != self.local_ts_id: url = f"http://{server['ts_ip']}:{server['app_port']}/db/sync" - tasks.append(self.call_db_sync(url)) + tasks.append(self.call_db_sync(server)) await asyncio.gather(*tasks) - async def call_db_sync(self, url): + + async def call_db_sync(self, server): + url = f"http://{server['ts_ip']}:{server['app_port']}/db/sync" + headers = { + "Authorization": f"Bearer {server['api_key']}" + } async with aiohttp.ClientSession() as session: try: - async with session.post(url, timeout=30) as response: + async with session.post(url, headers=headers, timeout=30) as response: if response.status == 200: l.info(f"Successfully called /db/sync on {url}") else: @@ -424,6 +429,7 @@ class Database: except Exception as e: l.error(f"Error calling /db/sync on {url}: {str(e)}") + async def close(self): for engine in self.engines.values(): await engine.dispose()