From ec715926bb827478f0433e716977af9b0ea835bc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Victor=20N=C3=A4slund?= Date: Wed, 16 Nov 2022 23:43:43 +0100 Subject: fixed healthcheck path --- data/healthcheck.py | 40 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100644 data/healthcheck.py (limited to 'data/healthcheck.py') diff --git a/data/healthcheck.py b/data/healthcheck.py new file mode 100644 index 0000000..1ccd8f3 --- /dev/null +++ b/data/healthcheck.py @@ -0,0 +1,40 @@ +""" +Send a healthcheck request +""" +import sys +import time +import json + +import requests + +from src.soc_collector.auth import load_api_keys + +def check_collector() -> bool: + """Check our collector using /info + + :return: bool + """ + time.sleep(2) # Prevent race condition with mongodb container healthcheck + + req = requests.get( + "https://localhost:8000/info", + headers={"API-KEY": load_api_keys("./api_keys.txt")[-1]}, + timeout=3, + verify="./collector_root_ca.crt", + ) + + if req.status_code != 200: + return False + + data = json.loads(req.text) + if isinstance(data["Estimated document count"], int) and data["Estimated document count"] >= 0: + return req.status_code == 200 + + return False + + +if __name__ == "__main__": + if sys.argv[1] == "COLLECTOR": + if check_collector(): + sys.exit(0) + sys.exit(1) -- cgit v1.1