Added amqp

This commit is contained in:
2021-08-19 03:25:30 +02:00
parent 5a5c510fa0
commit fdacc7a36d
6 changed files with 148 additions and 2 deletions

View File

@@ -1,5 +1,5 @@
#!/usr/bin/env python3
from utils import storage
from utils import storage, magic_amqp
from healthcheck import HealthCheck
from flask import Flask
@@ -15,7 +15,19 @@ def health_database_status():
return is_database_working, output
def amqp_connection_status():
if magic_amqp.is_healthy():
result = True
text = "amqp connection is ok"
else:
result = False
text = "amqp connection is unhealthy"
return result, text
def register_health_checks(app: Flask):
health = HealthCheck()
health.add_check(health_database_status)
health.add_check(amqp_connection_status)
app.add_url_rule("/healthz", "healthcheck", view_func=lambda: health.run())