Merge pull request #1971 from djdongjin/unify-multi-error

Optimize id walkers and muti error handling

Author: AkihiroSuda

cmd/nerdctl/container_stats.go

@@ -277,13 +277,8 @@ func statsAction(cmd *cobra.Command, args []string) error {
 			},
 		}
 
-		for _, req := range args {
-			n, err := walker.Walk(ctx, req)
-			if err != nil {
-				return err
-			} else if n == 0 {
-				return fmt.Errorf("no such container %s", req)
-			}
+		if err := walker.WalkAll(ctx, args, false); err != nil {
+			return err
 		}
 
 		// make sure each container get at least one valid stat data

cmd/nerdctl/container_update.go

@@ -110,15 +110,8 @@ func updateAction(cmd *cobra.Command, args []string) error {
 			return err
 		},
 	}
-	for _, req := range args {
-		n, err := walker.Walk(ctx, req)
-		if err != nil {
-			return err
-		} else if n == 0 {
-			return fmt.Errorf("no such container %s", req)
-		}
-	}
-	return nil
+
+	return walker.WalkAll(ctx, args, true)
 }
 
 func getUpdateOption(cmd *cobra.Command, globalOptions types.GlobalCommandOptions) (updateResourceOptions, error) {

cmd/nerdctl/image_history.go

@@ -143,19 +143,7 @@ func historyAction(cmd *cobra.Command, args []string) error {
 		},
 	}
 
-	var errs []error
-	for _, req := range args {
-		n, err := walker.Walk(ctx, req)
-		if err != nil {
-			errs = append(errs, err)
-		} else if n == 0 {
-			errs = append(errs, fmt.Errorf("no such object: %s", req))
-		}
-	}
-	if len(errs) > 0 {
-		return fmt.Errorf("%d errors: %v", len(errs), errs)
-	}
-	return nil
+	return walker.WalkAll(ctx, args, true)
 }
 
 type historyPrinter struct {

pkg/cmd/container/inspect.go

@@ -27,6 +27,7 @@ import (
 	"github.com/containerd/nerdctl/pkg/formatter"
 	"github.com/containerd/nerdctl/pkg/idutil/containerwalker"
 	"github.com/containerd/nerdctl/pkg/inspecttypes/dockercompat"
+	"github.com/sirupsen/logrus"
 )
 
 // Inspect prints detailed information for each container in `containers`.
@@ -40,20 +41,13 @@ func Inspect(ctx context.Context, client *containerd.Client, containers []string
 		OnFound: f.Handler,
 	}
 
-	var errs []error
-	for _, req := range containers {
-		n, err := walker.Walk(ctx, req)
-		if err != nil {
-			errs = append(errs, err)
-		} else if n == 0 {
-			errs = append(errs, fmt.Errorf("no such container: %s", req))
+	err := walker.WalkAll(ctx, containers, true)
+	if len(f.entries) > 0 {
+		if formatErr := formatter.FormatSlice(options.Format, options.Stdout, f.entries); formatErr != nil {
+			logrus.Error(formatErr)
 		}
 	}
-	if len(errs) > 0 {
-		return fmt.Errorf("%d errors: %v", len(errs), errs)
-	}
-
-	return formatter.FormatSlice(options.Format, options.Stdout, f.entries)
+	return err
 }
 
 type containerInspector struct {

pkg/cmd/container/kill.go

@@ -60,15 +60,8 @@ func Kill(ctx context.Context, client *containerd.Client, reqs []string, options
 			return err
 		},
 	}
-	for _, req := range reqs {
-		n, err := walker.Walk(ctx, req)
-		if err != nil {
-			return err
-		} else if n == 0 {
-			return fmt.Errorf("no such container %s", req)
-		}
-	}
-	return nil
+
+	return walker.WalkAll(ctx, reqs, true)
 }
 
 func killContainer(ctx context.Context, container containerd.Container, signal syscall.Signal) error {

pkg/cmd/container/pause.go

@@ -18,9 +18,7 @@ package container
 
 import (
 	"context"
-	"errors"
 	"fmt"
-	"strings"
 
 	"github.com/containerd/containerd"
 	"github.com/containerd/nerdctl/pkg/api/types"
@@ -45,18 +43,5 @@ func Pause(ctx context.Context, client *containerd.Client, reqs []string, option
 		},
 	}
 
-	var errs []string
-	for _, req := range reqs {
-		n, err := walker.Walk(ctx, req)
-		if err != nil {
-			errs = append(errs, err.Error())
-		} else if n == 0 {
-			errs = append(errs, fmt.Sprintf("no such container %s", req))
-		}
-	}
-
-	if len(errs) > 0 {
-		return errors.New(strings.Join(errs, "\n"))
-	}
-	return nil
+	return walker.WalkAll(ctx, reqs, true)
 }

pkg/cmd/container/remove.go

@@ -77,20 +77,13 @@ func Remove(ctx context.Context, client *containerd.Client, containers []string,
 			return err
 		},
 	}
-	for _, req := range containers {
-		n, err := walker.Walk(ctx, req)
-		if err == nil && n == 0 {
-			err = fmt.Errorf("no such container %s", req)
-		}
-		if err != nil {
-			if options.Force {
-				logrus.Error(err)
-			} else {
-				return err
-			}
-		}
+
+	err := walker.WalkAll(ctx, containers, true)
+	if err != nil && options.Force {
+		logrus.Error(err)
+		return nil
 	}
-	return nil
+	return err
 }
 
 // RemoveContainer removes a container from containerd store.

pkg/cmd/container/restart.go

@@ -44,14 +44,6 @@ func Restart(ctx context.Context, client *containerd.Client, containers []string
 			return err
 		},
 	}
-	for _, req := range containers {
-		n, err := walker.Walk(ctx, req)
-		if err != nil {
-			return err
-		} else if n == 0 {
-			return fmt.Errorf("no such container %s", req)
-		}
-	}
 
-	return nil
+	return walker.WalkAll(ctx, containers, true)
 }

pkg/cmd/container/start.go

@@ -52,13 +52,5 @@ func Start(ctx context.Context, client *containerd.Client, reqs []string, option
 		},
 	}
 
-	for _, req := range reqs {
-		n, err := walker.Walk(ctx, req)
-		if err != nil {
-			return err
-		} else if n == 0 {
-			return fmt.Errorf("no such container %s", req)
-		}
-	}
-	return nil
+	return walker.WalkAll(ctx, reqs, true)
 }

pkg/cmd/container/stop.go

@@ -27,6 +27,7 @@ import (
 	"github.com/containerd/nerdctl/pkg/idutil/containerwalker"
 )
 
+// Stop stops a list of containers specified by `reqs`.
 func Stop(ctx context.Context, client *containerd.Client, reqs []string, opt types.ContainerStopOptions) error {
 	walker := &containerwalker.ContainerWalker{
 		Client: client,
@@ -45,13 +46,6 @@ func Stop(ctx context.Context, client *containerd.Client, reqs []string, opt typ
 			return err
 		},
 	}
-	for _, req := range reqs {
-		n, err := walker.Walk(ctx, req)
-		if err != nil {
-			return err
-		} else if n == 0 {
-			return fmt.Errorf("no such container %s", req)
-		}
-	}
-	return nil
+
+	return walker.WalkAll(ctx, reqs, true)
 }