57 lines
1.7 KiB
Go
57 lines
1.7 KiB
Go
|
/*
|
||
|
|
Copyright 2014 The Kubernetes Authors.
|
||
|
|
|
||
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
you may not use this file except in compliance with the License.
|
||
|
|
You may obtain a copy of the License at
|
||
|
|
|
||
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
|
||
|
|
Unless required by applicable law or agreed to in writing, software
|
||
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
See the License for the specific language governing permissions and
|
||
|
|
limitations under the License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
// Modified version of the Kubernetes exponential-backoff code
|
||
|
|
|
||
|
|
package realis
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
"errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ConditionFunc returns true if the condition is satisfied, or an error
|
||
|
|
// if the loop should be aborted.
|
||
|
|
type ConditionFunc func() (done bool, err error)
|
||
|
|
|
||
|
|
// ExponentialBackoff repeats a condition check with exponential backoff.
|
||
|
|
//
|
||
|
|
// It checks the condition up to Steps times, increasing the wait by multiplying
|
||
|
|
// the previous duration by Factor.
|
||
|
|
//
|
||
|
|
// If Jitter is greater than zero, a random amount of each duration is added
|
||
|
|
// (between duration and duration*(1+jitter)).
|
||
|
|
//
|
||
|
|
// If the condition never returns true, ErrWaitTimeout is returned. All other
|
||
|
|
// errors terminate immediately.
|
||
|
|
func ExponentialBackoff(backoff Backoff, condition ConditionFunc) error {
|
||
|
|
duration := backoff.Duration
|
||
|
|
for i := 0; i < backoff.Steps; i++ {
|
||
|
|
if i != 0 {
|
||
|
|
adjusted := duration
|
||
|
|
if backoff.Jitter > 0.0 {
|
||
|
|
adjusted = Jitter(duration, backoff.Jitter)
|
||
|
|
}
|
||
|
|
time.Sleep(adjusted)
|
||
|
|
duration = time.Duration(float64(duration) * backoff.Factor)
|
||
|
|
}
|
||
|
|
if ok, err := condition(); err != nil || ok {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return errors.New("Timed out while retrying")
|
||
|
|
}
|