added saving producer ips to redis

Dockerfile

@@ -9,5 +9,5 @@ RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
 
 RUN pip3 install -r requirements.txt
 
-EXPOSE 8000
+EXPOSE 80
-CMD ["gunicorn", "-b", "0.0.0.0:8000", "app:app"]
+CMD ["gunicorn", "-b", "0.0.0.0:80", "app:app"]

consumer_api/views/consumers_view.py

@@ -1,7 +1,7 @@
 import json
 
 from db import redis_client
-from flask import jsonify
+from flask import jsonify, current_app
 from flask_classful import FlaskView
 
 
@@ -9,4 +9,5 @@ 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()])

consumer_api/views/log_view.py

@@ -1,9 +1,31 @@
 from flask import request, current_app, Response
 from flask_classful import FlaskView
+from db import redis_client
 
 
 class LogView(FlaskView):
 
     def post(self):
+        # Record the IP address of the producer
+        remote_uuid = request.json['uuid']
+        remote_ip = request.remote_addr
+
+        prod_key = f"producer_{remote_uuid}"
+
+        last_known_remote_ip = redis_client.get(prod_key)
+        if last_known_remote_ip:
+            last_known_remote_ip = last_known_remote_ip.decode('utf-8')
+
+        if not last_known_remote_ip:
+            current_app.logger.info(f"New producer {remote_uuid} at {remote_ip}")
+        elif last_known_remote_ip != remote_ip:
+            current_app.logger.info(f"IP address of producer {remote_uuid} have changed: {last_known_remote_ip} -> {remote_ip}")
+
+        # update expirity
+        redis_client.set(prod_key, remote_ip.encode('utf-8'))
+        redis_client.expire(prod_key, 240)
+
+        # print out message
         current_app.logger.info(f"New message: {request.json['message']}")
-        return Response(status = 204)
+
+        return Response(status=204)