exec.go

package cmd

import (
	"context"
	"fmt"

	"github.com/alexei-led/pumba/pkg/chaos"
	"github.com/alexei-led/pumba/pkg/chaos/docker"
	"github.com/pkg/errors"
	"github.com/urfave/cli"
)

type execContext struct {
	context context.Context
}

// NewExecCLICommand initialize CLI exec command and bind it to the execContext
func NewExecCLICommand(ctx context.Context) *cli.Command {
	cmdContext := &execContext{context: ctx}
	return &cli.Command{
		Name: "exec",
		Flags: []cli.Flag{
			cli.StringFlag{
				Name:  "command, s",
				Usage: "shell command, that will be sent by Pumba to the target container(s)",
				Value: "kill 1",
			},
			cli.StringSliceFlag{
				Name:  "args, a",
				Usage: "additional arguments for the command",
			},
			cli.IntFlag{
				Name:  "limit, l",
				Usage: "limit number of container to exec (0: exec all matching)",
				Value: 0,
			},
		},
		Usage:       "exec specified containers",
		ArgsUsage:   fmt.Sprintf("containers (name, list of names, or RE2 regex if prefixed with %q)", chaos.Re2Prefix),
		Description: "send command to target container(s)",
		Action:      cmdContext.exec,
	}
}

// EXEC Command
func (cmd *execContext) exec(c *cli.Context) error {
	// parse common chaos flags
	params, err := chaos.ParseGlobalParams(c)
	if err != nil {
		return errors.Wrap(err, "error parsing global parameters")
	}
	// get command
	command := c.String("command")
	// get args
	args := c.StringSlice("args")
	// get limit for number of containers to exec
	limit := c.Int("limit")
	// init exec command
	execCommand := docker.NewExecCommand(chaos.DockerClient, params, command, args, limit)
	// run exec command
	err = chaos.RunChaosCommand(cmd.context, execCommand, params)
	if err != nil {
		return errors.Wrap(err, "could not run exec command")
	}
	return nil
}