ps.go 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. // +build experimental
  2. package stack
  3. import (
  4. "fmt"
  5. "golang.org/x/net/context"
  6. "github.com/docker/docker/api/client"
  7. "github.com/docker/docker/api/client/idresolver"
  8. "github.com/docker/docker/api/client/task"
  9. "github.com/docker/docker/cli"
  10. "github.com/docker/docker/opts"
  11. "github.com/docker/engine-api/types"
  12. "github.com/docker/engine-api/types/swarm"
  13. "github.com/spf13/cobra"
  14. )
  15. type psOptions struct {
  16. all bool
  17. filter opts.FilterOpt
  18. noTrunc bool
  19. namespace string
  20. noResolve bool
  21. }
  22. func newPsCommand(dockerCli *client.DockerCli) *cobra.Command {
  23. opts := psOptions{filter: opts.NewFilterOpt()}
  24. cmd := &cobra.Command{
  25. Use: "ps [OPTIONS] STACK",
  26. Short: "List the tasks in the stack",
  27. Args: cli.ExactArgs(1),
  28. RunE: func(cmd *cobra.Command, args []string) error {
  29. opts.namespace = args[0]
  30. return runPS(dockerCli, opts)
  31. },
  32. }
  33. flags := cmd.Flags()
  34. flags.BoolVarP(&opts.all, "all", "a", false, "Display all tasks")
  35. flags.BoolVar(&opts.noTrunc, "no-trunc", false, "Do not truncate output")
  36. flags.BoolVar(&opts.noResolve, "no-resolve", false, "Do not map IDs to Names")
  37. flags.VarP(&opts.filter, "filter", "f", "Filter output based on conditions provided")
  38. return cmd
  39. }
  40. func runPS(dockerCli *client.DockerCli, opts psOptions) error {
  41. namespace := opts.namespace
  42. client := dockerCli.Client()
  43. ctx := context.Background()
  44. filter := opts.filter.Value()
  45. filter.Add("label", labelNamespace+"="+opts.namespace)
  46. if !opts.all && !filter.Include("desired-state") {
  47. filter.Add("desired-state", string(swarm.TaskStateRunning))
  48. filter.Add("desired-state", string(swarm.TaskStateAccepted))
  49. }
  50. tasks, err := client.TaskList(ctx, types.TaskListOptions{Filter: filter})
  51. if err != nil {
  52. return err
  53. }
  54. if len(tasks) == 0 {
  55. fmt.Fprintf(dockerCli.Out(), "Nothing found in stack: %s\n", namespace)
  56. return nil
  57. }
  58. return task.Print(dockerCli, ctx, tasks, idresolver.New(client, opts.noResolve), opts.noTrunc)
  59. }