youtubebeat/vendor/github.com/elastic/beats/libbeat/common/backoff.go

84 lines
1.8 KiB
Go
Raw Normal View History

2018-11-18 11:08:38 +01:00
// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you 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.
package common
import "time"
// A Backoff waits on errors with exponential backoff (limited by maximum
// backoff). Resetting Backoff will reset the next sleep timer to the initial
// backoff duration.
type Backoff struct {
duration time.Duration
done <-chan struct{}
init time.Duration
max time.Duration
last time.Time
}
func NewBackoff(done <-chan struct{}, init, max time.Duration) *Backoff {
return &Backoff{
duration: init,
done: done,
init: init,
max: max,
}
}
func (b *Backoff) Reset() {
b.duration = b.init
}
func (b *Backoff) Wait() bool {
backoff := b.duration
b.duration *= 2
if b.duration > b.max {
b.duration = b.max
}
select {
case <-b.done:
return false
case <-time.After(backoff):
b.last = time.Now()
return true
}
}
func (b *Backoff) WaitOnError(err error) bool {
if err == nil {
b.Reset()
return true
}
return b.Wait()
}
func (b *Backoff) TryWaitOnError(failTS time.Time, err error) bool {
if err == nil {
b.Reset()
return true
}
if failTS.Before(b.last) {
return true
}
return b.Wait()
}