mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 23:16:45 +00:00
232d6ebc1f
* rate limits: Make context aware * binance: rate limit allow for cancellation of reservation when deadline is exceeded * request: add context.done() before initiating any bulk work. * binance: update error return for rate limiting * request: updated dealine check to remove after time.Now procedure as this will obfuscate a deadline which will be limited by the context check on every attempt, so no need to sleep with delay.
39 lines
842 B
Go
39 lines
842 B
Go
package bithumb
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/request"
|
|
"golang.org/x/time/rate"
|
|
)
|
|
|
|
// Exchange specific rate limit consts
|
|
const (
|
|
bithumbRateInterval = time.Second
|
|
bithumbAuthRate = 95
|
|
bithumbUnauthRate = 95
|
|
)
|
|
|
|
// RateLimit implements the request.Limiter interface
|
|
type RateLimit struct {
|
|
Auth *rate.Limiter
|
|
UnAuth *rate.Limiter
|
|
}
|
|
|
|
// Limit limits requests
|
|
func (r *RateLimit) Limit(ctx context.Context, f request.EndpointLimit) error {
|
|
if f == request.Auth {
|
|
return r.Auth.Wait(ctx)
|
|
}
|
|
return r.UnAuth.Wait(ctx)
|
|
}
|
|
|
|
// SetRateLimit returns the rate limit for the exchange
|
|
func SetRateLimit() *RateLimit {
|
|
return &RateLimit{
|
|
Auth: request.NewRateLimit(bithumbRateInterval, bithumbAuthRate),
|
|
UnAuth: request.NewRateLimit(bithumbRateInterval, bithumbUnauthRate),
|
|
}
|
|
}
|