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

add restartInstances with sla requirements

Browse source
This commit is contained in:
lenhattan86 2021-06-07 15:37:27 -07:00
parent 2d81147aaa
commit 63e466ac0d
8 changed files with 1689 additions and 1132 deletions
Download patch file Download diff file Expand all files Collapse all files

14
realis.go
View file

@ -522,6 +522,20 @@ func (c *Client) RestartInstances(key aurora.JobKey, instances ...int32) error {
return nil
}
// Restarts specific instances with SLA specified
func (c *Client) SlaRestartInstances(key aurora.JobKey, policy *aurora.SlaPolicy, instances ...int32) error {
c.logger.DebugPrintf("SlaRestartInstances Thrift Payload: %+v %v\n", key, instances)
_, retryErr := c.thriftCallWithRetries(false, func() (*aurora.Response, error) {
return c.client.SlaRestartShards(context.TODO(), &key, instances, policy)
})
if retryErr != nil {
return errors.Wrap(retryErr, "error sending SlaRestartInstances command to Aurora Scheduler")
}
return nil
}
// Restarts all active tasks under a job configuration.
func (c *Client) RestartJob(key aurora.JobKey) error {