chore: add IsReady of MapStream (numaproj#147)

Signed-off-by: Yashash H L <[email protected]>
KeranYang · Jan 22, 2025 · f725281 · f725281
1 parent 4d4c9ed
commit f725281
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/main/java/io/numaproj/numaflow/mapstreamer/Service.java b/src/main/java/io/numaproj/numaflow/mapstreamer/Service.java
@@ -1,5 +1,6 @@
 package io.numaproj.numaflow.mapstreamer;
 
+import com.google.protobuf.Empty;
 import io.grpc.Status;
 import io.grpc.stub.StreamObserver;
 import io.numaproj.numaflow.map.v1.MapGrpc;
@@ -89,6 +90,17 @@ public void onCompleted() {
         };
     }
 
+    /**
+     * IsReady is the heartbeat endpoint for gRPC.
+     */
+    @Override
+    public void isReady(
+            Empty request,
+            StreamObserver<MapOuterClass.ReadyResponse> responseObserver) {
+        responseObserver.onNext(MapOuterClass.ReadyResponse.newBuilder().setReady(true).build());
+        responseObserver.onCompleted();
+    }
+
     // Construct a HandlerDatum from a MapRequest
     private HandlerDatum constructHandlerDatum(MapOuterClass.MapRequest d) {
         return new HandlerDatum(