jan/openreplay
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(chalice): health-check test

Browse source
This commit is contained in:
Taha Yassine Kraiem 2023-03-07 18:36:48 +01:00
parent 687dfc1857
commit d86b71f66e
2 changed files with 30 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
api/chalicelib/core/health.py
View file

@ -133,7 +133,7 @@ def __check_redis():
r = redis.Redis(host=u.hostname, port=u.port, socket_timeout=2)
r.ping()
except Exception as e:
print("!! Issue getting assist-health response")
print("!! Issue getting redis-health response")
print(str(e))
fail_response["details"]["errors"].append(str(e))
return fail_response

32
ee/api/chalicelib/core/health.py
View file

@ -3,6 +3,7 @@ from urllib.parse import urlparse
import redis
import requests
from decouple import config
import kafka
from chalicelib.utils import pg_client, ch_client
@ -133,7 +134,7 @@ def __check_redis():
r = redis.Redis(host=u.hostname, port=u.port, socket_timeout=2)
r.ping()
except Exception as e:
print("!! Issue getting assist-health response")
print("!! Issue getting redis-health response")
print(str(e))
fail_response["details"]["errors"].append(str(e))
return fail_response
@ -156,7 +157,7 @@ def get_health():
},
"ingestionPipeline": {
"redis": __check_redis,
"kafka": __not_supported
"kafka": __check_kafka
},
"backendServices": {
"alerts": __check_be_service("alerts"),
@ -219,4 +220,29 @@ def __check_database_ch():
def __check_kafka():
pass
fail_response = {
"health": False,
"details": {"errors": ["server health-check failed"]}
}
if config("KAFKA_SERVERS", default=None) is None:
fail_response["details"]["errors"].append("KAFKA_SERVERS not defined in env-vars")
return fail_response
try:
# consumer = kafka.KafkaConsumer(group_id='test', bootstrap_servers=[config("KAFKA_SERVERS")])
# topics = consumer.topics()
#
# if not topics:
# raise RuntimeError()
client =kafka.KafkaClient(bootstrap_servers=[config("KAFKA_SERVERS")])
except Exception as e:
print("!! Issue getting kafka-health response")
print(str(e))
fail_response["details"]["errors"].append(str(e))
return fail_response
return {
"health": True,
"details": {"version": r.execute_command('INFO')['redis_version']}
}