mirror of
https://github.com/ceph/ceph-csi.git
synced 2025-03-09 08:59:30 +00:00
f87d06ed85
Several packages are only used while running the e2e suite. These packages are less important to update, as the they can not influence the final executable that is part of the Ceph-CSI container-image. By moving these dependencies out of the main Ceph-CSI go.mod, it is easier to identify if a reported CVE affects Ceph-CSI, or only the testing (like most of the Kubernetes CVEs). Signed-off-by: Niels de Vos <ndevos@ibm.com>
63 lines
1.1 KiB
Go
63 lines
1.1 KiB
Go
package backoff
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
)
|
|
|
|
// BackOffContext is a backoff policy that stops retrying after the context
|
|
// is canceled.
|
|
type BackOffContext interface { // nolint: golint
|
|
BackOff
|
|
Context() context.Context
|
|
}
|
|
|
|
type backOffContext struct {
|
|
BackOff
|
|
ctx context.Context
|
|
}
|
|
|
|
// WithContext returns a BackOffContext with context ctx
|
|
//
|
|
// ctx must not be nil
|
|
func WithContext(b BackOff, ctx context.Context) BackOffContext { // nolint: golint
|
|
if ctx == nil {
|
|
panic("nil context")
|
|
}
|
|
|
|
if b, ok := b.(*backOffContext); ok {
|
|
return &backOffContext{
|
|
BackOff: b.BackOff,
|
|
ctx: ctx,
|
|
}
|
|
}
|
|
|
|
return &backOffContext{
|
|
BackOff: b,
|
|
ctx: ctx,
|
|
}
|
|
}
|
|
|
|
func getContext(b BackOff) context.Context {
|
|
if cb, ok := b.(BackOffContext); ok {
|
|
return cb.Context()
|
|
}
|
|
if tb, ok := b.(*backOffTries); ok {
|
|
return getContext(tb.delegate)
|
|
}
|
|
return context.Background()
|
|
}
|
|
|
|
func (b *backOffContext) Context() context.Context {
|
|
return b.ctx
|
|
}
|
|
|
|
func (b *backOffContext) NextBackOff() time.Duration {
|
|
select {
|
|
case <-b.ctx.Done():
|
|
return Stop
|
|
default:
|
|
return b.BackOff.NextBackOff()
|
|
}
|
|
}
|