134 lines
4.4 KiB
Go
134 lines
4.4 KiB
Go
package container
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/docker/cli/cli"
|
|
"github.com/docker/cli/cli/command"
|
|
"github.com/docker/cli/opts"
|
|
containertypes "github.com/docker/docker/api/types/container"
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/cobra"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
type updateOptions struct {
|
|
blkioWeight uint16
|
|
cpuPeriod int64
|
|
cpuQuota int64
|
|
cpuRealtimePeriod int64
|
|
cpuRealtimeRuntime int64
|
|
cpusetCpus string
|
|
cpusetMems string
|
|
cpuShares int64
|
|
memory opts.MemBytes
|
|
memoryReservation opts.MemBytes
|
|
memorySwap opts.MemSwapBytes
|
|
kernelMemory opts.MemBytes
|
|
restartPolicy string
|
|
cpus opts.NanoCPUs
|
|
|
|
nFlag int
|
|
|
|
containers []string
|
|
}
|
|
|
|
// NewUpdateCommand creates a new cobra.Command for `docker update`
|
|
func NewUpdateCommand(dockerCli command.Cli) *cobra.Command {
|
|
var options updateOptions
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "update [OPTIONS] CONTAINER [CONTAINER...]",
|
|
Short: "Update configuration of one or more containers",
|
|
Args: cli.RequiresMinArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
options.containers = args
|
|
options.nFlag = cmd.Flags().NFlag()
|
|
return runUpdate(dockerCli, &options)
|
|
},
|
|
}
|
|
|
|
flags := cmd.Flags()
|
|
flags.Uint16Var(&options.blkioWeight, "blkio-weight", 0, "Block IO (relative weight), between 10 and 1000, or 0 to disable (default 0)")
|
|
flags.Int64Var(&options.cpuPeriod, "cpu-period", 0, "Limit CPU CFS (Completely Fair Scheduler) period")
|
|
flags.Int64Var(&options.cpuQuota, "cpu-quota", 0, "Limit CPU CFS (Completely Fair Scheduler) quota")
|
|
flags.Int64Var(&options.cpuRealtimePeriod, "cpu-rt-period", 0, "Limit the CPU real-time period in microseconds")
|
|
flags.SetAnnotation("cpu-rt-period", "version", []string{"1.25"})
|
|
flags.Int64Var(&options.cpuRealtimeRuntime, "cpu-rt-runtime", 0, "Limit the CPU real-time runtime in microseconds")
|
|
flags.SetAnnotation("cpu-rt-runtime", "version", []string{"1.25"})
|
|
flags.StringVar(&options.cpusetCpus, "cpuset-cpus", "", "CPUs in which to allow execution (0-3, 0,1)")
|
|
flags.StringVar(&options.cpusetMems, "cpuset-mems", "", "MEMs in which to allow execution (0-3, 0,1)")
|
|
flags.Int64VarP(&options.cpuShares, "cpu-shares", "c", 0, "CPU shares (relative weight)")
|
|
flags.VarP(&options.memory, "memory", "m", "Memory limit")
|
|
flags.Var(&options.memoryReservation, "memory-reservation", "Memory soft limit")
|
|
flags.Var(&options.memorySwap, "memory-swap", "Swap limit equal to memory plus swap: '-1' to enable unlimited swap")
|
|
flags.Var(&options.kernelMemory, "kernel-memory", "Kernel memory limit")
|
|
flags.StringVar(&options.restartPolicy, "restart", "", "Restart policy to apply when a container exits")
|
|
|
|
flags.Var(&options.cpus, "cpus", "Number of CPUs")
|
|
flags.SetAnnotation("cpus", "version", []string{"1.29"})
|
|
|
|
return cmd
|
|
}
|
|
|
|
func runUpdate(dockerCli command.Cli, options *updateOptions) error {
|
|
var err error
|
|
|
|
if options.nFlag == 0 {
|
|
return errors.New("you must provide one or more flags when using this command")
|
|
}
|
|
|
|
var restartPolicy containertypes.RestartPolicy
|
|
if options.restartPolicy != "" {
|
|
restartPolicy, err = opts.ParseRestartPolicy(options.restartPolicy)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
resources := containertypes.Resources{
|
|
BlkioWeight: options.blkioWeight,
|
|
CpusetCpus: options.cpusetCpus,
|
|
CpusetMems: options.cpusetMems,
|
|
CPUShares: options.cpuShares,
|
|
Memory: options.memory.Value(),
|
|
MemoryReservation: options.memoryReservation.Value(),
|
|
MemorySwap: options.memorySwap.Value(),
|
|
KernelMemory: options.kernelMemory.Value(),
|
|
CPUPeriod: options.cpuPeriod,
|
|
CPUQuota: options.cpuQuota,
|
|
CPURealtimePeriod: options.cpuRealtimePeriod,
|
|
CPURealtimeRuntime: options.cpuRealtimeRuntime,
|
|
NanoCPUs: options.cpus.Value(),
|
|
}
|
|
|
|
updateConfig := containertypes.UpdateConfig{
|
|
Resources: resources,
|
|
RestartPolicy: restartPolicy,
|
|
}
|
|
|
|
ctx := context.Background()
|
|
|
|
var (
|
|
warns []string
|
|
errs []string
|
|
)
|
|
for _, container := range options.containers {
|
|
r, err := dockerCli.Client().ContainerUpdate(ctx, container, updateConfig)
|
|
if err != nil {
|
|
errs = append(errs, err.Error())
|
|
} else {
|
|
fmt.Fprintln(dockerCli.Out(), container)
|
|
}
|
|
warns = append(warns, r.Warnings...)
|
|
}
|
|
if len(warns) > 0 {
|
|
fmt.Fprintln(dockerCli.Out(), strings.Join(warns, "\n"))
|
|
}
|
|
if len(errs) > 0 {
|
|
return errors.New(strings.Join(errs, "\n"))
|
|
}
|
|
return nil
|
|
}
|