listeners: remove timeout arg for roundtriper

DataDog · May 23, 2024 · 23c17fb · 23c17fb
1 parent 89209f0
commit 23c17fb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/internal/appsec/listener/graphqlsec/graphql.go b/internal/appsec/listener/graphqlsec/graphql.go
@@ -85,7 +85,7 @@ func (l *wafEventListener) onEvent(request *types.RequestOperation, _ types.Requ
 	}
 
 	if _, ok := l.addresses[shared.ServerIoNetURLAddr]; ok {
-		shared.RegisterRoundTripper(request, wafCtx, l.limiter, l.config.WAFTimeout)
+		shared.RegisterRoundTripper(request, wafCtx, l.limiter)
 	}
 
 	// Add span tags notifying this trace is AppSec-enabled

diff --git a/internal/appsec/listener/grpcsec/grpc.go b/internal/appsec/listener/grpcsec/grpc.go
@@ -105,7 +105,7 @@ func (l *wafEventListener) onEvent(op *types.HandlerOperation, handlerArgs types
 	}
 
 	if _, ok := l.addresses[shared.ServerIoNetURLAddr]; ok {
-		shared.RegisterRoundTripper(op, wafCtx, l.limiter, l.config.WAFTimeout)
+		shared.RegisterRoundTripper(op, wafCtx, l.limiter)
 	}
 
 	// Listen to the UserID address if the WAF rules are using it

diff --git a/internal/appsec/listener/httpsec/http.go b/internal/appsec/listener/httpsec/http.go
@@ -69,6 +69,7 @@ type wafEventListener struct {
 	once      sync.Once
 }
 
+// newWAFEventListener returns the WAF event listener to register in order to enable it.
 func newWafEventListener(wafHandle *waf.Handle, cfg *config.Config, limiter limiter.Limiter) *wafEventListener {
 	if wafHandle == nil {
 		log.Debug("appsec: no WAF Handle available, the HTTP WAF Event Listener will not be registered")
@@ -90,7 +91,6 @@ func newWafEventListener(wafHandle *waf.Handle, cfg *config.Config, limiter limi
 	}
 }
 
-// NewWAFEventListener returns the WAF event listener to register in order to enable it.
 func (l *wafEventListener) onEvent(op *types.Operation, args types.HandlerOperationArgs) {
 	wafCtx, err := l.wafHandle.NewContextWithBudget(l.config.WAFTimeout)
 	if err != nil {