github.com/justincormack/cli@v0.0.0-20201215022714-831ebeae9675/cli/command/container/wait.go (about)

     1  package container
     2  
     3  import (
     4  	"context"
     5  	"fmt"
     6  	"strings"
     7  
     8  	"github.com/docker/cli/cli"
     9  	"github.com/docker/cli/cli/command"
    10  	"github.com/pkg/errors"
    11  	"github.com/spf13/cobra"
    12  )
    13  
    14  type waitOptions struct {
    15  	containers []string
    16  }
    17  
    18  // NewWaitCommand creates a new cobra.Command for `docker wait`
    19  func NewWaitCommand(dockerCli command.Cli) *cobra.Command {
    20  	var opts waitOptions
    21  
    22  	cmd := &cobra.Command{
    23  		Use:   "wait CONTAINER [CONTAINER...]",
    24  		Short: "Block until one or more containers stop, then print their exit codes",
    25  		Args:  cli.RequiresMinArgs(1),
    26  		RunE: func(cmd *cobra.Command, args []string) error {
    27  			opts.containers = args
    28  			return runWait(dockerCli, &opts)
    29  		},
    30  	}
    31  
    32  	return cmd
    33  }
    34  
    35  func runWait(dockerCli command.Cli, opts *waitOptions) error {
    36  	ctx := context.Background()
    37  
    38  	var errs []string
    39  	for _, container := range opts.containers {
    40  		resultC, errC := dockerCli.Client().ContainerWait(ctx, container, "")
    41  
    42  		select {
    43  		case result := <-resultC:
    44  			fmt.Fprintf(dockerCli.Out(), "%d\n", result.StatusCode)
    45  		case err := <-errC:
    46  			errs = append(errs, err.Error())
    47  		}
    48  	}
    49  	if len(errs) > 0 {
    50  		return errors.New(strings.Join(errs, "\n"))
    51  	}
    52  	return nil
    53  }