[image-builder-mk3] Avoid sending "Unknown" error code in all cases

Author: geropl

components/image-builder-mk3/pkg/orchestrator/orchestrator.go

@@ -260,7 +260,7 @@ func (o *Orchestrator) Build(req *protocol.BuildRequest, resp protocol.ImageBuil
 				BaseRef: req.BaseImageNameResolved,
 			})
 			if err != nil {
-				return err
+				return handleFailedBuildStreamResponse(err, "cannot send build response")
 			}
 			return nil
 		}
@@ -307,7 +307,7 @@ func (o *Orchestrator) Build(req *protocol.BuildRequest, resp protocol.ImageBuil
 			BaseRef: baseref,
 		})
 		if err != nil {
-			return err
+			return handleFailedBuildStreamResponse(err, "cannot send build response")
 		}
 		return nil
 	}
@@ -322,7 +322,7 @@ func (o *Orchestrator) Build(req *protocol.BuildRequest, resp protocol.ImageBuil
 
 	randomUUID, err := uuid.NewRandom()
 	if err != nil {
-		return
+		return status.Errorf(codes.Internal, "failed to generate build ID: %v", err)
 	}
 
 	var (
@@ -368,7 +368,7 @@ func (o *Orchestrator) Build(req *protocol.BuildRequest, resp protocol.ImageBuil
 
 	pbaseref, err := reference.ParseNormalizedNamed(baseref)
 	if err != nil {
-		return xerrors.Errorf("cannot parse baseref: %v", err)
+		return status.Errorf(codes.InvalidArgument, "cannot parse baseref: %v", err)
 	}
 	bobBaseref := "localhost:8080/base"
 	if r, ok := pbaseref.(reference.Digested); ok {
@@ -384,7 +384,7 @@ func (o *Orchestrator) Build(req *protocol.BuildRequest, resp protocol.ImageBuil
 		})
 		additionalAuth, err = json.Marshal(ath)
 		if err != nil {
-			return xerrors.Errorf("cannot marshal additional auth: %w", err)
+			return status.Errorf(codes.InvalidArgument, "cannot marshal additional auth: %v", err)
 		}
 	}
 
@@ -476,8 +476,8 @@ func (o *Orchestrator) Build(req *protocol.BuildRequest, resp protocol.ImageBuil
 
 		err := resp.Send(update)
 		if err != nil {
-			log.WithError(err).Error("cannot forward build update - dropping listener")
-			return status.Errorf(codes.Unknown, "cannot send update: %v", err)
+			log.WithError(err).Info("cannot forward build update - dropping listener")
+			return handleFailedBuildStreamResponse(err, "cannot send update")
 		}
 
 		if update.Status == protocol.BuildStatus_done_failure || update.Status == protocol.BuildStatus_done_success {
@@ -555,8 +555,8 @@ func (o *Orchestrator) Logs(req *protocol.LogsRequest, resp protocol.ImageBuilde
 
 		err := resp.Send(update)
 		if err != nil {
-			log.WithError(err).Error("cannot forward log output - dropping listener")
-			return status.Errorf(codes.Unknown, "cannot send log output: %v", err)
+			log.WithError(err).Info("cannot forward log output - dropping listener")
+			return handleFailedBuildStreamResponse(err, "cannot send log output")
 		}
 	}
 
@@ -709,6 +709,20 @@ func (o *Orchestrator) getWorkspaceImageRef(ctx context.Context, baseref string)
 	return fmt.Sprintf("%s:%x", o.Config.WorkspaceImageRepository, dst), nil
 }
 
+func handleFailedBuildStreamResponse(err error, msg string) error {
+	if status.Code(err) == codes.DeadlineExceeded {
+		// client disconnected before we could send the response - fine with us
+		return nil
+	}
+	log.WithError(err).WithField("code", status.Code(err)).Error(fmt.Sprintf("unexpected error while sending build response: %s", msg))
+
+	if _, ok := status.FromError(err); err != nil && ok {
+		return err
+	}
+
+	return status.Errorf(codes.Unavailable, "%s: %v", msg, err)
+}
+
 // parentCantCancelContext is a bit of a hack. We have some operations which we want to keep alive even after clients
 // disconnect. gRPC cancels the context once a client disconnects, thus we intercept the cancelation and act as if
 // nothing had happened.