PA-26431 Fix Insrequester Timeout Data Type

useinsider · Apr 25, 2024 · 6dcc60e · 6dcc60e
1 parent 307d596
commit 6dcc60e
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/insrequester/requester.go b/insrequester/requester.go
@@ -8,7 +8,7 @@ import (
 	"github.com/slok/goresilience/circuitbreaker"
 	goresilienceErrors "github.com/slok/goresilience/errors"
 	"github.com/slok/goresilience/retry"
-	"github.com/slok/goresilience/timeout"
+	resilienceTimeout "github.com/slok/goresilience/timeout"
 	"net/http"
 	"time"
 )
@@ -38,7 +38,7 @@ type Requester interface {
 	Delete(re RequestEntity) (*http.Response, error)
 	WithRetry(config RetryConfig) *Request
 	WithCircuitbreaker(config CircuitBreakerConfig) *Request
-	WithTimeout(timeoutSeconds int) *Request
+	WithTimeout(timeout time.Duration) *Request
 	WithHeaders(headers Headers) *Request
 	Load() *Request
 }
@@ -187,14 +187,14 @@ func (r *Request) WithCircuitbreaker(config CircuitBreakerConfig) *Request {
 	return r
 }
 
-func (r *Request) WithTimeout(timeoutSeconds int) *Request {
-	if timeoutSeconds == 0 {
+func (r *Request) WithTimeout(timeout time.Duration) *Request {
+	if timeout == 0 {
 		r.timeout = 30 * time.Second
 	} else {
-		r.timeout = time.Duration(timeoutSeconds) * time.Second
+		r.timeout = timeout
 	}
 
-	mw := timeout.NewMiddleware(timeout.Config{
+	mw := resilienceTimeout.NewMiddleware(resilienceTimeout.Config{
 		Timeout: r.timeout,
 	})
 	r.middlewares = append(r.middlewares, mw)