refactor: enhance stop computation

Signed-off-by: WashingtonKK <[email protected]>
WashingtonKK · Aug 22, 2024 · a558da2 · a558da2
1 parent f819902
commit a558da2
Show file tree

Hide file tree

Showing 3 changed files with 139 additions and 163 deletions.
diff --git a/manager/api/grpc/client.go b/manager/api/grpc/client.go
@@ -77,22 +77,8 @@ func (client ManagerClient) Process(ctx context.Context, cancel context.CancelFu
 					ComputationId: mes.StopComputation.ComputationId,
 				}}
 				if err := client.svc.Stop(ctx, mes.StopComputation.ComputationId); err != nil {
-					if errors.Contains(err, manager.ErrNotFound) {
-						msg.StopComputationRes.Message = err.Error()
-						msg.StopComputationRes.Stopped = false
-						if err := client.stream.Send(&pkgmanager.ClientStreamMessage{Message: msg}); err != nil {
-							return err
-						}
-						return nil
-					}
 					msg.StopComputationRes.Message = err.Error()
-					msg.StopComputationRes.Stopped = false
-					if err := client.stream.Send(&pkgmanager.ClientStreamMessage{Message: msg}); err != nil {
-						return err
-					}
-					return err
 				}
-				msg.StopComputationRes.Stopped = true
 				if err := client.stream.Send(&pkgmanager.ClientStreamMessage{Message: msg}); err != nil {
 					return err
 				}

diff --git a/manager/manager.proto b/manager/manager.proto
@@ -23,8 +23,7 @@ message StopComputation {
 
 message StopComputationResponse {
   string computation_id = 1;
-  bool stopped = 2;
-  string message = 3;
+  string message = 2;
 }
 
 message RunResponse{