update.go 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. package swarm
  2. import (
  3. "fmt"
  4. "golang.org/x/net/context"
  5. "github.com/docker/docker/api/types/swarm"
  6. "github.com/docker/docker/cli"
  7. "github.com/docker/docker/cli/command"
  8. "github.com/spf13/cobra"
  9. "github.com/spf13/pflag"
  10. )
  11. func newUpdateCommand(dockerCli *command.DockerCli) *cobra.Command {
  12. opts := swarmOptions{}
  13. cmd := &cobra.Command{
  14. Use: "update [OPTIONS]",
  15. Short: "Update the swarm",
  16. Args: cli.NoArgs,
  17. RunE: func(cmd *cobra.Command, args []string) error {
  18. return runUpdate(dockerCli, cmd.Flags(), opts)
  19. },
  20. }
  21. addSwarmFlags(cmd.Flags(), &opts)
  22. return cmd
  23. }
  24. func runUpdate(dockerCli *command.DockerCli, flags *pflag.FlagSet, opts swarmOptions) error {
  25. client := dockerCli.Client()
  26. ctx := context.Background()
  27. var updateFlags swarm.UpdateFlags
  28. swarm, err := client.SwarmInspect(ctx)
  29. if err != nil {
  30. return err
  31. }
  32. err = mergeSwarm(&swarm, flags)
  33. if err != nil {
  34. return err
  35. }
  36. err = client.SwarmUpdate(ctx, swarm.Version, swarm.Spec, updateFlags)
  37. if err != nil {
  38. return err
  39. }
  40. fmt.Fprintln(dockerCli.Out(), "Swarm updated.")
  41. return nil
  42. }
  43. func mergeSwarm(swarm *swarm.Swarm, flags *pflag.FlagSet) error {
  44. spec := &swarm.Spec
  45. if flags.Changed(flagTaskHistoryLimit) {
  46. taskHistoryRetentionLimit, _ := flags.GetInt64(flagTaskHistoryLimit)
  47. spec.Orchestration.TaskHistoryRetentionLimit = &taskHistoryRetentionLimit
  48. }
  49. if flags.Changed(flagDispatcherHeartbeat) {
  50. if v, err := flags.GetDuration(flagDispatcherHeartbeat); err == nil {
  51. spec.Dispatcher.HeartbeatPeriod = v
  52. }
  53. }
  54. if flags.Changed(flagCertExpiry) {
  55. if v, err := flags.GetDuration(flagCertExpiry); err == nil {
  56. spec.CAConfig.NodeCertExpiry = v
  57. }
  58. }
  59. if flags.Changed(flagExternalCA) {
  60. value := flags.Lookup(flagExternalCA).Value.(*ExternalCAOption)
  61. spec.CAConfig.ExternalCAs = value.Value()
  62. }
  63. return nil
  64. }