This commit is contained in:
Kocsis Richárd 2020-03-30 15:42:48 +02:00
parent abc87057bf
commit df1875ca56

19
app.py
View File

@ -6,6 +6,7 @@ import requests.exceptions
import os
import redis
import json
import logging
"""
Scheduler
@ -20,15 +21,18 @@ sentry_sdk.init("https://0a106e104e114bc9a3fa47f9cb0db2f4@sentry.kmlabz.com/10")
def main():
logging.basicConfig(filename='', level=logging.DEBUG)
r = redis.Redis(host='localhost', port=6379, db=0)
ip_list = os.environ['INITIAL_SERVERS'].split(',')
logging.DEBUG('Get consumer list from environ at first: Done')
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:
response = requests.post(f"http://{ip}/sync", json={'uuid': os.environ['LOCAL_UUID']})
except requests.exceptions.ConnectionError:
@ -38,13 +42,12 @@ def main():
temp_dict[response.json()['uuid']] = {'ip': ip}
consumer_list_redis.update(temp_dict)
r.set('consumer_list', json.dumps(consumer_list_redis).encode('utf-8'))
logging.DEBUG('Update redis consumers ip list from first answers: Done')
while True:
logging.DEBUG('Infinite Cycle start : Done')
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:
@ -57,9 +60,13 @@ def main():
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')
time.sleep(30)
LOG.ERROR('While get rekt')
if __name__ == "__main__":
try: