From 8790cde5a1450fc6c3f877ee2ab8a8e7b4266820 Mon Sep 17 00:00:00 2001
From: Aakash Singh <mail@singhaakash.dev>
Date: Wed, 5 Jun 2024 23:12:56 +0530
Subject: [PATCH] Add health check for celery queue length (#2244)

add health check for celery queue length
---
 config/settings/base.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/config/settings/base.py b/config/settings/base.py
index 03e764e8b3..dda4a2067f 100644
--- a/config/settings/base.py
+++ b/config/settings/base.py
@@ -9,6 +9,9 @@
 
 import environ
 from authlib.jose import JsonWebKey
+from healthy_django.healthcheck.celery_queue_length import (
+    DjangoCeleryQueueLengthHealthCheck,
+)
 from healthy_django.healthcheck.django_cache import DjangoCacheHealthCheck
 from healthy_django.healthcheck.django_database import DjangoDatabaseHealthCheck
 
@@ -415,6 +418,15 @@
         "Database", slug="main_database", connection_name="default"
     ),
     DjangoCacheHealthCheck("Cache", slug="main_cache", connection_name="default"),
+    DjangoCeleryQueueLengthHealthCheck(
+        "Celery Queue Length",
+        slug="celery_queue_length",
+        broker=REDIS_URL,
+        queue_name="celery",
+        info_length=50,
+        warning_length=0,  # this skips the 300 status code
+        alert_length=200,
+    ),
 ]
 
 # Audit logs