pr feedback

Author: aphralG

internal/command/command_service.go

@@ -560,11 +560,6 @@ func (cs *CommandService) connectCallback(
 
 		validatedError := grpc.ValidateGrpcError(connectErr)
 		if validatedError != nil {
-			codeError, ok := status.FromError(validatedError)
-			if ok && codeError.Message() == "grpc: the client connection is closing" {
-				return nil, backoff.Permanent(fmt.Errorf("failed to create connection: %w, "+
-					"stopping retry", validatedError))
-			}
 			slog.ErrorContext(ctx, "Failed to create connection", "error", validatedError)
 
 			return nil, validatedError

internal/grpc/grpc.go

@@ -353,7 +353,8 @@ func ProtoValidatorStreamClientInterceptor() (grpc.StreamClientInterceptor, erro
 func ValidateGrpcError(err error) error {
 	if err != nil {
 		if statusError, ok := status.FromError(err); ok {
-			if statusError.Code() == codes.InvalidArgument || statusError.Code() == codes.Unimplemented {
+			if statusError.Code() == codes.InvalidArgument || statusError.Code() == codes.Unimplemented ||
+				statusError.Code() == codes.Canceled {
 				return backoff.Permanent(err)
 			}
 		}