Implemented ip change checking
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
2020-05-08 22:21:15 +02:00
parent 7ce88f1a74
commit 0713cabc6b
3 changed files with 47 additions and 17 deletions

View File

@@ -2,12 +2,11 @@
import redis
import os
import json
import socket
import time
import logging
class RedisSuperStorage:
def __init__(self, redis_url: str, timeout: int):
self.r = redis.from_url(redis_url)
self._timeout = timeout
@@ -38,21 +37,6 @@ class RedisSuperStorage:
return list_of_producer_ip
def is_ip_changed(self) -> bool:
old_ip = self.r.get('current_ip')
if old_ip:
old_ip = old_ip.decode('utf-8')
host_name = socket.gethostname()
current_ip = socket.gethostbyname(host_name)
if current_ip != old_ip:
logging.info(f'IP changed: {old_ip} -> {current_ip}')
self.r.set('current_ip', current_ip.encode('utf-8'))
return current_ip != old_ip
def update_consumer(self, uuid: str, ip: str):
cust_key = f"consumer_{uuid}"
@@ -65,3 +49,16 @@ class RedisSuperStorage:
self.r.set(cust_key, json.dumps(info).encode('utf-8'))
self.r.expire(cust_key, os.environ["CUSTOMER_TIMEOUT"])
def get_current_ip(self) -> str:
ip = self.r.get('current_ip')
if ip:
ip = ip.decode('utf-8')
return ip
def set_current_ip(self, ip: str):
self.r.set('current_ip', ip.encode('utf-8'))
current_ip = property(get_current_ip, set_current_ip)