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.
67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
package bitflyer
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/request"
|
|
"golang.org/x/time/rate"
|
|
)
|
|
|
|
// Exchange specific rate limit consts
|
|
const (
|
|
biflyerRateInterval = time.Minute * 5
|
|
bitflyerPrivateRequestRate = 500
|
|
bitflyerPrivateLowVolumeRequestRate = 100
|
|
bitflyerPrivateSendOrderRequestRate = 300
|
|
bitflyerPublicRequestRate = 500
|
|
)
|
|
|
|
// RateLimit implements the rate.Limiter interface
|
|
type RateLimit struct {
|
|
Auth *rate.Limiter
|
|
UnAuth *rate.Limiter
|
|
|
|
// Send a New Order
|
|
// Submit New Parent Order (Special order)
|
|
// Cancel All Orders
|
|
Order *rate.Limiter
|
|
LowVolume *rate.Limiter
|
|
}
|
|
|
|
// Limit limits outbound requests
|
|
func (r *RateLimit) Limit(ctx context.Context, f request.EndpointLimit) error {
|
|
switch f {
|
|
case request.Auth:
|
|
return r.Auth.Wait(ctx)
|
|
case orders:
|
|
err := r.Auth.Wait(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return r.Order.Wait(ctx)
|
|
case lowVolume:
|
|
err := r.LowVolume.Wait(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = r.Order.Wait(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return r.Auth.Wait(ctx)
|
|
default:
|
|
return r.UnAuth.Wait(ctx)
|
|
}
|
|
}
|
|
|
|
// SetRateLimit returns the rate limit for the exchange
|
|
func SetRateLimit() *RateLimit {
|
|
return &RateLimit{
|
|
Auth: request.NewRateLimit(biflyerRateInterval, bitflyerPrivateRequestRate),
|
|
UnAuth: request.NewRateLimit(biflyerRateInterval, bitflyerPublicRequestRate),
|
|
Order: request.NewRateLimit(biflyerRateInterval, bitflyerPrivateSendOrderRequestRate),
|
|
LowVolume: request.NewRateLimit(time.Minute, bitflyerPrivateLowVolumeRequestRate),
|
|
}
|
|
}
|