chore: add serve response type (#180)

Signed-off-by: Yashash H L <[email protected]>

pkg/apis/proto/sink/v1/sink.proto

@@ -66,6 +66,7 @@ enum Status {
   SUCCESS = 0;
   FAILURE = 1;
   FALLBACK = 2;
+  SERVE = 3;
 }
 
 /**

pkg/sinker/service.go

@@ -223,6 +223,12 @@ func (fs *Service) processData(ctx context.Context, stream sinkpb.Sink_SinkFnSer
 				Status:        sinkpb.Status_SUCCESS,
 				ServeResponse: msg.ServeResponse,
 			})
+		} else if msg.Serve {
+			resultList = append(resultList, &sinkpb.SinkResponse_Result{
+				Id:            msg.ID,
+				Status:        sinkpb.Status_SERVE,
+				ServeResponse: msg.ServeResponse,
+			})
 		} else {
 			resultList = append(resultList, &sinkpb.SinkResponse_Result{
 				Id:     msg.ID,

pkg/sinker/types.go

@@ -9,7 +9,10 @@ type Response struct {
 	// Err represents the error message when "success" is false.
 	Err string `json:"err,omitempty"`
 	// Fallback is true if the message to be sent to the fallback sink.
-	Fallback      bool   `json:"fallback,omitempty"`
+	Fallback bool `json:"fallback,omitempty"`
+	// Serve is true if the message to be sent to serving store.
+	Serve bool `json:"serve,omitempty"`
+	// ServeResponse is the response that will be sent to the serving store.
 	ServeResponse []byte `json:"serve_reponse,omitempty"`
 }
 
@@ -50,5 +53,5 @@ func ResponseFallback(id string) Response {
 }
 
 func ResponseServe(id string, result []byte) Response {
-	return Response{ID: id, Success: true, ServeResponse: result}
+	return Response{ID: id, Serve: true, ServeResponse: result}
 }