RetailCRM-Mirror/mg-transport-api-client-go
1
0
Fork 0
mirror of synced 2025-01-30 12:31:41 +03:00
Code Issues Projects Releases Wiki Activity

fix linter

Browse Source
This commit is contained in:
Pavel 2024-12-06 19:11:34 +03:00
parent 5fb9f0f895
commit fabaf40050
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
v1/rate_limit.go
View File

@ -8,6 +8,9 @@ import (
"time" "time"
) )
// Use double the CPU count for sharding
const shardsPerCoreMultiplier = 2
var NoopLimiter Limiter = &noopLimiter{} var NoopLimiter Limiter = &noopLimiter{}
type token struct { type token struct {
@ -38,7 +41,7 @@ type tokenShard struct {
// NewTokensBucket creates a sharded token bucket limiter. // NewTokensBucket creates a sharded token bucket limiter.
func NewTokensBucket(maxRPS uint32, unusedTokenTime, checkTokenTime time.Duration) Limiter { func NewTokensBucket(maxRPS uint32, unusedTokenTime, checkTokenTime time.Duration) Limiter {
shardCount := uint32(runtime.NumCPU() * 2) // Use double the CPU count for sharding shardCount := uint32(runtime.NumCPU() * shardsPerCoreMultiplier)
shards := make([]*tokenShard, shardCount) shards := make([]*tokenShard, shardCount)
for i := range shards { for i := range shards {
shards[i] = &tokenShard{tokens: make(map[string]*token)} shards[i] = &tokenShard{tokens: make(map[string]*token)}