Merge pull request #139 from mysteriumnetwork/configure-timeout

Allow to configure timeouts
mysteriumnetwork · Oct 4, 2021 · 439f093 · 439f093
2 parents 0d7c424 + 334f0c7
commit 439f093
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/transfer/gas/eth.go b/transfer/gas/eth.go
@@ -27,7 +27,7 @@ type EthStation struct {
 }
 
 // NewEthStation returns a new instance of defi pulse api for gas price checks.
-func NewEthStation(apiKey, endpointURI string, upperBound *big.Int) *EthStation {
+func NewEthStation(timeout time.Duration, apiKey, endpointURI string, upperBound *big.Int) *EthStation {
 	endpoint := endpointURI
 	if !strings.HasSuffix(endpoint, "/") {
 		endpoint += "/"
@@ -44,7 +44,7 @@ func NewEthStation(apiKey, endpointURI string, upperBound *big.Int) *EthStation
 				ExpectContinueTimeout: 4 * time.Second,
 				ResponseHeaderTimeout: 3 * time.Second,
 			},
-			Timeout: 10 * time.Second,
+			Timeout: timeout,
 		},
 		APIKey:      apiKey,
 		EndpointURI: endpoint,

diff --git a/transfer/gas/etherscan.go b/transfer/gas/etherscan.go
@@ -27,15 +27,15 @@ type EtherscanStation struct {
 }
 
 // NewEtherscanStation returns a new instance of etherscan api for gas price checks.
-func NewEtherscanStation(apiKey, endpointURI string, upperBound *big.Int) *EtherscanStation {
+func NewEtherscanStation(timeout time.Duration, apiKey, endpointURI string, upperBound *big.Int) *EtherscanStation {
 	endpoint := endpointURI
 	if !strings.HasSuffix(endpoint, "/") {
 		endpoint += "/"
 	}
 
 	return &EtherscanStation{
 		client: &http.Client{
-			Timeout: 10 * time.Second,
+			Timeout: timeout,
 		},
 		endpointURI: endpointURI,
 		upperBound:  upperBound,