This commit is contained in:
parent
b8fbbd9efb
commit
ff222eb543
12
app.py
12
app.py
@ -21,19 +21,24 @@ sentry_sdk.init("https://0a106e104e114bc9a3fa47f9cb0db2f4@sentry.kmlabz.com/10")
|
||||
|
||||
|
||||
def main():
|
||||
|
||||
# set logging preferences
|
||||
logging.basicConfig(filename = '', level = logging.DEBUG)
|
||||
|
||||
# connect to redis
|
||||
r = redis.Redis(host = 'localhost', port = 6379, db = 0)
|
||||
|
||||
# set initial consumer addresses
|
||||
ip_list = os.environ['INITIAL_SERVERS'].split(',')
|
||||
logging.debug('Get consumer list from environ at first: Done')
|
||||
# get the dictionary of the currently available consumers
|
||||
consumer_list_redis = json.loads((r.get('consumer_list') or b'{}').decode('utf-8'))
|
||||
logging.debug('Get consumer list from redis at first: Done')
|
||||
temp_dict = { }
|
||||
|
||||
for ip in ip_list:
|
||||
|
||||
try:
|
||||
# request synchronization
|
||||
response = requests.post(f"http://{ip}/sync", json = { 'uuid': os.environ['LOCAL_UUID'] })
|
||||
except requests.exceptions.ConnectionError:
|
||||
continue
|
||||
@ -46,11 +51,13 @@ def main():
|
||||
logging.debug('Update redis consumers ip list from first answers: Done')
|
||||
while True:
|
||||
logging.debug('Infinite Cycle start : Done')
|
||||
# get the dictionary of the currently available consumers
|
||||
consumer_list_redis = json.loads((r.get('consumer_list') or b'{}').decode('utf-8'))
|
||||
logging.debug('Get consumer list from redis: Done')
|
||||
for uuid, info in consumer_list_redis.items():
|
||||
ip = info['ip']
|
||||
try:
|
||||
# request synchronization
|
||||
response = requests.post(f"http://{ip}/sync", json = { 'uuid': os.environ['LOCAL_UUID'] })
|
||||
except requests.exceptions.ConnectionError:
|
||||
continue
|
||||
@ -58,11 +65,14 @@ def main():
|
||||
if response.status_code == 200:
|
||||
temp_dict[response.json()['uuid']] = { 'ip': ip }
|
||||
|
||||
|
||||
# update the dictionary of the currently available consumers
|
||||
consumer_list_redis.update(temp_dict)
|
||||
r.set('consumer_list', json.dumps(consumer_list_redis).encode('utf-8'))
|
||||
logging.debug('Update redis consumer ip list from answers: Done')
|
||||
|
||||
logging.debug('Waiting for next turn')
|
||||
# wait for the next update time
|
||||
time.sleep(30)
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user