ctr/commands/images/push: don't show progress if it is debug mode

If user sets debug mode, the command push should only show the debug log
information. If the stdout is with flush by the progress status, it is
hard to see the debug log.

Signed-off-by: Wei Fu <fuweid89@gmail.com>

cmd/ctr/commands/images/push.go

@@ -63,16 +63,19 @@ var pushCommand = cli.Command{
 		var (
 			ref   = context.Args().First()
 			local = context.Args().Get(1)
+			debug = context.GlobalBool("debug")
 			desc  ocispec.Descriptor
 		)
 		if ref == "" {
 			return errors.New("please provide a remote image reference to push")
 		}
+
 		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
 		defer cancel()
+
 		if manifest := context.String("manifest"); manifest != "" {
 			desc.Digest, err = digest.Parse(manifest)
 			if err != nil {
@@ -98,7 +101,12 @@ var pushCommand = cli.Command{
 
 		eg, ctx := errgroup.WithContext(ctx)
 
+		// used to notify the progress writer
+		doneCh := make(chan struct{})
+
 		eg.Go(func() error {
+			defer close(doneCh)
+
 			log.G(ctx).WithField("image", ref).WithField("digest", desc.Digest).Debug("pushing")
 
 			jobHandler := images.HandlerFunc(func(ctx gocontext.Context, desc ocispec.Descriptor) ([]ocispec.Descriptor, error) {
@@ -112,18 +120,16 @@ var pushCommand = cli.Command{
 			)
 		})
 
-		errs := make(chan error)
+		// don't show progress if debug mode is set
-		go func() {
+		if !debug {
-			defer close(errs)
+			eg.Go(func() error {
-			errs <- eg.Wait()
-		}()
-
 				var (
 					ticker = time.NewTicker(100 * time.Millisecond)
 					fw     = progress.NewWriter(os.Stdout)
 					start  = time.Now()
 					done   bool
 				)
+
 				defer ticker.Stop()
 
 				for {
@@ -140,15 +146,15 @@ var pushCommand = cli.Command{
 							fw.Flush()
 							return nil
 						}
-			case err := <-errs:
+					case <-doneCh:
-				if err != nil {
-					return err
-				}
 						done = true
 					case <-ctx.Done():
 						done = true // allow ui to update once more
 					}
 				}
+			})
+		}
+		return eg.Wait()
 	},
 }