serverless/appsec: use agent logger backend (#21446)

DataDog · Dec 11, 2023 · 6a0adec · 6a0adec
1 parent ad51d4b
commit 6a0adec
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pkg/serverless/appsec/appsec.go b/pkg/serverless/appsec/appsec.go
@@ -12,6 +12,7 @@ import (
 	"math/rand"
 	"time"
 
+	appsecLog "github.com/DataDog/appsec-internal-go/log"
 	"github.com/DataDog/datadog-agent/pkg/serverless/appsec/config"
 	"github.com/DataDog/datadog-agent/pkg/serverless/appsec/httpsec"
 	"github.com/DataDog/datadog-agent/pkg/serverless/proxy"
@@ -161,3 +162,13 @@ func wafHealth() error {
 func (a *AppSec) canExtractSchemas() bool {
 	return a.cfg.APISec.Enabled && a.cfg.APISec.SampleRate >= rand.Float64()
 }
+
+func init() {
+	appsecLog.SetBackend(appsecLog.Backend{
+		Trace:     log.Tracef,
+		Debug:     log.Debugf,
+		Info:      log.Infof,
+		Errorf:    log.Errorf,
+		Criticalf: log.Criticalf,
+	})
+}