From 70639675490f007765e6ad0ced7f461ea1e36e44 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ren=C3=A1n=20Del=20Valle?= Date: Fri, 7 May 2021 15:17:45 -0700 Subject: [PATCH] Returning Summaries api back to what it was --- realis.go | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/realis.go b/realis.go index 27eac29..a3c4e2e 100644 --- a/realis.go +++ b/realis.go @@ -51,7 +51,7 @@ type Realis interface { DescheduleCronJob(key *aurora.JobKey) (*aurora.Response, error) FetchTaskConfig(instKey aurora.InstanceKey) (*aurora.TaskConfig, error) GetInstanceIds(key *aurora.JobKey, states []aurora.ScheduleStatus) ([]int32, error) - GetJobUpdateSummaries(jobUpdateQuery *aurora.JobUpdateQuery) ([]*aurora.JobUpdateSummary, error) + GetJobUpdateSummaries(jobUpdateQuery *aurora.JobUpdateQuery) (*aurora.Response, error) GetTaskStatus(query *aurora.TaskQuery) ([]*aurora.ScheduledTask, error) GetTasksWithoutConfigs(query *aurora.TaskQuery) ([]*aurora.ScheduledTask, error) GetJobs(role string) (*aurora.Response, *aurora.GetJobsResult_, error) @@ -574,7 +574,7 @@ func (r *realisClient) GetInstanceIds(key *aurora.JobKey, states []aurora.Schedu } -func (r *realisClient) GetJobUpdateSummaries(jobUpdateQuery *aurora.JobUpdateQuery) ([]*aurora.JobUpdateSummary, error) { +func (r *realisClient) GetJobUpdateSummaries(jobUpdateQuery *aurora.JobUpdateQuery) (*aurora.Response, error) { r.logger.debugPrintf("GetJobUpdateSummaries Thrift Payload: %+v\n", jobUpdateQuery) @@ -587,15 +587,14 @@ func (r *realisClient) GetJobUpdateSummaries(jobUpdateQuery *aurora.JobUpdateQue ) if retryErr != nil { - return nil, errors.Wrap(retryErr, "error getting job update summaries from Aurora Scheduler") + return resp, errors.Wrap(retryErr, "error getting job update summaries from Aurora Scheduler") } if resp.GetResult_() == nil || resp.GetResult_().GetGetJobUpdateSummariesResult_() == nil { return nil, errors.New("unexpected response from scheduler") - } - return resp.GetResult_().GetGetJobUpdateSummariesResult_().GetUpdateSummaries(), nil + return resp, nil } func (r *realisClient) GetJobs(role string) (*aurora.Response, *aurora.GetJobsResult_, error) {