#96 query clusters in 5 second interval

This commit is contained in:
Henning Jacobs
2017-01-15 17:12:28 +01:00
parent a60f1b58b3
commit 1b3819a290

View File

@@ -27,23 +27,7 @@ def get_short_error_message(e: requests.exceptions.RequestException):
return str(e)
def update_clusters(cluster_discoverer, query_cluster: callable, store, debug: bool):
while True:
lock = store.acquire_lock()
if lock:
try:
clusters = cluster_discoverer.get_clusters()
cluster_ids = set()
for cluster in clusters:
cluster_ids.add(cluster.id)
backoff_key = '{}:backoff'.format(cluster.id)
backoff = store.get(backoff_key)
if backoff and time.time() < backoff['next_try']:
# cluster is still in backoff, skip
continue
try:
data = query_cluster(cluster)
except Exception as e:
def handle_query_failure(e: Exception, cluster, backoff: dict):
if not backoff:
backoff = {}
tries = backoff.get('tries', 0) + 1
@@ -58,11 +42,38 @@ def update_clusters(cluster_discoverer, query_cluster: callable, store, debug: b
log = logger.exception
log('Failed to query cluster {} ({}): {} (try {}, wait {} seconds)'.format(
cluster.id, cluster.api_server_url, message, tries, round(wait_seconds)))
store.set(backoff_key, backoff)
return backoff
def update_clusters(cluster_discoverer, query_cluster: callable, store, query_interval=5, debug: bool=False):
while True:
lock = store.acquire_lock()
if lock:
try:
clusters = cluster_discoverer.get_clusters()
cluster_ids = set()
for cluster in clusters:
cluster_ids.add(cluster.id)
status_key = '{}:status'.format(cluster.id)
status = store.get(status_key) or {}
now = time.time()
if now < status.get('last_query_time', 0) + query_interval:
continue
backoff = status.get('backoff')
if backoff and now < backoff['next_try']:
# cluster is still in backoff, skip
continue
try:
logger.debug('Querying cluster {} ({})..'.format(cluster.id, cluster.api_server_url))
data = query_cluster(cluster)
except Exception as e:
backoff = handle_query_failure(e, cluster, backoff)
status['backoff'] = backoff
else:
status['last_query_time'] = now
if backoff:
# reset backoff
store.set(backoff_key, None)
logger.info('Cluster {} ({}) recovered after {} tries.'.format(cluster.id, cluster.api_server_url, backoff['tries']))
del status['backoff']
old_data = store.get(data['id'])
if old_data:
# https://pikacode.com/phijaro/json_delta/ticket/11/
@@ -72,11 +83,13 @@ def update_clusters(cluster_discoverer, query_cluster: callable, store, debug: b
if delta:
store.set(cluster.id, data)
else:
logger.info('Discovered new cluster {} ({}).'.format(cluster.id, cluster.api_server_url))
store.publish('clusterupdate', data)
store.set(cluster.id, data)
store.set(status_key, status)
store.set('cluster-ids', list(sorted(cluster_ids)))
except:
logger.exception('Failed to update')
finally:
store.release_lock(lock)
gevent.sleep(5)
gevent.sleep(random_jitter(1))