func getTask(ctx context.Context, c swarmapi.ControlClient, input string) (*swarmapi.Task, error) { // GetTask to match via full ID. rg, err := c.GetTask(ctx, &swarmapi.GetTaskRequest{TaskID: input}) if err != nil { // If any error (including NotFound), ListTasks to match via full name. rl, err := c.ListTasks(ctx, &swarmapi.ListTasksRequest{Filters: &swarmapi.ListTasksRequest_Filters{Names: []string{input}}}) if err != nil || len(rl.Tasks) == 0 { // If any error or 0 result, ListTasks to match via ID prefix. rl, err = c.ListTasks(ctx, &swarmapi.ListTasksRequest{Filters: &swarmapi.ListTasksRequest_Filters{IDPrefixes: []string{input}}}) } if err != nil { return nil, err } if len(rl.Tasks) == 0 { return nil, fmt.Errorf("task %s not found", input) } if l := len(rl.Tasks); l > 1 { return nil, fmt.Errorf("task %s is ambigious (%d matches found)", input, l) } return rl.Tasks[0], nil } return rg.Task, nil }