Readiness probe (#191)

* Consider IDLE connection to Core as connected state * Remove debugging code
feast-dev · May 10, 2019 · 5be925c · 5be925c
1 parent a401ae3
commit 5be925c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/serving/src/main/java/feast/serving/service/CoreService.java b/serving/src/main/java/feast/serving/service/CoreService.java
@@ -142,7 +142,8 @@ public Map<String, FeatureSpec> getAllFeatureSpecs() {
    */
   public boolean isConnected() {
     ConnectivityState state = channel.getState(true);
-    return state.compareTo(ConnectivityState.READY) == 0;
+    return state == ConnectivityState.IDLE
+        || state == ConnectivityState.READY;
   }
 
   /**