trace: add GetStartOptions config (#153)

moonrhythm · Dec 18, 2023 · 5486824 · 5486824
1 parent 64ea3be
commit 5486824
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/trace/trace.go b/pkg/trace/trace.go
@@ -23,6 +23,7 @@ type Trace struct {
 	IsPublicEndpoint bool
 	FormatSpanName   func(r *http.Request) string
 	StartOptions     trace.StartOptions
+	GetStartOptions  func(r *http.Request) trace.StartOptions
 }
 
 // ServeHandler implements middleware interface
@@ -42,6 +43,7 @@ func (m Trace) ServeHandler(h http.Handler) http.Handler {
 		Propagation:      m.Propagation,
 		FormatSpanName:   m.FormatSpanName,
 		StartOptions:     m.StartOptions,
+		GetStartOptions:  m.GetStartOptions,
 		IsPublicEndpoint: m.IsPublicEndpoint,
 	}
 }