class ConsulHandler:
auth_address = ""
club_address = ""
def __init__(self):
self.consul = Consul(host=ConsulConfig.host, port=ConsulConfig.port)
self.consul_agent = self.consul.Agent(self.consul)
self.consul_service = self.consul_agent.Service(self.consul)
self.consul_check = self.consul_agent.Check(self.consul)
ConsulConfig.service_host = gethostbyname(gethostname())
ConsulConfig.service_port = self.create_service_port()
ConsulConfig.service_id = f"{ConsulConfig.service_name}-{uuid.uuid4()}"
def register_consul(self, port):
self.consul_service.register(
name=ConsulConfig.service_name,
service_id=ConsulConfig.service_id,
address=ConsulConfig.service_host,
port=port,
token=ConsulConfig.token,
)
self.consul_check.register(
name=f"service '{ConsulConfig.service_name}' check",
check=Check.ttl("10000000s"),
check_id=ConsulConfig.check_id,
service_id=ConsulConfig.service_id,
token=ConsulConfig.token,
)
self.consul_check.ttl_pass(ConsulConfig.check_id)
def deregister_consul(self):
self.consul_service.deregister(ConsulConfig.service_id)
self.consul_check.deregister(ConsulConfig.check_id)
def update_address(self):
ConsulHandler.auth_address = self.get_address(auth_service_name)
ConsulHandler.club_address = self.get_address(club_service_name)
def get_address(self, service: str) -> Optional[str]:
services = self.consul_agent.services()
checks = self.consul_agent.checks()
for service_id in services:
if services[service_id]["Service"] == service:
if checks[f"service:{service_id}"]["Status"] == "passing":
return (
f"{services[service_id]['Address']}:{services[service_id]['Port']}"
)
return None
def get_db_info(self) -> dict:
return json.loads(self.consul.kv.get("db/outing/local")[1]["Value"].decode())
def get_redis_info(self) -> dict:
return json.loads(self.consul.kv.get("redis/outing/local")[1]["Value"].decode())
def create_service_port(self) -> int:
port = self.generate_service_port()
while self.check_service_port(port): port = self.generate_service_port()
return port
def check_service_port(self, port) -> bool:
services = self.consul_agent.services()
for service in services:
if services[service]["Port"] == port: return True
else:
return False
def generate_service_port(self) -> int:
return random.randrange(10101, 10200)
