Report & metrics server

This commit is contained in:
Ethanell 2021-07-10 15:58:31 +02:00
parent c039b2410a
commit c13b08ec8b
2 changed files with 69 additions and 0 deletions

61
app.py Normal file
View file

@ -0,0 +1,61 @@
from re import sub
from flask import Flask, request, Response
from prometheus_client import make_wsgi_app, Gauge, Info
from werkzeug.middleware.dispatcher import DispatcherMiddleware
app = Flask(__name__)
app.wsgi_app = DispatcherMiddleware(app.wsgi_app, {
"/metrics": make_wsgi_app()
})
metrics = {}
def to_snake_case(name):
name = sub('(.)([A-Z][a-z]+)', r'\1_\2', name)
name = sub('__([A-Z])', r'_\1', name)
name = sub('([a-z0-9])([A-Z])', r'\1_\2', name)
return name.lower()
def pars_data(data_name: str, data_value, operation_name: str, backup_name: str):
if isinstance(data_value, dict):
for dd, vv in data_value.items():
pars_data(f"{data_name}_{dd}", vv, operation_name, backup_name)
else:
set_metric(data_name, data_value, operation_name, backup_name)
def set_metric(name: str, value, operation_name: str, backup_name: str):
metric_name = to_snake_case(name)
if name not in metrics:
if isinstance(value, int):
metrics[name] = Gauge(metric_name, name, ["operation", "name"])
elif isinstance(value, str) or isinstance(value, bool):
metrics[name] = Info(metric_name, name, ["operation", "name"])
else:
return
m = metrics[name]
if isinstance(m, Gauge):
m.labels(operation_name, backup_name).set(value)
elif isinstance(m, Info):
m.labels(operation_name, backup_name).info({"data": value})
@app.route("/report", methods=["POST"])
def report() -> Response:
request.get_json()
operation_name = request.json["Extra"]["OperationName"]
backup_name = request.json["Extra"]["backup-name"]
pars_data("duplicati", request.json["Data"], operation_name, backup_name)
return Response("Ok", 200)
if __name__ == "__main__":
app.run(host="127.0.0.1", port=80)

8
requirements.txt Normal file
View file

@ -0,0 +1,8 @@
click==8.0.1
Flask==2.0.1
itsdangerous==2.0.1
Jinja2==3.0.1
MarkupSafe==2.0.1
prometheus-client==0.11.0
uWSGI==2.0.19.1
Werkzeug==2.0.1