Mirrors/wireguard_webadmin
1
0
Fork 0
mirror of https://github.com/eduardogsilva/wireguard_webadmin.git synced 2026-01-02 06:36:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add ping functionality to worker for status updates

Browse Source
This commit is contained in:
Eduardo Silva
2025-12-31 20:34:16 -03:00
parent d8955ebd6d
commit 21e1e94d63
1 changed files with 20 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
containers/cluster_node/cluster_worker.py
View File

@@ -144,6 +144,19 @@ class ClusterWorker:
logger.error(f"Error updating DNS config: {e}") logger.error(f"Error updating DNS config: {e}")
return False return False
def send_ping(self):
params = {
'token': TOKEN,
'worker_config_version': self.config_version,
'worker_dns_version': self.dns_version,
'worker_version': WORKER_VERSION
}
try:
logger.info("Sending ping to Master...")
self.session.get(f"{self.base_url}/worker/ping/", params=params, timeout=REQUEST_TIMEOUT)
except Exception as e:
logger.error(f"Error sending ping: {e}")
def apply_configs(self, data): def apply_configs(self, data):
logger.info("Applying new configurations...") logger.info("Applying new configurations...")
files = data.get('files', {}) files = data.get('files', {})
@@ -209,25 +222,25 @@ class ClusterWorker:
remote_config_version = data.get('cluster_settings', {}).get('config_version', 0) remote_config_version = data.get('cluster_settings', {}).get('config_version', 0)
# Check WireGuard Config # Check WireGuard Config
if remote_config_version != self.config_version: if int(remote_config_version) != self.config_version:
logger.info(f"Config version mismatch (Local: {self.config_version}, Remote: {remote_config_version}). Updating...") logger.info(f"Config version mismatch (Local: {self.config_version}, Remote: {remote_config_version}). Updating...")
config_data = self.download_configs() config_data = self.download_configs()
if config_data: if config_data:
self.apply_configs(config_data) self.apply_configs(config_data)
logger.info("Sending post-update status to Master...") self.send_ping()
self.get_status() continue
else: else:
logger.error("Failed to download config files.") logger.error("Failed to download config files.")
# Check DNS Config # Check DNS Config
remote_dns_version = data.get('cluster_settings', {}).get('dns_version', 0) remote_dns_version = int(data.get('cluster_settings', {}).get('dns_version', 0))
if remote_dns_version != self.dns_version: if remote_dns_version != self.dns_version:
logger.info(f"DNS version mismatch (Local: {self.dns_version}, Remote: {remote_dns_version}). Updating...") logger.info(f"DNS version mismatch (Local: {self.dns_version}, Remote: {remote_dns_version}). Updating...")
if self.download_dns_config(): if self.download_dns_config():
logger.info("Sending post-update status to Master...") self.send_ping()
self.get_status() continue
if remote_config_version == self.config_version and remote_dns_version == self.dns_version: if int(remote_config_version) == self.config_version and remote_dns_version == self.dns_version:
logger.info(f"No changes detected. Configuration is up to date (WG: {self.config_version}, DNS: {self.dns_version}).") logger.info(f"No changes detected. Configuration is up to date (WG: {self.config_version}, DNS: {self.dns_version}).")
except Exception as e: except Exception as e: