healthcheck

Haidra-Org · Jun 20, 2024 · 5061bc9 · 5061bc9
1 parent 2272165
commit 5061bc9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/horde/apis/v2/base.py b/horde/apis/v2/base.py
@@ -2644,13 +2644,17 @@ class Heartbeat(Resource):
     def get(self):
         """If this loads, this node is available
         Includes some other metrics to gauge the health of this node"""
-        db_conn = True
+        health = 'OK'
+        db_conn = True        
         try:
             db.session.execute(text("SELECT 1"))
         except Exception:
             db_conn = False
+            health = 'DOWN'
+        if waitress_metrics.queue > 0:
+            health = 'OVERLOADED'
         return {
-            "message": "OK",
+            "message": health,
             "version": HORDE_VERSION,
             "queue": waitress_metrics.queue,
             "threads": waitress_metrics.threads,

diff --git a/horde/consts.py b/horde/consts.py
@@ -1,4 +1,4 @@
-HORDE_VERSION = "4.38.0"
+HORDE_VERSION = "4.38.1"
 
 WHITELISTED_SERVICE_IPS = {
     "212.227.227.178",  # Turing Bot