From 5aae9702e14a08322bfdf2f5e5ac0dbe7fddcd06 Mon Sep 17 00:00:00 2001
From: Leon Kiefer <leon.k97@gmx.de>
Date: Tue, 19 Mar 2024 14:01:05 +0100
Subject: [PATCH] Add kv-store connection check to readiness probe

Signed-off-by: Leon Kiefer <leon.k97@gmx.de>
---
 src/main/java/com/ibm/watson/modelmesh/ModelMesh.java | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/main/java/com/ibm/watson/modelmesh/ModelMesh.java b/src/main/java/com/ibm/watson/modelmesh/ModelMesh.java
index f8779651..3795178d 100644
--- a/src/main/java/com/ibm/watson/modelmesh/ModelMesh.java
+++ b/src/main/java/com/ibm/watson/modelmesh/ModelMesh.java
@@ -1327,6 +1327,11 @@ protected boolean isReady() {
             reportReady = true;
         }
 
+        if (!verifyKvStoreConnection()) {
+            logger.info("Returning NOT READY to readiness probe due to kv-store connection failure");
+            return false;
+        }
+
         return true;
     }