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.
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package hitbtc
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/request"
|
|
"golang.org/x/time/rate"
|
|
)
|
|
|
|
const (
|
|
hitbtcRateInterval = time.Second
|
|
hitbtcMarketDataReqRate = 100
|
|
hitbtcTradingReqRate = 300
|
|
hitbtcAllOthers = 10
|
|
|
|
marketRequests request.EndpointLimit = iota
|
|
tradingRequests
|
|
otherRequests
|
|
)
|
|
|
|
// RateLimit implements the request.Limiter interface
|
|
type RateLimit struct {
|
|
MarketData *rate.Limiter
|
|
Trading *rate.Limiter
|
|
Other *rate.Limiter
|
|
}
|
|
|
|
// Limit limits outbound requests
|
|
func (r *RateLimit) Limit(ctx context.Context, f request.EndpointLimit) error {
|
|
switch f {
|
|
case marketRequests:
|
|
return r.MarketData.Wait(ctx)
|
|
case tradingRequests:
|
|
return r.Trading.Wait(ctx)
|
|
case otherRequests:
|
|
return r.Other.Wait(ctx)
|
|
default:
|
|
return errors.New("functionality not found")
|
|
}
|
|
}
|
|
|
|
// SetRateLimit returns the rate limit for the exchange
|
|
func SetRateLimit() *RateLimit {
|
|
return &RateLimit{
|
|
MarketData: request.NewRateLimit(hitbtcRateInterval, hitbtcMarketDataReqRate),
|
|
Trading: request.NewRateLimit(hitbtcRateInterval, hitbtcTradingReqRate),
|
|
Other: request.NewRateLimit(hitbtcRateInterval, hitbtcAllOthers),
|
|
}
|
|
}
|