From 812b3ffb5d6c562fa79e00edaef5409cd053f4d0 Mon Sep 17 00:00:00 2001
From: Samuel Berthe <dev@samuel-berthe.fr>
Date: Sun, 21 Apr 2024 23:13:31 +0200
Subject: [PATCH] feat: include tracing only for the recording spans

---
 middleware.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/middleware.go b/middleware.go
index b102286..eab9827 100644
--- a/middleware.go
+++ b/middleware.go
@@ -1,6 +1,7 @@
 package sloggin
 
 import (
+	"context"
 	"log/slog"
 	"net/http"
 	"strings"
@@ -168,7 +169,7 @@ func NewWithConfig(logger *slog.Logger, config Config) gin.HandlerFunc {
 		}
 
 		// otel
-		baseAttributes = append(baseAttributes, extractTraceSpanID(c, config.WithTraceID, config.WithSpanID)...)
+		baseAttributes = append(baseAttributes, extractTraceSpanID(c.Request.Context(), config.WithTraceID, config.WithSpanID)...)
 
 		// request body
 		requestAttributes = append(requestAttributes, slog.Int("length", br.bytes))
@@ -283,12 +284,11 @@ func AddCustomAttributes(c *gin.Context, attr slog.Attr) {
 	}
 }
 
-func extractTraceSpanID(c *gin.Context, withTraceID bool, withSpanID bool) []slog.Attr {
+func extractTraceSpanID(ctx context.Context, withTraceID bool, withSpanID bool) []slog.Attr {
 	if !(withTraceID || withSpanID) {
 		return []slog.Attr{}
 	}
 
-	ctx := c.Request.Context()
 	span := trace.SpanFromContext(ctx)
 	if !span.IsRecording() {
 		return []slog.Attr{}