chore: pass ctx to thorchain funcs

strangelove-ventures · Oct 1, 2024 · 024e307 · 024e307
1 parent 048309d
commit 024e307
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 11 deletions.
diff --git a/examples/thorchain/features/helpers.go b/examples/thorchain/features/helpers.go
@@ -66,7 +66,7 @@ func PollForPool(ctx context.Context, thorchain *tc.Thorchain, deltaBlocks int64
 		return fmt.Errorf("failed to get height: %w", err)
 	}
 	doPoll := func(ctx context.Context, height int64) (any, error) {
-		pool, err := thorchain.APIGetPool(nil, asset)
+		pool, err := thorchain.APIGetPool(ctx, asset)
 		if err != nil {
 			time.Sleep(time.Second) // rate limit
 			return nil, err
@@ -143,7 +143,7 @@ func PollSwapCompleted(ctx context.Context, thorchain *tc.Thorchain, deltaBlocks
 		return tc.Saver{}, fmt.Errorf("failed to get height: %w", err)
 	}
 	doPoll := func(ctx context.Context, height int64) (any, error) {
-		stages, err := thorchain.APIGetTxStages(nil, txHash)
+		stages, err := thorchain.APIGetTxStages(ctx, txHash)
 		if err != nil {
 			time.Sleep(time.Second) // rate limit
 			return nil, err
@@ -167,7 +167,7 @@ func PollOutboundSigned(ctx context.Context, thorchain *tc.Thorchain, deltaBlock
 		return tc.Saver{}, fmt.Errorf("failed to get height: %w", err)
 	}
 	doPoll := func(ctx context.Context, height int64) (any, error) {
-		stages, err := thorchain.APIGetTxStages(nil, txHash)
+		stages, err := thorchain.APIGetTxStages(ctx, txHash)
 		if err != nil {
 			time.Sleep(time.Second) // rate limit
 			return nil, err
@@ -215,7 +215,7 @@ func PollForPoolSuspended(ctx context.Context, thorchain *tc.Thorchain, deltaBlo
 		return fmt.Errorf("failed to get height: %w", err)
 	}
 	doPoll := func(ctx context.Context, height int64) (any, error) {
-		pool, err := thorchain.APIGetPool(nil, exoAsset)
+		pool, err := thorchain.APIGetPool(ctx, exoAsset)
 		if err != nil {
 			time.Sleep(time.Second) // rate limit
 			return nil, err

diff --git a/examples/thorchain/features/ragnarok.go b/examples/thorchain/features/ragnarok.go
@@ -29,7 +29,7 @@ func Ragnarok(
 	}
 	exoAsset := exoChainType.GetGasAsset()
 
-	_, err = thorchain.APIGetPool(nil, exoAsset)
+	_, err = thorchain.APIGetPool(ctx, exoAsset)
 	if err != nil {
 		return fmt.Errorf("pool (%s) not found to ragnarok, %w", exoChain.Config().Name, err)
 	}

diff --git a/examples/thorchain/features/saver_eject.go b/examples/thorchain/features/saver_eject.go
@@ -36,7 +36,7 @@ func SaverEject(
 
 	// Reset mimirs
 	mimirLock.Lock()
-	mimirs, err := thorchain.APIGetMimirs(nil)
+	mimirs, err := thorchain.APIGetMimirs(ctx)
 	if err != nil {
 		mimirLock.Unlock()
 		return exoUser, err
@@ -65,15 +65,15 @@ func SaverEject(
 	}
 	exoAsset := exoChainType.GetGasAsset()
 
-	pool, err := thorchain.APIGetPool(nil, exoAsset)
+	pool, err := thorchain.APIGetPool(ctx, exoAsset)
 	if err != nil {
 		mimirLock.Unlock()
 		return exoUser, err
 	}
 	saveAmount := sdkmath.NewUintFromString(pool.BalanceAsset).
 		MulUint64(2000).QuoUint64(10_000)
 
-	saverQuote, err := thorchain.APIGetSaverDepositQuote(nil, exoAsset, saveAmount)
+	saverQuote, err := thorchain.APIGetSaverDepositQuote(ctx, exoAsset, saveAmount)
 	if err != nil {
 		mimirLock.Unlock()
 		return exoUser, err
@@ -150,7 +150,7 @@ func SaverEject(
 		}
 	}
 
-	mimirs, err = thorchain.APIGetMimirs(nil)
+	mimirs, err = thorchain.APIGetMimirs(ctx)
 	if err != nil {
 		mimirLock.Unlock()
 		return exoUser, err

diff --git a/examples/thorchain/features/savers.go b/examples/thorchain/features/savers.go
@@ -32,14 +32,14 @@ func Saver(
 	}
 	exoAsset := exoChainType.GetGasAsset()
 
-	pool, err := thorchain.APIGetPool(nil, exoAsset)
+	pool, err := thorchain.APIGetPool(ctx, exoAsset)
 	if err != nil {
 		return exoUser, err
 	}
 	saveAmount := sdkmath.NewUintFromString(pool.BalanceAsset).
 		MulUint64(500).QuoUint64(10_000)
 
-	saverQuote, err := thorchain.APIGetSaverDepositQuote(nil, exoAsset, saveAmount)
+	saverQuote, err := thorchain.APIGetSaverDepositQuote(ctx, exoAsset, saveAmount)
 	if err != nil {
 		return exoUser, err
 	}