117 lines
2.6 KiB
Go
117 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/projectatomic/libpod/cmd/podman/libpodruntime"
|
|
"github.com/projectatomic/libpod/libpod"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
var (
|
|
restartFlags = []cli.Flag{
|
|
cli.UintFlag{
|
|
Name: "timeout, time, t",
|
|
Usage: "Seconds to wait for stop before killing the container",
|
|
Value: libpod.CtrRemoveTimeout,
|
|
},
|
|
LatestFlag,
|
|
}
|
|
restartDescription = `Restarts one or more running containers. The container ID or name can be used. A timeout before forcibly stopping can be set, but defaults to 10 seconds`
|
|
|
|
restartCommand = cli.Command{
|
|
Name: "restart",
|
|
Usage: "Restart one or more containers",
|
|
Description: restartDescription,
|
|
Flags: restartFlags,
|
|
Action: restartCmd,
|
|
ArgsUsage: "CONTAINER [CONTAINER ...]",
|
|
UseShortOptionHandling: true,
|
|
}
|
|
)
|
|
|
|
func restartCmd(c *cli.Context) error {
|
|
args := c.Args()
|
|
if len(args) < 1 && !c.Bool("latest") {
|
|
return errors.Wrapf(libpod.ErrInvalidArg, "you must provide at least one container name or ID")
|
|
}
|
|
|
|
if err := validateFlags(c, restartFlags); err != nil {
|
|
return err
|
|
}
|
|
|
|
runtime, err := libpodruntime.GetRuntime(c)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "error creating libpod runtime")
|
|
}
|
|
defer runtime.Shutdown(false)
|
|
|
|
var lastError error
|
|
|
|
timeout := c.Uint("timeout")
|
|
useTimeout := c.IsSet("timeout")
|
|
|
|
// Handle --latest
|
|
if c.Bool("latest") {
|
|
lastCtr, err := runtime.GetLatestContainer()
|
|
if err != nil {
|
|
lastError = errors.Wrapf(err, "unable to get latest container")
|
|
} else {
|
|
ctrTimeout := lastCtr.StopTimeout()
|
|
if useTimeout {
|
|
ctrTimeout = timeout
|
|
}
|
|
|
|
lastError = restartCtr(ctrTimeout, lastCtr)
|
|
}
|
|
}
|
|
|
|
for _, id := range args {
|
|
ctr, err := runtime.LookupContainer(id)
|
|
if err != nil {
|
|
if lastError != nil {
|
|
fmt.Fprintln(os.Stderr, lastError)
|
|
}
|
|
lastError = errors.Wrapf(err, "unable to find container %s", id)
|
|
continue
|
|
}
|
|
|
|
ctrTimeout := ctr.StopTimeout()
|
|
if useTimeout {
|
|
ctrTimeout = timeout
|
|
}
|
|
|
|
if err := restartCtr(ctrTimeout, ctr); err != nil {
|
|
if lastError != nil {
|
|
fmt.Fprintln(os.Stderr, lastError)
|
|
}
|
|
lastError = errors.Wrapf(err, "error restarting container %s", ctr.ID())
|
|
}
|
|
}
|
|
|
|
return lastError
|
|
}
|
|
|
|
// Restart a single container
|
|
func restartCtr(timeout uint, ctr *libpod.Container) error {
|
|
state, err := ctr.State()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if state == libpod.ContainerStateRunning {
|
|
if err := ctr.StopWithTimeout(timeout); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if err := ctr.Start(getContext()); err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("%s\n", ctr.ID())
|
|
|
|
return nil
|
|
}
|