diff --git a/consumer_api/app.py b/consumer_api/app.py index dbedea3..3f5282f 100644 --- a/consumer_api/app.py +++ b/consumer_api/app.py @@ -19,6 +19,8 @@ sentry_sdk.init("https://0a106e104e114bc9a3fa47f9cb0db2f4@sentry.kmlabz.com/10") app = Flask(__name__) app.config['REDIS_URL'] = os.environ['REDIS_URL'] app.config['LOCAL_UUID'] = os.environ['LOCAL_UUID'] +app.config['CUSTOMER_TIMEOUT'] = int(os.environ.get('CUSTOMER_TIMEOUT', 30)) +app.config['PRODUCER_TIMEOUT'] = int(os.environ.get('PRODUCER_TIMEOUT', 60)) redis_client.init_app(app) diff --git a/consumer_api/views/consumers_view.py b/consumer_api/views/consumers_view.py index 0582bc4..097c42d 100644 --- a/consumer_api/views/consumers_view.py +++ b/consumer_api/views/consumers_view.py @@ -8,6 +8,13 @@ from flask_classful import FlaskView class ConsumersView(FlaskView): def get(self): - consumer_list = json.loads((redis_client.get("consumer_list") or b"{}").decode('utf-8')) - current_app.logger.warning(jsonify([v['ip'] for k, v in consumer_list.items()])) - return jsonify([v['ip'] for k, v in consumer_list.items()]) + + keys = redis_client.keys('consumer_*') + + list_of_customer_ips = [] + + for key in keys: + info = json.loads((redis_client.get(key) or b"{}").decode('utf-8')) + list_of_customer_ips.append(info['ip']) + + return jsonify(list_of_customer_ips) diff --git a/consumer_api/views/log_view.py b/consumer_api/views/log_view.py index d2a4d15..2ebf4ff 100644 --- a/consumer_api/views/log_view.py +++ b/consumer_api/views/log_view.py @@ -23,7 +23,7 @@ class LogView(FlaskView): # update expirity redis_client.set(prod_key, remote_ip.encode('utf-8')) - redis_client.expire(prod_key, 240) + redis_client.expire(prod_key, current_app.config["PRODUCER_TIMEOUT"]) # print out message current_app.logger.info(f"New message: {request.json['message']}") diff --git a/consumer_api/views/sync_view.py b/consumer_api/views/sync_view.py index a6f8eb5..e0dc13a 100644 --- a/consumer_api/views/sync_view.py +++ b/consumer_api/views/sync_view.py @@ -1,4 +1,5 @@ import json +import time from flask import request, current_app, jsonify from flask_classful import FlaskView from db import redis_client # ez nagyon otvar @@ -10,20 +11,25 @@ class SyncView(FlaskView): remote_uuid = request.json['uuid'] remote_ip = request.remote_addr - consumer_list = json.loads((redis_client.get("consumer_list") or b"{}").decode('utf-8')) + cust_key = f"consumer_{remote_uuid}" - # Log something about it - if remote_uuid not in consumer_list.keys(): + last_known_info = json.loads((redis_client.get(cust_key) or b"{}").decode('utf-8')) + + if not last_known_info: current_app.logger.info(f"New consumer registered (unknown UUID): {remote_uuid} at {remote_ip}") - else: # known - if consumer_list[remote_uuid]['ip'] != remote_ip: + + else: + if last_known_info['ip'] != remote_ip: current_app.logger.info(f"Address of consumer {remote_uuid} changed to {remote_ip}") - consumer_list.update( - {remote_uuid: {"ip": remote_ip}} - ) + info = { + "uuid": remote_uuid, + "ip": remote_ip, + "last_seen": time.time() + } - redis_client.set("consumer_list", json.dumps(consumer_list).encode('utf-8')) + redis_client.set(cust_key, json.dumps(info).encode('utf-8')) + redis_client.expire(cust_key, current_app.config["CUSTOMER_TIMEOUT"]) response = { "uuid": current_app.config['LOCAL_UUID']