Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

added prometheus endpoint to tunnel metrics from analysis endpoint

Browse Source
This commit is contained in:
Julius Unverfehrt 2022-03-09 15:26:19 +01:00
parent 21592342d5
commit ee88be4c80
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pyinfra/flask.py
View File

@ -1,3 +1,4 @@
import requests
from flask import Flask, jsonify
from waitress import serve
@ -5,7 +6,6 @@ from pyinfra.config import CONFIG
def run_probing_webserver(app, host=None, port=None, mode=None):
if not host:
host = CONFIG.probing_webserver.host
@ -23,7 +23,6 @@ def run_probing_webserver(app, host=None, port=None, mode=None):
def set_up_probing_webserver():
app = Flask(__name__)
@app.route("/ready", methods=["GET"])
@ -38,4 +37,10 @@ def set_up_probing_webserver():
resp.status_code = 200
return resp
@app.route("/prometheus", methods=["GET"])
def get_analysis_prometheus():
prom_endpoint = f"{CONFIG.rabbitmq.callback.analysis_endpoint}/prometheus"
metric = requests.get(prom_endpoint)
return metric.text
return app