default policy for slaDrainHosts

This commit is contained in:
lenhattan86 2021-10-21 14:44:52 -07:00
parent fc1ecd7034
commit fb69e15d1d
3 changed files with 58 additions and 0 deletions

View file

@ -59,6 +59,21 @@ func (c *Client) SLADrainHosts(policy *aurora.SlaPolicy, timeout int64, hosts ..
return nil, errors.New("no hosts provided to drain")
}
if policy == nil ||
(policy.PercentageSlaPolicy == nil &&
policy.CountSlaPolicy == nil &&
policy.CoordinatorSlaPolicy == nil) {
policy = &defaultSlaPolicy
c.logger.Printf("Warning: start draining with default sla policy %v", policy)
}
if timeout < 0 {
c.logger.Printf("Warning: timeout %d secs is invalid, draining with default timeout %d secs",
timeout,
defaultSlaDrainTimeoutSecs)
timeout = defaultSlaDrainTimeoutSecs
}
drainList := aurora.NewHosts()
drainList.HostNames = hosts