ridv/gorealis
1
0
Fork 0
Code Issues Pull requests Projects Releases 23 Packages Wiki Activity Actions

Making sure words instance and tasks are used correctly

Browse source
This commit is contained in:
Renan DelValle 2016-08-09 14:21:43 -07:00
parent 07173df33c
commit 8e7d7b72d4
1 changed files with 16 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

34
realis.go
View file

@ -89,7 +89,7 @@ func (r *Realis) Close() {
} }
// Uses predefined set of states to retrieve a set of active jobs in Apache Aurora // Uses predefined set of states to retrieve a set of active jobs in Apache Aurora
func (r *Realis) getActiveTaskIds(key *aurora.JobKey) (map[int32]bool, error) { func (r *Realis) getActiveInstanceIds(key *aurora.JobKey) (map[int32]bool, error) {
taskQ := &aurora.TaskQuery{Role: key.Role, taskQ := &aurora.TaskQuery{Role: key.Role,
Environment: key.Environment, Environment: key.Environment,
JobName: key.Name, JobName: key.Name,
@ -102,40 +102,38 @@ func (r *Realis) getActiveTaskIds(key *aurora.JobKey) (map[int32]bool, error) {
tasks := response.GetResult_().GetScheduleStatusResult_().GetTasks() tasks := response.GetResult_().GetScheduleStatusResult_().GetTasks()
jobTaskIds := make(map[int32]bool) jobInstanceIds := make(map[int32]bool)
for _, task := range tasks { for _, task := range tasks {
jobTaskIds[task.GetAssignedTask().GetInstanceId()] = true jobInstanceIds[task.GetAssignedTask().GetInstanceId()] = true
} }
return jobTaskIds, nil return jobInstanceIds, nil
} }
func (r *Realis) KillInstance(key *aurora.JobKey, instanceId int32) (*aurora.Response, error) {
func (r *Realis) KillTask(key *aurora.JobKey, taskId int32) { instanceIds := make(map[int32]bool)
instanceIds[instanceId] = true
taskIds := make(map[int32]bool) response, err := r.client.KillTasks(key, instanceIds)
taskIds[taskId] = true
response, err := r.client.KillTasks(key, taskIds)
if err != nil { if err != nil {
return nil, errors.Wrap(err, "Error sending Kill command to Aurora Scheduler.") return nil, errors.Wrap(err, "Error sending Kill command to Aurora Scheduler.")
} }
return response, nil return response, nil
} }
// Sends a kill message to the scheduler for all active tasks under a job // Sends a kill message to the scheduler for all active tasks under a job
func (r *Realis) KillJob(key *aurora.JobKey) (*aurora.Response, error) { func (r *Realis) KillJob(key *aurora.JobKey) (*aurora.Response, error) {
taskIds, err := r.getActiveTaskIds(key) instanceIds, err := r.getActiveInstanceIds(key)
if err != nil { if err != nil {
return nil, errors.Wrap(err, "Could not retrieve relevant task IDs.") return nil, errors.Wrap(err, "Could not retrieve relevant task instance IDs.")
} }
if len(taskIds) > 0 { if len(instanceIds) > 0 {
response, err := r.client.KillTasks(key, taskIds) response, err := r.client.KillTasks(key, instanceIds)
if err != nil { if err != nil {
return nil, errors.Wrap(err, "Error sending Kill command to Aurora Scheduler.") return nil, errors.Wrap(err, "Error sending Kill command to Aurora Scheduler.")
@ -161,13 +159,13 @@ func (r *Realis) CreateJob(auroraJob *Job) (*aurora.Response, error) {
// Restarts all active tasks under a job configuration // Restarts all active tasks under a job configuration
func (r *Realis) RestartJob(key *aurora.JobKey) (*aurora.Response, error) { func (r *Realis) RestartJob(key *aurora.JobKey) (*aurora.Response, error) {
taskIds, err := r.getActiveTaskIds(key) instanceIds, err := r.getActiveInstanceIds(key)
if err != nil { if err != nil {
return nil, errors.Wrap(err, "Could not retrieve relevant task IDs.") return nil, errors.Wrap(err, "Could not retrieve relevant task instance IDs.")
} }
if len(taskIds) > 0 { if len(instanceIds) > 0 {
response, err := r.client.RestartShards(key, taskIds) response, err := r.client.RestartShards(key, instanceIds)
if err != nil { if err != nil {
return nil, errors.Wrap(err, "Error sending Restart command to Aurora Scheduler.") return nil, errors.Wrap(err, "Error sending Restart command to Aurora Scheduler.")