Implemented ip change checking
All checks were successful
continuous-integration/drone/push Build is passing
All checks were successful
continuous-integration/drone/push Build is passing
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user