143 lines
3.9 KiB
Python
143 lines
3.9 KiB
Python
from pathlib import Path
|
|
from stats import AppStats
|
|
import errno
|
|
import json
|
|
import traceback
|
|
import socket
|
|
import stats
|
|
from stats import WorkerAppStats
|
|
from stats import AppStats
|
|
from collections import Counter
|
|
from rich.console import Console
|
|
import sys
|
|
import glob
|
|
|
|
from stats import RouterSubscription
|
|
import time
|
|
from rich.live import Live
|
|
from rich.table import Table
|
|
import pydantic
|
|
|
|
class SocketStats(pydantic.BaseModel):
|
|
name: str
|
|
queue: int
|
|
|
|
class WorkerStats(pydantic.BaseModel):
|
|
id: int
|
|
accepting: int
|
|
|
|
class AppStats(pydantic.BaseModel):
|
|
model_config = pydantic.ConfigDict(strict=True)
|
|
version: str
|
|
listen_queue: int = 0
|
|
listen_queue_errors: int = 0
|
|
signal_queue: int
|
|
load: int
|
|
pid: int
|
|
uid: int
|
|
gid: int
|
|
cwd: str
|
|
locks: list[dict[str, int]]
|
|
sockets: list[SocketStats]
|
|
workers: list[WorkerStats]
|
|
|
|
# import ipdb;ipdb.set_trace()
|
|
|
|
|
|
|
|
class RouterStats(pydantic.BaseModel):
|
|
model_config = pydantic.ConfigDict(strict=True)
|
|
version: str
|
|
pid: int = pydantic.Field(ge=0)
|
|
uid: int = pydantic.Field(ge=0)
|
|
gid: int = pydantic.Field(ge=0)
|
|
cwd: str
|
|
active_sessions: int = pydantic.Field(ge=0)
|
|
http: list[str] # ['0.0.0.0:8034', '127.0.0.1:5700'],
|
|
subscriptions: list[RouterSubscription]
|
|
cheap: int = pydantic.Field(ge=0)
|
|
|
|
def read_stats(stats_address):
|
|
if not all([stats_address.exists(), stats_address.is_socket()]):
|
|
raise Exception(f"unable to read stats from {(stats_address)}")
|
|
|
|
def unix_addr(arg):
|
|
sfamily = socket.AF_UNIX
|
|
addr = arg
|
|
return sfamily, addr, socket.gethostname()
|
|
|
|
js = ""
|
|
sfamily, addr, host = unix_addr(stats_address)
|
|
# console.info(f"{sfamily=} {str(addr)=} {host=}")
|
|
|
|
try:
|
|
s = None
|
|
s = socket.socket(sfamily, socket.SOCK_STREAM)
|
|
s.connect(str(addr))
|
|
while True:
|
|
data = s.recv(4096)
|
|
if len(data) < 1:
|
|
break
|
|
js += data.decode('utf8', 'ignore')
|
|
if s:
|
|
s.close()
|
|
except ConnectionRefusedError as e:
|
|
print('connection refused')
|
|
except FileNotFoundError as e:
|
|
print(f"socket @ {addr} not available")
|
|
except IOError as e:
|
|
if e.errno != errno.EINTR:
|
|
#uwsgi.log(f"socket @ {addr} not available")
|
|
pass
|
|
except Exception:
|
|
print(traceback.format_exc())
|
|
else:
|
|
try:
|
|
return json.loads(js)
|
|
except json.JSONDecodeError:
|
|
print(traceback.format_exc())
|
|
print(js)
|
|
|
|
if __name__ == "__main__":
|
|
stats_list = ["run/sub1-stats.sock", "run/sub2-stats.sock"]
|
|
|
|
table = Table()
|
|
table.add_column("Virtual Hosts", justify="right", style="cyan", no_wrap=True)
|
|
table.add_column("Nodes", style="magenta")
|
|
table.add_column("Total Workers", style="magenta")
|
|
|
|
for sock in stats_list:
|
|
stats_address = Path(sock)
|
|
stats = read_stats(stats_address)
|
|
print(stats)
|
|
|
|
router_stats = RouterStats(**stats)
|
|
|
|
all_app_stats = {}
|
|
total_workers = 0
|
|
|
|
for sub in router_stats.subscriptions:
|
|
print(sub)
|
|
for node in sub.nodes:
|
|
print(node.name)
|
|
port = node.name.split(":")[-1]
|
|
print(port)
|
|
stats_address2 = Path(f"run/{port}_stats.sock")
|
|
stats2 = read_stats(stats_address2)
|
|
print(stats2)
|
|
app_stats = AppStats(**stats2)
|
|
print(f"App Stats for {port}:{app_stats}")
|
|
print(app_stats.workers)
|
|
print(len(app_stats.workers))
|
|
worker_amount = len(app_stats.workers)
|
|
total_workers = total_workers + worker_amount
|
|
print(total_workers)
|
|
for a in router_stats.subscriptions:
|
|
print(a.key)
|
|
|
|
table.add_row(a.key, f"{len(sub.nodes)}", f"{total_workers}")
|
|
|
|
console = Console()
|
|
console.print(table)
|
|
|