mirror of
https://github.com/optim-enterprises-bv/openlan-cgw.git
synced 2025-10-31 18:27:46 +00:00
49555495e8
Force redis, PSQL, kafka clients to have a strict timeout value for any requests. It's done to ensure our tests do not hang for an unknown period of time in case if something goes wrong. Signed-off-by: Oleksandr Mazur <oleksandr.mazur@plvision.eu>
118 lines
3.3 KiB
Python
118 lines
3.3 KiB
Python
import re
|
|
import requests
|
|
|
|
|
|
def cgw_metric_get(host: str = "localhost", port: int = 8080) -> str:
|
|
metrics = ""
|
|
|
|
try:
|
|
# Try to fetch metrics with 5 seconds timeout value
|
|
r = requests.get(f"http://{host}:{port}/metrics", timeout=5)
|
|
print("CGW metrics ret code: " + str(r.status_code))
|
|
assert r is not None and r.status_code == 200, \
|
|
f"CGW metrics is not available"
|
|
metrics = r.text
|
|
except Exception as e:
|
|
print("CGW metrics: raised exception when tried to fetch metrics:" + e)
|
|
raise Exception('CGW metrics fetch failed (Not running?)')
|
|
|
|
return metrics
|
|
|
|
|
|
def cgw_metrics_get_active_shards_num() -> int:
|
|
active_shards_num = 0
|
|
metrics = cgw_metric_get()
|
|
|
|
match = re.search(r"cgw_active_shards_num (\d+)", metrics)
|
|
if match:
|
|
active_shards_num = int(match.group(1))
|
|
print(f"Active shards num: {active_shards_num}")
|
|
else:
|
|
print("Active shards num not found.")
|
|
|
|
return active_shards_num
|
|
|
|
|
|
def cgw_metrics_get_connections_num() -> int:
|
|
wss_connections_num = 0
|
|
metrics = cgw_metric_get()
|
|
|
|
match = re.search(r"cgw_connections_num (\d+)", metrics)
|
|
if match:
|
|
wss_connections_num = int(match.group(1))
|
|
print(f"WSS conections num: {wss_connections_num}")
|
|
else:
|
|
print("WSS conections num not found.")
|
|
|
|
return wss_connections_num
|
|
|
|
|
|
def cgw_metrics_get_groups_assigned_num() -> int:
|
|
groups_assigned_num = 0
|
|
metrics = cgw_metric_get()
|
|
|
|
match = re.search(r"cgw_groups_assigned_num (\d+)", metrics)
|
|
if match:
|
|
groups_assigned_num = int(match.group(1))
|
|
print(f"Groups assigned num: {groups_assigned_num}")
|
|
else:
|
|
print("Groups assigned num not found.")
|
|
|
|
return groups_assigned_num
|
|
|
|
|
|
def cgw_metrics_get_group_infras_assigned_num(group_id: int) -> int:
|
|
group_infras_assigned_num = 0
|
|
metrics = cgw_metric_get()
|
|
|
|
match = re.search(rf"cgw_group_{group_id}_infras_assigned_num (\d+)", metrics)
|
|
if match:
|
|
group_infras_assigned_num = int(match.group(1))
|
|
print(f"Group {group_id} infras assigned num: {group_infras_assigned_num}")
|
|
else:
|
|
print(f"Group {group_id} infras assigned num not found.")
|
|
|
|
return group_infras_assigned_num
|
|
|
|
|
|
def cgw_metrics_get_groups_capacity() -> int:
|
|
groups_capacity = 0
|
|
metrics = cgw_metric_get()
|
|
|
|
match = re.search(r"cgw_groups_capacity (\d+)", metrics)
|
|
if match:
|
|
groups_capacity = int(match.group(1))
|
|
print(f"Groups capacity: {groups_capacity}")
|
|
else:
|
|
print("Groups capacity.")
|
|
|
|
return groups_capacity
|
|
|
|
|
|
def cgw_metrics_get_groups_threshold() -> int:
|
|
groups_threshold = 0
|
|
metrics = cgw_metric_get()
|
|
|
|
match = re.search(r"cgw_groups_threshold (\d+)", metrics)
|
|
if match:
|
|
groups_threshold = int(match.group(1))
|
|
print(f"Groups assigned num: {groups_threshold}")
|
|
else:
|
|
print("Groups assigned num not found.")
|
|
|
|
return groups_threshold
|
|
|
|
|
|
def cgw_metrics_get_group_ifras_capacity() -> int:
|
|
group_infras_capacity = 0
|
|
metrics = cgw_metric_get()
|
|
|
|
match = re.search(r"cgw_group_ifras_capacity (\d+)", metrics)
|
|
if match:
|
|
group_infras_capacity = int(match.group(1))
|
|
print(f"Groups capacity: {group_infras_capacity}")
|
|
else:
|
|
print("Groups capacity.")
|
|
|
|
return group_infras_capacity
|