🔥 feat: Add Context Support to RequestID Middleware

middleware/requestid/requestid.go

@@ -1,6 +1,8 @@
 package requestid
 
 import (
+	"context"
+
 	"github.com/gofiber/fiber/v3"
 )
 
@@ -36,6 +38,10 @@ func New(config ...Config) fiber.Handler {
 		// Add the request ID to locals
 		c.Locals(requestIDKey, rid)
 
+		// Add the request ID to UserContext
+		ctx := context.WithValue(c.UserContext(), requestIDKey, rid)
+		c.SetUserContext(ctx)
+
 		// Continue stack
 		return c.Next()
 	}
@@ -49,3 +55,13 @@ func FromContext(c fiber.Ctx) string {
 	}
 	return ""
 }
+
+// FromUserContext returns the request ID from the UserContext.
+// If there is no request ID, an empty string is returned.
+// Compared to Local, UserContext is more suitable for transmitting requests between microservices
+func FromUserContext(ctx context.Context) string {
+	if rid, ok := ctx.Value(requestIDKey).(string); ok {
+		return rid
+	}
+	return ""
+}

middleware/requestid/requestid_test.go

@@ -74,3 +74,27 @@ func Test_RequestID_FromContext(t *testing.T) {
 	require.NoError(t, err)
 	require.Equal(t, reqID, ctxVal)
 }
+
+// go test -run Test_RequestID_FromUserContext
+func Test_RequestID_FromUserContext(t *testing.T) {
+	t.Parallel()
+	reqID := "ThisIsARequestId"
+
+	app := fiber.New()
+	app.Use(New(Config{
+		Generator: func() string {
+			return reqID
+		},
+	}))
+
+	var ctxVal string
+
+	app.Use(func(c fiber.Ctx) error {
+		ctxVal = FromUserContext(c.UserContext())
+		return c.Next()
+	})
+
+	_, err := app.Test(httptest.NewRequest(fiber.MethodGet, "/", nil))
+	require.NoError(t, err)
+	require.Equal(t, reqID, ctxVal)
+}