Adds flag for flow job entry removal (#608)

PeerDB-io · Nov 2, 2023 · b2c9686 · b2c9686
1 parent 0aec792
commit b2c9686
Show file tree

Hide file tree

Showing 10 changed files with 600 additions and 555 deletions.
diff --git a/flow/cmd/handler.go b/flow/cmd/handler.go
@@ -340,12 +340,14 @@ func (h *FlowRequestHandler) ShutdownFlow(
 		}
 	}
 
-	delErr := h.removeFlowEntryInCatalog(req.FlowJobName)
-	if delErr != nil {
-		return &protos.ShutdownResponse{
-			Ok:           false,
-			ErrorMessage: err.Error(),
-		}, err
+	if req.RemoveFlowEntry {
+		delErr := h.removeFlowEntryInCatalog(req.FlowJobName)
+		if delErr != nil {
+			return &protos.ShutdownResponse{
+				Ok:           false,
+				ErrorMessage: err.Error(),
+			}, err
+		}
 	}
 
 	return &protos.ShutdownResponse{

diff --git a/flow/generated/protos/route.pb.go b/flow/generated/protos/route.pb.go
diff --git a/nexus/flow-rs/src/grpc.rs b/nexus/flow-rs/src/grpc.rs
@@ -129,6 +129,7 @@ impl FlowGrpcClient {
             workflow_id: workflow_details.workflow_id,
             source_peer: Some(workflow_details.source_peer),
             destination_peer: Some(workflow_details.destination_peer),
+            remove_flow_entry:false
         };
         let response = self.client.shutdown_flow(shutdown_flow_req).await?;
         let shutdown_response = response.into_inner();