Refactor database connection count retrieval and improve code readabi…

…lity in core views
OWASP-BLT · Jan 1, 2025 · 3ed77c0 · 3ed77c0
1 parent 77e0bae
commit 3ed77c0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/website/views/core.py b/website/views/core.py
@@ -154,7 +154,9 @@ def check_status(request):
         )[:5]
 
         # Get database connection count
-        status["db_connection_count"] = len(connection.queries)
+        with connection.cursor() as cursor:
+            cursor.execute("SELECT COUNT(*) FROM pg_stat_activity WHERE state = 'active'")
+            status["db_connection_count"] = cursor.fetchone()[0]
 
         # Get Redis stats using django-redis
         redis_client = get_redis_connection("default")