durch
120 lines
3.1 KiB
Python
120 lines
3.1 KiB
Python
import argparse
|
|
import os
|
|
import requests
|
|
import json
|
|
from collections import namedtuple
|
|
|
|
Config = namedtuple("Config", ["port", "outfile", "outlink", "env"])
|
|
|
|
|
|
def gateway_targets(entry):
|
|
targets = [
|
|
# Config(
|
|
# None,
|
|
# "/tmp/temp_targets_gateway.json",
|
|
# f"/tmp/prom_targets_gateway_{entry['env']}.json",
|
|
# entry["env"],
|
|
# ),
|
|
Config(
|
|
9100,
|
|
"/tmp/temp_targets_gateway_node.json",
|
|
f"/tmp/prom_targets_gateway_node_{entry['env']}.json",
|
|
entry["env"],
|
|
),
|
|
]
|
|
|
|
gateways = requests.get(f"{entry['nym_api']}/api/v1/gateways").json()
|
|
|
|
for config in targets:
|
|
config_to_targets(config, gateways, {"kind": "gateway"})
|
|
|
|
|
|
def mixnode_targets(entry):
|
|
targets = [
|
|
Config(
|
|
None,
|
|
"/tmp/temp_targets_mix.json",
|
|
f"/tmp/prom_targets_mix_{entry['env']}.json",
|
|
entry["env"],
|
|
),
|
|
Config(
|
|
9100,
|
|
"/tmp/temp_targets_node.json",
|
|
f"/tmp/prom_targets_node_{entry['env']}.json",
|
|
entry["env"],
|
|
),
|
|
]
|
|
|
|
mixnodes = requests.get(f"{entry['nym_api']}/api/v1/mixnodes").json()
|
|
|
|
for config in targets:
|
|
config_to_targets(config, mixnodes, {"kind": "mixnode"})
|
|
|
|
|
|
def validate_config_entry(entry):
|
|
return entry.get("nym_api") and entry.get("env")
|
|
|
|
|
|
def config_to_targets(config, mixnodes, labels=None):
|
|
prom_targets = [
|
|
make_prom_target(config.env, mixnode, config.port, labels)
|
|
for mixnode in mixnodes
|
|
]
|
|
with open(config.outfile, "w") as f:
|
|
json.dump(prom_targets, f)
|
|
|
|
os.chmod(config.outfile, 0o777)
|
|
os.rename(config.outfile, config.outlink)
|
|
os.chmod(config.outlink, 0o777)
|
|
|
|
print(f"Prometheus -> {len(prom_targets)} targets written to {config.outlink}")
|
|
|
|
|
|
def make_prom_target(env, mixnode, port=None, labels=None):
|
|
bond_info = mixnode.get("bond_information", {})
|
|
mix_node = bond_info.get("mix_node")
|
|
if mix_node is None:
|
|
mix_node = mixnode.get("gateway")
|
|
host = mix_node.get("host", None)
|
|
port = port if port else mix_node.get("http_api_port", None)
|
|
if host is None or port is None:
|
|
return None
|
|
|
|
target = {
|
|
"targets": [f"{host}:{port}"],
|
|
"labels": {
|
|
"mix_node_host": host,
|
|
"identity_key": mix_node.get("identity_key", None),
|
|
"sphinx_key": mix_node.get("sphinx_key", None),
|
|
"mix_node_version": mix_node.get("version", None),
|
|
"mixnet_env": env,
|
|
},
|
|
}
|
|
|
|
for k, v in labels.items():
|
|
target["labels"][k] = v
|
|
|
|
return target
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(
|
|
description="Create prometheus targets for rewarded set mixnodes."
|
|
)
|
|
parser.add_argument(
|
|
"config",
|
|
type=str,
|
|
help="Config file, see scripts/prom_targets_config.json",
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
config_file = args.config
|
|
|
|
with open(config_file, "r") as f:
|
|
config = json.load(f)
|
|
|
|
for entry in config:
|
|
mixnode_targets(entry)
|
|
|
|
gateway_targets(entry)
|