83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
from tabulate import tabulate
|
|
|
|
# ANSI Color Codes
|
|
GREEN = "\033[92m"
|
|
RED = "\033[91m"
|
|
CYAN = "\033[96m"
|
|
YELLOW = "\033[93m"
|
|
RESET = "\033[0m"
|
|
BOLD = "\033[1m"
|
|
|
|
def colorize(text, color, use_color=True):
|
|
if not use_color:
|
|
return text
|
|
return f"{color}{text}{RESET}"
|
|
|
|
def format_summary(cluster_data, use_color=True):
|
|
"""
|
|
Formats the cluster health summary.
|
|
"""
|
|
health = cluster_data["health"]
|
|
color = GREEN if health == "Healthy" else RED
|
|
if health == "Split Brain Detected (Multiple Primaries)":
|
|
color = YELLOW
|
|
|
|
summary = [
|
|
f"{BOLD}Cluster Health:{RESET} {colorize(health, color, use_color)}",
|
|
f"{BOLD}Total Nodes:{RESET} {len(cluster_data['nodes'])}",
|
|
f"{BOLD}Primaries:{RESET} {cluster_data['primary_count']}",
|
|
"-" * 30
|
|
]
|
|
return "\n".join(summary)
|
|
|
|
def format_node_table(nodes, use_color=True):
|
|
"""
|
|
Formats the node list as a table.
|
|
"""
|
|
headers = ["Node", "Role", "Consul Status", "LiteFS Role", "Uptime", "Replication Lag", "LiteFS Info"]
|
|
table_data = []
|
|
|
|
for node in nodes:
|
|
# Consul status color
|
|
status = node["status"]
|
|
status_color = GREEN if status == "passing" else RED
|
|
colored_status = colorize(status, status_color, use_color)
|
|
|
|
# Role color
|
|
role = node["role"]
|
|
role_color = CYAN if role == "primary" else RESET
|
|
colored_role = colorize(role, role_color, use_color)
|
|
|
|
# LiteFS role color & consistency check
|
|
litefs_primary = node["litefs_primary"]
|
|
litefs_role = "primary" if litefs_primary else "replica"
|
|
|
|
# Highlight discrepancy if consul and litefs disagree
|
|
litefs_role_color = RESET
|
|
if (role == "primary" and not litefs_primary) or (role == "replica" and litefs_primary):
|
|
litefs_role_color = YELLOW
|
|
litefs_role = f"!! {litefs_role} !!"
|
|
elif litefs_primary:
|
|
litefs_role_color = CYAN
|
|
|
|
colored_litefs_role = colorize(litefs_role, litefs_role_color, use_color)
|
|
|
|
# Error info
|
|
info = ""
|
|
if node.get("litefs_error"):
|
|
info = colorize("LiteFS API Error", RED, use_color)
|
|
else:
|
|
info = node.get("advertise_url", "")
|
|
|
|
table_data.append([
|
|
colorize(node["node"], BOLD, use_color),
|
|
colored_role,
|
|
colored_status,
|
|
colored_litefs_role,
|
|
node["uptime"],
|
|
node["replication_lag"],
|
|
info
|
|
])
|
|
|
|
return tabulate(table_data, headers=headers, tablefmt="simple")
|