syncthing-monitor/syncthing_monitor/etcd_client.py

45 lines
1.5 KiB
Python
Raw Normal View History

2021-01-29 08:28:35 +00:00
import json
import etcd3 as etcd3
CLUSTER_INFO_KEY = '/syncthing_monitor/cluster_info'
class EtcdClient:
def __init__(self, host, port, key=CLUSTER_INFO_KEY):
2021-01-29 10:56:51 +00:00
self.etcd = etcd3.client(host=host, port=port)
self.key = key
2021-01-29 08:28:35 +00:00
2021-01-29 10:56:51 +00:00
def load_cluster_info(self):
2021-02-02 08:48:19 +00:00
print("Retrieving cluster info from etcd...", flush=True)
raw_value = self.etcd.get(self.key)[0]
2021-01-29 10:56:51 +00:00
if raw_value is None:
return {'devices': []}
return json.loads(raw_value)
2021-01-29 08:28:35 +00:00
def add_device_to_cluster(self, device_id, node_name, address):
2021-01-29 10:56:51 +00:00
with self.etcd.lock('syncthing_monitor'):
cluster_info = self.load_cluster_info()
2021-01-29 08:28:35 +00:00
if any(device_id in device['id'] for device in cluster_info['devices']):
return
new_device = {
'id': device_id,
'node_name': node_name,
'address': address
}
cluster_info['devices'].append(new_device)
2021-01-29 09:59:59 +00:00
2021-02-01 14:50:03 +00:00
print("Updating etcd value for '{0}': {1}".format(self.key, json.dumps(cluster_info)), flush=True)
self.etcd.put(self.key, json.dumps(cluster_info))
2021-01-29 09:59:59 +00:00
2021-01-29 10:56:51 +00:00
def get_device_list(self):
device_list = self.load_cluster_info()['devices']
2021-02-01 14:50:03 +00:00
print("Obtained device_list devices from etcd: {0}".format(json.dumps(device_list)), flush=True)
2021-01-29 10:56:51 +00:00
return device_list
def register_device_update_handler(self, device_update_handler):
self.etcd.add_watch_callback(self.key, device_update_handler)