feat: add GetRequestIDFromContext

samber · Jul 7, 2024 · cc0678b · cc0678b
1 parent de72e88
commit cc0678b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/middleware.go b/middleware.go
@@ -254,7 +254,12 @@ func NewWithConfig(logger *slog.Logger, config Config) func(http.Handler) http.H
 
 // GetRequestID returns the request identifier
 func GetRequestID(r *http.Request) string {
-	requestID := r.Context().Value(requestIDCtxKey)
+	return GetRequestIDFromContext(r.Context())
+}
+
+// GetRequestID returns the request identifier from the context
+func GetRequestIDFromContext(ctx context.Context) string {
+	requestID := ctx.Value(requestIDCtxKey)
 	if id, ok := requestID.(string); ok {
 		return id
 	}