cmd.go 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. package node
  2. import (
  3. "fmt"
  4. "golang.org/x/net/context"
  5. "github.com/spf13/cobra"
  6. "github.com/docker/docker/api/client"
  7. "github.com/docker/docker/cli"
  8. apiclient "github.com/docker/engine-api/client"
  9. )
  10. // NewNodeCommand returns a cobra command for `node` subcommands
  11. func NewNodeCommand(dockerCli *client.DockerCli) *cobra.Command {
  12. cmd := &cobra.Command{
  13. Use: "node",
  14. Short: "Manage Docker Swarm nodes",
  15. Args: cli.NoArgs,
  16. Run: func(cmd *cobra.Command, args []string) {
  17. fmt.Fprintf(dockerCli.Err(), "\n"+cmd.UsageString())
  18. },
  19. }
  20. cmd.AddCommand(
  21. newAcceptCommand(dockerCli),
  22. newDemoteCommand(dockerCli),
  23. newInspectCommand(dockerCli),
  24. newListCommand(dockerCli),
  25. newPromoteCommand(dockerCli),
  26. newRemoveCommand(dockerCli),
  27. newTasksCommand(dockerCli),
  28. newUpdateCommand(dockerCli),
  29. )
  30. return cmd
  31. }
  32. func nodeReference(client apiclient.APIClient, ctx context.Context, ref string) (string, error) {
  33. // The special value "self" for a node reference is mapped to the current
  34. // node, hence the node ID is retrieved using the `/info` endpoint.
  35. if ref == "self" {
  36. info, err := client.Info(ctx)
  37. if err != nil {
  38. return "", err
  39. }
  40. return info.Swarm.NodeID, nil
  41. }
  42. return ref, nil
  43. }