d0zingcat/gocryptotrader
1
0
Fork 0
mirror of https://github.com/d0zingcat/gocryptotrader.git synced 2026-05-13 23:16:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

cmd/wrapper_coverage: prevent non-needed REST requests via context timeout (#1154)

Browse Source 
* wrapper_coverage/exchanges: cancel context to not send rest requests/ populate context through functions that do rest requests

* linter: fix

* exchange_template: fix test

---------

Co-authored-by: Ryan O'Hara-Reid <ryan.oharareid@thrasher.io>
This commit is contained in:
Ryan O'Hara-Reid
2023-04-02 11:20:25 +10:00
committed by GitHub
parent c69f48f8d7
commit 7cadc5cffb
62 changed files with 233 additions and 228 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
exchanges/bithumb/bithumb_test.go
View File

@@ -61,12 +61,12 @@ func TestMain(m *testing.M) {
func TestStart(t *testing.T) {
t.Parallel()
err := b.Start(nil)
err := b.Start(context.Background(), nil)
if !errors.Is(err, common.ErrNilPointer) {
t.Fatalf("received: '%v' but expected: '%v'", err, common.ErrNilPointer)
}
var testWg sync.WaitGroup
err = b.Start(&testWg)
err = b.Start(context.Background(), &testWg)
if err != nil {
t.Fatal(err)
}

14
exchanges/bithumb/bithumb_wrapper.go
View File

@@ -36,7 +36,7 @@ const wsRateLimitMillisecond = 1000
var errNotEnoughPairs = errors.New("at least one currency is required to fetch order history")
// GetDefaultConfig returns a default exchange config
func (b *Bithumb) GetDefaultConfig() (*config.Exchange, error) {
func (b *Bithumb) GetDefaultConfig(ctx context.Context) (*config.Exchange, error) {
b.SetDefaults()
exchCfg := new(config.Exchange)
exchCfg.Name = b.Name
@@ -49,7 +49,7 @@ func (b *Bithumb) GetDefaultConfig() (*config.Exchange, error) {
}
if b.Features.Supports.RESTCapabilities.AutoPairUpdates {
err = b.UpdateTradablePairs(context.TODO(), true)
err = b.UpdateTradablePairs(ctx, true)
if err != nil {
return nil, err
}
@@ -195,25 +195,25 @@ func (b *Bithumb) Setup(exch *config.Exchange) error {
}
// Start starts the Bithumb go routine
func (b *Bithumb) Start(wg *sync.WaitGroup) error {
func (b *Bithumb) Start(ctx context.Context, wg *sync.WaitGroup) error {
if wg == nil {
return fmt.Errorf("%T %w", wg, common.ErrNilPointer)
}
wg.Add(1)
go func() {
b.Run()
b.Run(ctx)
wg.Done()
}()
return nil
}
// Run implements the Bithumb wrapper
func (b *Bithumb) Run() {
func (b *Bithumb) Run(ctx context.Context) {
if b.Verbose {
b.PrintEnabledPairs()
}
err := b.UpdateOrderExecutionLimits(context.TODO(), asset.Empty)
err := b.UpdateOrderExecutionLimits(ctx, asset.Empty)
if err != nil {
log.Errorf(log.ExchangeSys,
"%s failed to set exchange order execution limits. Err: %v",
@@ -225,7 +225,7 @@ func (b *Bithumb) Run() {
return
}
err = b.UpdateTradablePairs(context.TODO(), false)
err = b.UpdateTradablePairs(ctx, false)
if err != nil {
log.Errorf(log.ExchangeSys, "%s failed to update tradable pairs. Err: %s", b.Name, err)
}