mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-07-21 11:11:03 +02:00
68 lines
1.2 KiB
Python
Executable file
68 lines
1.2 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
"""
|
|
Dump out TrustGraph processor states.
|
|
"""
|
|
|
|
import requests
|
|
import argparse
|
|
import json
|
|
|
|
default_metrics_url = "http://localhost:8088/api/metrics"
|
|
|
|
def dump_status(url, flow_id):
|
|
|
|
url = f"{url}/query"
|
|
|
|
expr = f"consumer_state=\"running\",flow=\"{flow_id}\""
|
|
|
|
params = {
|
|
"query": "consumer_state{" + expr + "}"
|
|
}
|
|
|
|
resp = requests.get(url, params=params)
|
|
|
|
obj = resp.json()
|
|
|
|
tbl = [
|
|
[
|
|
m["metric"]["job"],
|
|
"\U0001f49a" if int(m["value"][1]) > 0 else "\U0000274c"
|
|
]
|
|
for m in obj["data"]["result"]
|
|
]
|
|
|
|
for row in tbl:
|
|
print(f"- {row[0]:30} {row[1]}")
|
|
|
|
def main():
|
|
|
|
parser = argparse.ArgumentParser(
|
|
prog='tg-show-flow-state',
|
|
description=__doc__,
|
|
)
|
|
|
|
parser.add_argument(
|
|
'-f', '--flow-id',
|
|
default="0000",
|
|
help=f'Flow ID (default: 0000)'
|
|
)
|
|
|
|
parser.add_argument(
|
|
'-m', '--metrics-url',
|
|
default=default_metrics_url,
|
|
help=f'Metrics URL (default: {default_metrics_url})',
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
|
|
dump_status(args.metrics_url, args.flow_id)
|
|
|
|
except Exception as e:
|
|
|
|
print("Exception:", e, flush=True)
|
|
|
|
main()
|
|
|