Apply suggestions from code review

Co-authored-by: Julio Guerra <julio@datadog.com>
DataDog · Feb 24, 2022 · 927399e · 927399e
1 parent 7716114
commit 927399e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/contrib/gin-gonic/gin/appsec.go b/contrib/gin-gonic/gin/appsec.go
@@ -27,8 +27,8 @@ func useAppSec(c *gin.Context, span tracer.Span) func() {
 		}
 	}
 	args := httpsec.MakeHandlerOperationArgs(req, params)
-	ctx, op := httpsec.StartOperation(c.Request.Context(), args)
-	*req = *req.WithContext(ctx)
+	ctx, op := httpsec.StartOperation(req.Context(), args)
+	c.Request = req.WithContext(ctx)
 	return func() {
 		events := op.Finish(httpsec.HandlerOperationRes{Status: c.Writer.Status()})
 		if len(events) > 0 {

diff --git a/contrib/labstack/echo.v4/appsec.go b/contrib/labstack/echo.v4/appsec.go
@@ -28,7 +28,7 @@ func withAppSec(next echo.HandlerFunc) echo.HandlerFunc {
 		}
 		args := httpsec.MakeHandlerOperationArgs(req, params)
 		ctx, op := httpsec.StartOperation(req.Context(), args)
-		*req = *req.WithContext(ctx)
+		c.SetRequest(req.WithContext(ctx))
 		defer func() {
 			events := op.Finish(httpsec.HandlerOperationRes{Status: c.Response().Status})
 			if len(events) > 0 {