mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-20 07:26:46 +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.
67 lines
1.9 KiB
Go
67 lines
1.9 KiB
Go
package btcmarkets
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/request"
|
|
"golang.org/x/time/rate"
|
|
)
|
|
|
|
// BTCMarkets Rate limit consts
|
|
const (
|
|
btcmarketsRateInterval = time.Second * 10
|
|
btcmarketsAuthLimit = 50
|
|
btcmarketsUnauthLimit = 50
|
|
btcmarketsOrderLimit = 30
|
|
btcmarketsBatchOrderLimit = 5
|
|
btcmarketsWithdrawLimit = 10
|
|
btcmarketsCreateNewReportLimit = 1
|
|
|
|
// Used to match endpints to rate limits
|
|
orderFunc request.EndpointLimit = iota
|
|
batchFunc
|
|
withdrawFunc
|
|
newReportFunc
|
|
)
|
|
|
|
// RateLimit implements the request.Limiter interface
|
|
type RateLimit struct {
|
|
Auth *rate.Limiter
|
|
UnAuth *rate.Limiter
|
|
OrderPlacement *rate.Limiter
|
|
BatchOrders *rate.Limiter
|
|
WithdrawRequest *rate.Limiter
|
|
CreateNewReport *rate.Limiter
|
|
}
|
|
|
|
// Limit limits the outbound requests
|
|
func (r *RateLimit) Limit(ctx context.Context, f request.EndpointLimit) error {
|
|
switch f {
|
|
case request.Auth:
|
|
return r.Auth.Wait(ctx)
|
|
case orderFunc:
|
|
return r.OrderPlacement.Wait(ctx)
|
|
case batchFunc:
|
|
return r.BatchOrders.Wait(ctx)
|
|
case withdrawFunc:
|
|
return r.WithdrawRequest.Wait(ctx)
|
|
case newReportFunc:
|
|
return r.CreateNewReport.Wait(ctx)
|
|
default:
|
|
return r.UnAuth.Wait(ctx)
|
|
}
|
|
}
|
|
|
|
// SetRateLimit returns the rate limit for the exchange
|
|
func SetRateLimit() *RateLimit {
|
|
return &RateLimit{
|
|
Auth: request.NewRateLimit(btcmarketsRateInterval, btcmarketsAuthLimit),
|
|
UnAuth: request.NewRateLimit(btcmarketsRateInterval, btcmarketsUnauthLimit),
|
|
OrderPlacement: request.NewRateLimit(btcmarketsRateInterval, btcmarketsOrderLimit),
|
|
BatchOrders: request.NewRateLimit(btcmarketsRateInterval, btcmarketsBatchOrderLimit),
|
|
WithdrawRequest: request.NewRateLimit(btcmarketsRateInterval, btcmarketsWithdrawLimit),
|
|
CreateNewReport: request.NewRateLimit(btcmarketsRateInterval, btcmarketsCreateNewReportLimit),
|
|
}
|
|
}
|