diff --git a/examples/client.go b/examples/client.go index dc71f8e..5eae5c4 100644 --- a/examples/client.go +++ b/examples/client.go @@ -553,9 +553,9 @@ func main() { case "taskStatus": fmt.Println("Getting task status") taskQ := &aurora.TaskQuery{ - Role: job.JobKey().Role, - Environment: job.JobKey().Environment, - JobName: job.JobKey().Name, + Role: &job.JobKey().Role, + Environment: &job.JobKey().Environment, + JobName: &job.JobKey().Name, } tasks, err := r.GetTaskStatus(taskQ) if err != nil { @@ -567,9 +567,9 @@ func main() { case "tasksWithoutConfig": fmt.Println("Getting task status") taskQ := &aurora.TaskQuery{ - Role: job.JobKey().Role, - Environment: job.JobKey().Environment, - JobName: job.JobKey().Name, + Role: &job.JobKey().Role, + Environment: &job.JobKey().Environment, + JobName: &job.JobKey().Name, } tasks, err := r.GetTasksWithoutConfigs(taskQ) if err != nil { diff --git a/realis.go b/realis.go index 82a5eac..3e4837a 100644 --- a/realis.go +++ b/realis.go @@ -497,9 +497,9 @@ func (r *realisClient) Close() { // Uses predefined set of states to retrieve a set of active jobs in Apache Aurora. func (r *realisClient) GetInstanceIds(key *aurora.JobKey, states map[aurora.ScheduleStatus]bool) (map[int32]bool, error) { taskQ := &aurora.TaskQuery{ - Role: key.Role, - Environment: key.Environment, - JobName: key.Name, + Role: &key.Role, + Environment: &key.Environment, + JobName: &key.Name, Statuses: states, } @@ -880,9 +880,9 @@ func (r *realisClient) FetchTaskConfig(instKey aurora.InstanceKey) (*aurora.Task ids[instKey.InstanceId] = true taskQ := &aurora.TaskQuery{ - Role: instKey.JobKey.Role, - Environment: instKey.JobKey.Environment, - JobName: instKey.JobKey.Name, + Role: &instKey.JobKey.Role, + Environment: &instKey.JobKey.Environment, + JobName: &instKey.JobKey.Name, InstanceIds: ids, Statuses: aurora.ACTIVE_STATES, } diff --git a/realis_e2e_test.go b/realis_e2e_test.go index 342dbb1..cbf2804 100644 --- a/realis_e2e_test.go +++ b/realis_e2e_test.go @@ -75,11 +75,7 @@ func TestNonExistentEndpoint(t *testing.T) { ) defer r.Close() - taskQ := &aurora.TaskQuery{ - Role: "no", - Environment: "task", - JobName: "here", - } + taskQ := &aurora.TaskQuery{} _, err = r.GetTasksWithoutConfigs(taskQ)