123 lines
3.0 KiB
Python
123 lines
3.0 KiB
Python
# Copyright (c) 2021, JINGROW
|
|
# For license information, please see license.txt
|
|
|
|
|
|
from itertools import groupby
|
|
|
|
import jingrow
|
|
|
|
from jcloud.exceptions import AlertRuleNotEnabled
|
|
from jcloud.utils import log_error
|
|
|
|
|
|
def get_benches():
|
|
self_hosted_stand_alone_servers = jingrow.get_all(
|
|
"Server",
|
|
{"is_standalone": True, "is_self_hosted": True, "status": "Active"},
|
|
pluck="name",
|
|
)
|
|
sites = jingrow.get_all(
|
|
"Site",
|
|
["name", "bench"],
|
|
{"status": "Active", "server": ("not in", self_hosted_stand_alone_servers)},
|
|
ignore_ifnull=True,
|
|
)
|
|
sites.sort(key=lambda x: (x.bench, x.name))
|
|
|
|
bench_map = {
|
|
bench.name: bench
|
|
for bench in jingrow.get_all(
|
|
"Bench",
|
|
{"name": ("in", set(site.bench for site in sites))},
|
|
["name", "cluster", "server", "group"],
|
|
ignore_ifnull=True,
|
|
)
|
|
}
|
|
benches = []
|
|
for bench_name, _sites in groupby(sites, lambda x: x.bench):
|
|
bench = bench_map[bench_name]
|
|
bench.update({"sites": [site.name for site in _sites]})
|
|
benches.append(bench)
|
|
|
|
return benches
|
|
|
|
|
|
def get_clusters():
|
|
servers = {}
|
|
servers["proxy"] = jingrow.get_all("Proxy Server", {"status": ("!=", "Archived")}, ["name", "cluster"])
|
|
servers["app"] = jingrow.get_all("Server", {"status": ("!=", "Archived")}, ["name", "cluster"])
|
|
servers["database"] = jingrow.get_all(
|
|
"Database Server", {"status": ("!=", "Archived")}, ["name", "cluster"]
|
|
)
|
|
clusters = jingrow.get_all("Cluster")
|
|
job_map = {
|
|
"proxy": ["node", "nginx", "proxysql", "mariadb_proxy"],
|
|
"app": ["node", "nginx", "docker", "cadvisor", "gunicorn", "rq"],
|
|
"database": ["node", "mariadb"],
|
|
}
|
|
for cluster in clusters:
|
|
cluster["jobs"] = {}
|
|
|
|
for server_type, server_type_servers in servers.items():
|
|
for server in server_type_servers:
|
|
if server.cluster == cluster.name:
|
|
for job in job_map[server_type]:
|
|
cluster["jobs"].setdefault(job, []).append(server.name)
|
|
|
|
return clusters
|
|
|
|
|
|
def get_domains():
|
|
return jingrow.get_all(
|
|
"Site Domain", ["name", "site"], {"tls_certificate": ("is", "set")}, order_by="name"
|
|
)
|
|
|
|
|
|
def get_tls():
|
|
tls = []
|
|
server_types = [
|
|
"Server",
|
|
"Proxy Server",
|
|
"Database Server",
|
|
"Registry Server",
|
|
"Log Server",
|
|
"Monitor Server",
|
|
"Analytics Server",
|
|
"Trace Server",
|
|
]
|
|
for server_type in server_types:
|
|
tls += jingrow.get_all(server_type, {"status": ("!=", "Archived")}, ["name"])
|
|
|
|
return tls
|
|
|
|
|
|
@jingrow.whitelist(allow_guest=True)
|
|
def targets(token):
|
|
monitor_token = jingrow.db.get_single_value("Jcloud Settings", "monitor_token")
|
|
if token != monitor_token:
|
|
return None
|
|
|
|
return {"benches": get_benches(), "clusters": get_clusters(), "domains": get_domains(), "tls": get_tls()}
|
|
|
|
|
|
@jingrow.whitelist(allow_guest=True, xss_safe=True)
|
|
def alert(*args, **kwargs):
|
|
try:
|
|
user = str(jingrow.session.user)
|
|
jingrow.set_user("Administrator")
|
|
|
|
pg = jingrow.get_pg(
|
|
{
|
|
"pagetype": "Alertmanager Webhook Log",
|
|
"payload": jingrow.request.get_data().decode(),
|
|
}
|
|
)
|
|
pg.insert()
|
|
except AlertRuleNotEnabled:
|
|
pass
|
|
except Exception:
|
|
log_error("Alertmanager Webhook Error", args=args, kwargs=kwargs)
|
|
raise
|
|
finally:
|
|
jingrow.set_user(user)
|