address PR comments

This commit is contained in:
lenhattan86 2021-11-01 15:12:23 -07:00
parent fb69e15d1d
commit a7b90591bc

View file

@ -59,10 +59,7 @@ 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) {
if policy == nil || policy.CountSetFieldsSlaPolicy() == 0 {
policy = &defaultSlaPolicy
c.logger.Printf("Warning: start draining with default sla policy %v", policy)
}