GoldenPogacsa/consumer-scheduler
Archived
7
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

documentation
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Füleki Fábián
2020-04-04 13:58:08 +02:00
parent b8fbbd9efb
commit ff222eb543
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
app.py
View File

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