Consistent readiness check.

dbca-wa · Mar 1, 2024 · 0cc21bf · 0cc21bf
1 parent aeeb38b
commit 0cc21bf
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/itassets/middleware.py b/itassets/middleware.py
@@ -30,14 +30,13 @@ def readiness(self, request):
         """
         try:
             from django.db import connections
-            for name in connections:
-                cursor = connections[name].cursor()
-                cursor.execute("SELECT 1;")
-                row = cursor.fetchone()
-                if row is None:
-                    return HttpResponseServerError("db: invalid response")
+            cursor = connections["default"].cursor()
+            cursor.execute("SELECT 1;")
+            row = cursor.fetchone()
+            if row is None:
+                return HttpResponseServerError("Database: invalid response")
         except Exception as e:
             LOGGER.exception(e)
-            return HttpResponseServerError("db: cannot connect to database.")
+            return HttpResponseServerError("Database: unable to connect")
 
         return HttpResponse("OK")