tasks.go 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  1. package cluster
  2. import (
  3. apitypes "github.com/docker/docker/api/types"
  4. "github.com/docker/docker/api/types/filters"
  5. types "github.com/docker/docker/api/types/swarm"
  6. "github.com/docker/docker/daemon/cluster/convert"
  7. swarmapi "github.com/docker/swarmkit/api"
  8. "golang.org/x/net/context"
  9. )
  10. // GetTasks returns a list of tasks matching the filter options.
  11. func (c *Cluster) GetTasks(options apitypes.TaskListOptions) ([]types.Task, error) {
  12. c.mu.RLock()
  13. defer c.mu.RUnlock()
  14. state := c.currentNodeState()
  15. if !state.IsActiveManager() {
  16. return nil, c.errNoManager(state)
  17. }
  18. byName := func(filter filters.Args) error {
  19. if filter.Include("service") {
  20. serviceFilters := filter.Get("service")
  21. for _, serviceFilter := range serviceFilters {
  22. service, err := c.GetService(serviceFilter, false)
  23. if err != nil {
  24. return err
  25. }
  26. filter.Del("service", serviceFilter)
  27. filter.Add("service", service.ID)
  28. }
  29. }
  30. if filter.Include("node") {
  31. nodeFilters := filter.Get("node")
  32. for _, nodeFilter := range nodeFilters {
  33. node, err := c.GetNode(nodeFilter)
  34. if err != nil {
  35. return err
  36. }
  37. filter.Del("node", nodeFilter)
  38. filter.Add("node", node.ID)
  39. }
  40. }
  41. return nil
  42. }
  43. filters, err := newListTasksFilters(options.Filters, byName)
  44. if err != nil {
  45. return nil, err
  46. }
  47. ctx, cancel := c.getRequestContext()
  48. defer cancel()
  49. r, err := state.controlClient.ListTasks(
  50. ctx,
  51. &swarmapi.ListTasksRequest{Filters: filters})
  52. if err != nil {
  53. return nil, err
  54. }
  55. tasks := make([]types.Task, 0, len(r.Tasks))
  56. for _, task := range r.Tasks {
  57. if task.Spec.GetContainer() != nil {
  58. tasks = append(tasks, convert.TaskFromGRPC(*task))
  59. }
  60. }
  61. return tasks, nil
  62. }
  63. // GetTask returns a task by an ID.
  64. func (c *Cluster) GetTask(input string) (types.Task, error) {
  65. var task *swarmapi.Task
  66. if err := c.lockedManagerAction(func(ctx context.Context, state nodeState) error {
  67. t, err := getTask(ctx, state.controlClient, input)
  68. if err != nil {
  69. return err
  70. }
  71. task = t
  72. return nil
  73. }); err != nil {
  74. return types.Task{}, err
  75. }
  76. return convert.TaskFromGRPC(*task), nil
  77. }