From 034f0087c492e273ae24e05bf4dc35562c6493e6 Mon Sep 17 00:00:00 2001
From: Renan DelValle <renanidelvalle+noreply@gmail.com>
Date: Fri, 3 May 2019 18:22:00 -0700
Subject: [PATCH] Optimized GetInstancesIds to create a copy of the JobKey
 being passed down in order to avoid unexpected behavior. Instead of setting
 every variable name separately, now a JobKey array is being created.

---
 realis.go | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/realis.go b/realis.go
index 21ce193..1b61e7d 100644
--- a/realis.go
+++ b/realis.go
@@ -495,10 +495,8 @@ 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 []aurora.ScheduleStatus) ([]int32, error) {
 	taskQ := &aurora.TaskQuery{
-		Role:        &key.Role,
-		Environment: &key.Environment,
-		JobName:     &key.Name,
-		Statuses:    states,
+		JobKeys:  []*aurora.JobKey{{Environment: key.Environment, Role: key.Role, Name: key.Name}},
+		Statuses: states,
 	}
 
 	r.logger.DebugPrintf("GetTasksWithoutConfigs Thrift Payload: %+v\n", taskQ)