first commit

2025-01-30 18:17:43 -05:00
commit ac281878ce
22 changed files with 622 additions and 0 deletions
--- a/read_stats.py
+++ b/read_stats.py
@@ -0,0 +1,143 @@
+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 rich.console import Console
+
+from stats import RouterSubscription
+import time
+from rich.live import Live
+from rich.table import Table
+import pydantic
+
+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 run(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_address = Path("run/sub1-stats.sock")
+    stats = run(stats_address)
+
+
+class AppStats(pydantic.BaseModel):
+    model_config = pydantic.ConfigDict(strict=True)
+    version: str
+    listen_queue: int
+    listen_queue_errors: int
+    signal_queue: int
+    load: int
+    pid: int
+    uid: int
+    gid: int
+    cwd: str
+    locks: list[dict[str, int]]
+    sockets: list[stats.SocketStats]
+    workers: list[stats.WorkerStats]
+
+    def read_stats(socket_address):
+        sockets = {
+            "9027_stats.sock": "{app1 - stats}"
+        }
+        if socket_address in sockets:
+            return sockets[socket_address]
+        else:
+            return "could not lookup socket"
+
+    if __name__ == "__main__":
+        counter = 0
+        nodes = ["127.0.0.1:9027", "127.0.0.1:9030"]
+
+
+        for node in nodes:
+            port = node.split(":")[-1]
+            socket_address = f"{port}_stats.sock"
+            stats = read_stats(socket_address)
+            print(stats)
+
+            stats2 = run(Path("run/app1_stats.sock"))
+            app_stats = AppStats(**stats2)
+            print(app_stats.workers)
+            worker_amount = app_stats.workers
+            print(len(app_stats.workers))
+
+
+
+
+
+    router_stats = RouterStats(**stats)
+    print(router_stats)
+
+    # import ipdb;ipdb.set_trace()
+    for sub in router_stats.subscriptions:
+        print(f"{sub.key=}")
+
+    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")
+
+    table.add_row(sub.key, f"{len(sub.nodes)}")
+
+
+
+    console = Console()
+    console.print(table)
+