This commit is contained in:
		
							
								
								
									
										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