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

add retry for prometheus endpoint metric getter

Browse Source
This commit is contained in:
Julius Unverfehrt 2022-03-11 10:09:39 +01:00
parent 963132afb1
commit 91980ff3ed
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pyinfra/flask.py
View File

@ -1,5 +1,6 @@
import requests
from flask import Flask, jsonify
from retry import retry
from waitress import serve
from pyinfra.config import CONFIG
@ -39,8 +40,12 @@ def set_up_probing_webserver():
@app.route("/prometheus", methods=["GET"])
def get_analysis_prometheus_endpoint():
prom_endpoint = f"{CONFIG.rabbitmq.callback.analysis_endpoint}/prometheus"
metric = requests.get(prom_endpoint)
return metric.text
@retry(requests.exceptions.ConnectionError, tries=3, delay=5, jitter=(1, 3))
def inner():
prom_endpoint = f"{CONFIG.rabbitmq.callback.analysis_endpoint}/prometheus"
metric = requests.get(prom_endpoint)
metric.raise_for_status()
return metric.text
return inner()
return app