mirror of
https://github.com/podman-container-tools/podman.git
synced 2026-08-28 19:27:58 +00:00
52371057cd
in cases where commands require input and we dont provide it, we often would segv. This can be attributed in many cases to the subcommand not picked up the cobra Args attribute or neither had them. Signed-off-by: baude <bbaude@redhat.com>
103 lines
2.7 KiB
Go
103 lines
2.7 KiB
Go
package containers
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/containers/libpod/cmd/podman/registry"
|
|
"github.com/containers/libpod/cmd/podman/utils"
|
|
"github.com/containers/libpod/cmd/podman/validate"
|
|
"github.com/containers/libpod/libpod/define"
|
|
"github.com/containers/libpod/pkg/domain/entities"
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/pflag"
|
|
)
|
|
|
|
var (
|
|
waitDescription = `Block until one or more containers stop and then print their exit codes.
|
|
`
|
|
waitCommand = &cobra.Command{
|
|
Use: "wait [flags] CONTAINER [CONTAINER...]",
|
|
Short: "Block on one or more containers",
|
|
Long: waitDescription,
|
|
RunE: wait,
|
|
Args: validate.IdOrLatestArgs,
|
|
Example: `podman wait --latest
|
|
podman wait --interval 5000 ctrID
|
|
podman wait ctrID1 ctrID2`,
|
|
}
|
|
|
|
containerWaitCommand = &cobra.Command{
|
|
Use: waitCommand.Use,
|
|
Short: waitCommand.Short,
|
|
Long: waitCommand.Long,
|
|
RunE: waitCommand.RunE,
|
|
Args: validate.IdOrLatestArgs,
|
|
Example: `podman container wait --latest
|
|
podman container wait --interval 5000 ctrID
|
|
podman container wait ctrID1 ctrID2`,
|
|
}
|
|
)
|
|
|
|
var (
|
|
waitOptions = entities.WaitOptions{}
|
|
waitCondition string
|
|
)
|
|
|
|
func waitFlags(flags *pflag.FlagSet) {
|
|
flags.DurationVarP(&waitOptions.Interval, "interval", "i", time.Duration(250), "Milliseconds to wait before polling for completion")
|
|
flags.BoolVarP(&waitOptions.Latest, "latest", "l", false, "Act on the latest container podman is aware of")
|
|
flags.StringVar(&waitCondition, "condition", "stopped", "Condition to wait on")
|
|
if registry.IsRemote() {
|
|
// TODO: This is the same as V1. We could skip creating the flag altogether in V2...
|
|
_ = flags.MarkHidden("latest")
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
registry.Commands = append(registry.Commands, registry.CliCommand{
|
|
Mode: []entities.EngineMode{entities.ABIMode, entities.TunnelMode},
|
|
Command: waitCommand,
|
|
})
|
|
flags := waitCommand.Flags()
|
|
waitFlags(flags)
|
|
|
|
registry.Commands = append(registry.Commands, registry.CliCommand{
|
|
Mode: []entities.EngineMode{entities.ABIMode, entities.TunnelMode},
|
|
Command: containerWaitCommand,
|
|
Parent: containerCmd,
|
|
})
|
|
|
|
containerWaitFlags := containerWaitCommand.Flags()
|
|
waitFlags(containerWaitFlags)
|
|
}
|
|
|
|
func wait(cmd *cobra.Command, args []string) error {
|
|
var (
|
|
err error
|
|
errs utils.OutputErrors
|
|
)
|
|
if waitOptions.Interval == 0 {
|
|
return errors.New("interval must be greater then 0")
|
|
}
|
|
|
|
waitOptions.Condition, err = define.StringToContainerStatus(waitCondition)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
responses, err := registry.ContainerEngine().ContainerWait(context.Background(), args, waitOptions)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, r := range responses {
|
|
if r.Error == nil {
|
|
fmt.Println(r.ExitCode)
|
|
} else {
|
|
errs = append(errs, r.Error)
|
|
}
|
|
}
|
|
return errs.PrintErrors()
|
|
}
|