57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
// Copyright 2016 Google Inc. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package gensupport
|
|
|
|
import (
|
|
"io"
|
|
"time"
|
|
)
|
|
|
|
// errReader reads out of a buffer until it is empty, then returns the specified error.
|
|
type errReader struct {
|
|
buf []byte
|
|
err error
|
|
}
|
|
|
|
func (er *errReader) Read(p []byte) (int, error) {
|
|
if len(er.buf) == 0 {
|
|
if er.err == nil {
|
|
return 0, io.EOF
|
|
}
|
|
return 0, er.err
|
|
}
|
|
n := copy(p, er.buf)
|
|
er.buf = er.buf[n:]
|
|
return n, nil
|
|
}
|
|
|
|
// UniformPauseStrategy implements BackoffStrategy with uniform pause.
|
|
type UniformPauseStrategy time.Duration
|
|
|
|
func (p UniformPauseStrategy) Pause() (time.Duration, bool) { return time.Duration(p), true }
|
|
func (p UniformPauseStrategy) Reset() {}
|
|
|
|
// NoPauseStrategy implements BackoffStrategy with infinite 0-length pauses.
|
|
const NoPauseStrategy = UniformPauseStrategy(0)
|
|
|
|
// LimitRetryStrategy wraps a BackoffStrategy but limits the number of retries.
|
|
type LimitRetryStrategy struct {
|
|
Max int
|
|
Strategy BackoffStrategy
|
|
n int
|
|
}
|
|
|
|
func (l *LimitRetryStrategy) Pause() (time.Duration, bool) {
|
|
l.n++
|
|
if l.n > l.Max {
|
|
return 0, false
|
|
}
|
|
return l.Strategy.Pause()
|
|
}
|
|
|
|
func (l *LimitRetryStrategy) Reset() {
|
|
l.n = 0
|
|
l.Strategy.Reset()
|
|
}
|