mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
create a new file task.go and move task part codes from cluster.go into tasks.go
Signed-off-by: allencloud <allen.sun@daocloud.io>
This commit is contained in:
parent
6ff14b48c7
commit
6be675c752
2 changed files with 90 additions and 81 deletions
|
@ -784,87 +784,6 @@ func (c *Cluster) errNoManager(st nodeState) error {
|
||||||
return errors.New("This node is not a swarm manager. Worker nodes can't be used to view or modify cluster state. Please run this command on a manager node or promote the current node to a manager.")
|
return errors.New("This node is not a swarm manager. Worker nodes can't be used to view or modify cluster state. Please run this command on a manager node or promote the current node to a manager.")
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetTasks returns a list of tasks matching the filter options.
|
|
||||||
func (c *Cluster) GetTasks(options apitypes.TaskListOptions) ([]types.Task, error) {
|
|
||||||
c.mu.RLock()
|
|
||||||
defer c.mu.RUnlock()
|
|
||||||
|
|
||||||
state := c.currentNodeState()
|
|
||||||
if !state.IsActiveManager() {
|
|
||||||
return nil, c.errNoManager(state)
|
|
||||||
}
|
|
||||||
|
|
||||||
byName := func(filter filters.Args) error {
|
|
||||||
if filter.Include("service") {
|
|
||||||
serviceFilters := filter.Get("service")
|
|
||||||
for _, serviceFilter := range serviceFilters {
|
|
||||||
service, err := c.GetService(serviceFilter)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
filter.Del("service", serviceFilter)
|
|
||||||
filter.Add("service", service.ID)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if filter.Include("node") {
|
|
||||||
nodeFilters := filter.Get("node")
|
|
||||||
for _, nodeFilter := range nodeFilters {
|
|
||||||
node, err := c.GetNode(nodeFilter)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
filter.Del("node", nodeFilter)
|
|
||||||
filter.Add("node", node.ID)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
filters, err := newListTasksFilters(options.Filters, byName)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx, cancel := c.getRequestContext()
|
|
||||||
defer cancel()
|
|
||||||
|
|
||||||
r, err := state.controlClient.ListTasks(
|
|
||||||
ctx,
|
|
||||||
&swarmapi.ListTasksRequest{Filters: filters})
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
tasks := []types.Task{}
|
|
||||||
|
|
||||||
for _, task := range r.Tasks {
|
|
||||||
if task.Spec.GetContainer() != nil {
|
|
||||||
tasks = append(tasks, convert.TaskFromGRPC(*task))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return tasks, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetTask returns a task by an ID.
|
|
||||||
func (c *Cluster) GetTask(input string) (types.Task, error) {
|
|
||||||
c.mu.RLock()
|
|
||||||
defer c.mu.RUnlock()
|
|
||||||
|
|
||||||
state := c.currentNodeState()
|
|
||||||
if !state.IsActiveManager() {
|
|
||||||
return types.Task{}, c.errNoManager(state)
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx, cancel := c.getRequestContext()
|
|
||||||
defer cancel()
|
|
||||||
|
|
||||||
task, err := getTask(ctx, state.controlClient, input)
|
|
||||||
if err != nil {
|
|
||||||
return types.Task{}, err
|
|
||||||
}
|
|
||||||
return convert.TaskFromGRPC(*task), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetNetwork returns a cluster network by an ID.
|
// GetNetwork returns a cluster network by an ID.
|
||||||
func (c *Cluster) GetNetwork(input string) (apitypes.NetworkResource, error) {
|
func (c *Cluster) GetNetwork(input string) (apitypes.NetworkResource, error) {
|
||||||
c.mu.RLock()
|
c.mu.RLock()
|
||||||
|
|
90
daemon/cluster/tasks.go
Normal file
90
daemon/cluster/tasks.go
Normal file
|
@ -0,0 +1,90 @@
|
||||||
|
package cluster
|
||||||
|
|
||||||
|
import (
|
||||||
|
apitypes "github.com/docker/docker/api/types"
|
||||||
|
"github.com/docker/docker/api/types/filters"
|
||||||
|
types "github.com/docker/docker/api/types/swarm"
|
||||||
|
"github.com/docker/docker/daemon/cluster/convert"
|
||||||
|
swarmapi "github.com/docker/swarmkit/api"
|
||||||
|
)
|
||||||
|
|
||||||
|
// GetTasks returns a list of tasks matching the filter options.
|
||||||
|
func (c *Cluster) GetTasks(options apitypes.TaskListOptions) ([]types.Task, error) {
|
||||||
|
c.mu.RLock()
|
||||||
|
defer c.mu.RUnlock()
|
||||||
|
|
||||||
|
state := c.currentNodeState()
|
||||||
|
if !state.IsActiveManager() {
|
||||||
|
return nil, c.errNoManager(state)
|
||||||
|
}
|
||||||
|
|
||||||
|
byName := func(filter filters.Args) error {
|
||||||
|
if filter.Include("service") {
|
||||||
|
serviceFilters := filter.Get("service")
|
||||||
|
for _, serviceFilter := range serviceFilters {
|
||||||
|
service, err := c.GetService(serviceFilter)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
filter.Del("service", serviceFilter)
|
||||||
|
filter.Add("service", service.ID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if filter.Include("node") {
|
||||||
|
nodeFilters := filter.Get("node")
|
||||||
|
for _, nodeFilter := range nodeFilters {
|
||||||
|
node, err := c.GetNode(nodeFilter)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
filter.Del("node", nodeFilter)
|
||||||
|
filter.Add("node", node.ID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
filters, err := newListTasksFilters(options.Filters, byName)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx, cancel := c.getRequestContext()
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
r, err := state.controlClient.ListTasks(
|
||||||
|
ctx,
|
||||||
|
&swarmapi.ListTasksRequest{Filters: filters})
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
tasks := []types.Task{}
|
||||||
|
|
||||||
|
for _, task := range r.Tasks {
|
||||||
|
if task.Spec.GetContainer() != nil {
|
||||||
|
tasks = append(tasks, convert.TaskFromGRPC(*task))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return tasks, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetTask returns a task by an ID.
|
||||||
|
func (c *Cluster) GetTask(input string) (types.Task, error) {
|
||||||
|
c.mu.RLock()
|
||||||
|
defer c.mu.RUnlock()
|
||||||
|
|
||||||
|
state := c.currentNodeState()
|
||||||
|
if !state.IsActiveManager() {
|
||||||
|
return types.Task{}, c.errNoManager(state)
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx, cancel := c.getRequestContext()
|
||||||
|
defer cancel()
|
||||||
|
|
||||||
|
task, err := getTask(ctx, state.controlClient, input)
|
||||||
|
if err != nil {
|
||||||
|
return types.Task{}, err
|
||||||
|
}
|
||||||
|
return convert.TaskFromGRPC(*task), nil
|
||||||
|
}
|
Loading…
Add table
Reference in a new issue