mirror of
https://github.com/rclone/rclone.git
synced 2025-03-03 18:01:38 +01:00
401cf81034
This commit modernizes Go usage. This was done with:
go run golang.org/x/tools/gopls/internal/analysis/modernize/cmd/modernize@latest -fix -test ./...
Then files needed to be `go fmt`ed and a few comments needed to be
restored.
The modernizations include replacing
- if/else conditional assignment by a call to the built-in min or max functions added in go1.21
- sort.Slice(x, func(i, j int) bool) { return s[i] < s[j] } by a call to slices.Sort(s), added in go1.21
- interface{} by the 'any' type added in go1.18
- append([]T(nil), s...) by slices.Clone(s) or slices.Concat(s), added in go1.21
- loop around an m[k]=v map update by a call to one of the Collect, Copy, Clone, or Insert functions from the maps package, added in go1.21
- []byte(fmt.Sprintf...) by fmt.Appendf(nil, ...), added in go1.19
- append(s[:i], s[i+1]...) by slices.Delete(s, i, i+1), added in go1.21
- a 3-clause for i := 0; i < n; i++ {} loop by for i := range n {}, added in go1.22
35 lines
910 B
Go
35 lines
910 B
Go
package accounting
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/rclone/rclone/fs"
|
|
"golang.org/x/time/rate"
|
|
)
|
|
|
|
var (
|
|
tpsBucket *rate.Limiter // for limiting number of http transactions per second
|
|
)
|
|
|
|
// StartLimitTPS starts the token bucket for transactions per second
|
|
// limiting if necessary
|
|
func StartLimitTPS(ctx context.Context) {
|
|
ci := fs.GetConfig(ctx)
|
|
if ci.TPSLimit > 0 {
|
|
tpsBurst := max(ci.TPSLimitBurst, 1)
|
|
tpsBucket = rate.NewLimiter(rate.Limit(ci.TPSLimit), tpsBurst)
|
|
fs.Infof(nil, "Starting transaction limiter: max %g transactions/s with burst %d", ci.TPSLimit, tpsBurst)
|
|
}
|
|
}
|
|
|
|
// LimitTPS limits the number of transactions per second if enabled.
|
|
// It should be called once per transaction.
|
|
func LimitTPS(ctx context.Context) {
|
|
if tpsBucket != nil {
|
|
tbErr := tpsBucket.Wait(ctx)
|
|
if tbErr != nil && tbErr != context.Canceled {
|
|
fs.Errorf(nil, "HTTP token bucket error: %v", tbErr)
|
|
}
|
|
}
|
|
}
|