mirror of
https://gitea.com/Lydanne/buildx.git
synced 2025-05-19 09:57:45 +08:00
b35a0f4718
Removes gogo/protobuf from buildx and updates to a version of moby/buildkit where gogo is removed. This also changes how the proto files are generated. This is because newer versions of protobuf are more strict about name conflicts. If two files have the same name (even if they are relative paths) and are used in different protoc commands, they'll conflict in the registry. Since protobuf file generation doesn't work very well with `paths=source_relative`, this removes the `go:generate` expression and just relies on the dockerfile to perform the generation. Signed-off-by: Jonathan A. Sternberg <jonathan.sternberg@docker.com>
110 lines
3.3 KiB
Go
110 lines
3.3 KiB
Go
/*
|
|
*
|
|
* Copyright 2017 gRPC 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.
|
|
*
|
|
*/
|
|
|
|
// Package backoff implement the backoff strategy for gRPC.
|
|
//
|
|
// This is kept in internal until the gRPC project decides whether or not to
|
|
// allow alternative backoff strategies.
|
|
package backoff
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"math/rand"
|
|
"time"
|
|
|
|
grpcbackoff "google.golang.org/grpc/backoff"
|
|
)
|
|
|
|
// Strategy defines the methodology for backing off after a grpc connection
|
|
// failure.
|
|
type Strategy interface {
|
|
// Backoff returns the amount of time to wait before the next retry given
|
|
// the number of consecutive failures.
|
|
Backoff(retries int) time.Duration
|
|
}
|
|
|
|
// DefaultExponential is an exponential backoff implementation using the
|
|
// default values for all the configurable knobs defined in
|
|
// https://github.com/grpc/grpc/blob/master/doc/connection-backoff.md.
|
|
var DefaultExponential = Exponential{Config: grpcbackoff.DefaultConfig}
|
|
|
|
// Exponential implements exponential backoff algorithm as defined in
|
|
// https://github.com/grpc/grpc/blob/master/doc/connection-backoff.md.
|
|
type Exponential struct {
|
|
// Config contains all options to configure the backoff algorithm.
|
|
Config grpcbackoff.Config
|
|
}
|
|
|
|
// Backoff returns the amount of time to wait before the next retry given the
|
|
// number of retries.
|
|
func (bc Exponential) Backoff(retries int) time.Duration {
|
|
if retries == 0 {
|
|
return bc.Config.BaseDelay
|
|
}
|
|
backoff, max := float64(bc.Config.BaseDelay), float64(bc.Config.MaxDelay)
|
|
for backoff < max && retries > 0 {
|
|
backoff *= bc.Config.Multiplier
|
|
retries--
|
|
}
|
|
if backoff > max {
|
|
backoff = max
|
|
}
|
|
// Randomize backoff delays so that if a cluster of requests start at
|
|
// the same time, they won't operate in lockstep.
|
|
backoff *= 1 + bc.Config.Jitter*(rand.Float64()*2-1)
|
|
if backoff < 0 {
|
|
return 0
|
|
}
|
|
return time.Duration(backoff)
|
|
}
|
|
|
|
// ErrResetBackoff is the error to be returned by the function executed by RunF,
|
|
// to instruct the latter to reset its backoff state.
|
|
var ErrResetBackoff = errors.New("reset backoff state")
|
|
|
|
// RunF provides a convenient way to run a function f repeatedly until the
|
|
// context expires or f returns a non-nil error that is not ErrResetBackoff.
|
|
// When f returns ErrResetBackoff, RunF continues to run f, but resets its
|
|
// backoff state before doing so. backoff accepts an integer representing the
|
|
// number of retries, and returns the amount of time to backoff.
|
|
func RunF(ctx context.Context, f func() error, backoff func(int) time.Duration) {
|
|
attempt := 0
|
|
timer := time.NewTimer(0)
|
|
for ctx.Err() == nil {
|
|
select {
|
|
case <-timer.C:
|
|
case <-ctx.Done():
|
|
timer.Stop()
|
|
return
|
|
}
|
|
|
|
err := f()
|
|
if errors.Is(err, ErrResetBackoff) {
|
|
timer.Reset(0)
|
|
attempt = 0
|
|
continue
|
|
}
|
|
if err != nil {
|
|
return
|
|
}
|
|
timer.Reset(backoff(attempt))
|
|
attempt++
|
|
}
|
|
}
|