-
Notifications
You must be signed in to change notification settings - Fork 242
feat: http client integration #876
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: master
Are you sure you want to change the base?
Changes from all commits
6b48923
fd66575
439e98b
5f4c5ae
5214ea5
ed105c1
4e55daa
15aa59a
099dad4
7d9555d
f018dde
d6f2663
6af10bf
c5ad7e4
081fc0b
a3a6367
2cb97af
ce3d4ea
1075de4
ae861f3
a5eb96f
791feb5
c912fe9
3e9a23b
912797f
0e21a00
b31001c
389ca62
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,75 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "io" | ||
| "net/http" | ||
|
|
||
| "github.com/getsentry/sentry-go" | ||
| sentryhttpclient "github.com/getsentry/sentry-go/httpclient" | ||
| ) | ||
|
|
||
| func main() { | ||
| _ = sentry.Init(sentry.ClientOptions{ | ||
| Dsn: "", | ||
| EnableTracing: true, | ||
| TracesSampleRate: 1.0, | ||
| BeforeSend: func(event *sentry.Event, hint *sentry.EventHint) *sentry.Event { | ||
| fmt.Println(event) | ||
| return event | ||
| }, | ||
| BeforeSendTransaction: func(event *sentry.Event, hint *sentry.EventHint) *sentry.Event { | ||
| fmt.Println(event) | ||
| return event | ||
| }, | ||
| Debug: true, | ||
| }) | ||
|
|
||
| // With custom HTTP client | ||
| ctx := sentry.SetHubOnContext(context.Background(), sentry.CurrentHub().Clone()) | ||
| httpClient := &http.Client{ | ||
| Transport: sentryhttpclient.NewSentryRoundTripper(nil), | ||
| } | ||
|
|
||
| err := getExamplePage(ctx, httpClient) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| // With Sentry's HTTP client | ||
| err = getExamplePage(ctx, sentryhttpclient.Client) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| } | ||
|
|
||
| func getExamplePage(ctx context.Context, httpClient *http.Client) error { | ||
| span := sentry.StartSpan(ctx, "getExamplePage") | ||
| ctx = span.Context() | ||
| defer span.Finish() | ||
|
|
||
| request, err := http.NewRequestWithContext(ctx, http.MethodGet, "https://example.com", nil) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| response, err := httpClient.Do(request) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| defer func() { | ||
| if response.Body != nil { | ||
| _ = response.Body.Close() | ||
| } | ||
| }() | ||
|
|
||
| body, err := io.ReadAll(response.Body) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| fmt.Println(string(body)) | ||
|
|
||
| return nil | ||
| } |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,139 @@ | ||
| // Package sentryhttpclient provides Sentry integration for Requests modules to enable distributed tracing between services. | ||
| // It is compatible with `net/http.RoundTripper`. | ||
| // | ||
| // import sentryhttpclient "github.com/getsentry/sentry-go/httpclient" | ||
| // | ||
| // roundTrippper := sentryhttpclient.NewSentryRoundTripper(nil, nil) | ||
| // client := &http.Client{ | ||
| // Transport: roundTripper, | ||
| // } | ||
| // | ||
| // request, err := client.Do(request) | ||
| package sentryhttpclient | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "net/http" | ||
| "strings" | ||
|
|
||
| "github.com/getsentry/sentry-go" | ||
| ) | ||
|
|
||
| // SentryRoundTripTracerOption provides a specific type in which defines the option for SentryRoundTripper. | ||
| type SentryRoundTripTracerOption func(*SentryRoundTripper) | ||
|
|
||
| // WithTracePropagationTargets configures additional trace propagation targets URL for the RoundTripper. | ||
| // Does not support regex patterns. | ||
| func WithTracePropagationTargets(targets []string) SentryRoundTripTracerOption { | ||
| return func(t *SentryRoundTripper) { | ||
| if t.tracePropagationTargets == nil { | ||
| t.tracePropagationTargets = targets | ||
| } else { | ||
| t.tracePropagationTargets = append(t.tracePropagationTargets, targets...) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| // NewSentryRoundTripper provides a wrapper to existing http.RoundTripper to have required span data and trace headers for outgoing HTTP requests. | ||
| // | ||
| // - If `nil` is passed to `originalRoundTripper`, it will use http.DefaultTransport instead. | ||
| func NewSentryRoundTripper(originalRoundTripper http.RoundTripper, opts ...SentryRoundTripTracerOption) http.RoundTripper { | ||
| if originalRoundTripper == nil { | ||
| originalRoundTripper = http.DefaultTransport | ||
| } | ||
|
|
||
| // Configure trace propagation targets | ||
| var tracePropagationTargets []string | ||
| if hub := sentry.CurrentHub(); hub != nil { | ||
| client := hub.Client() | ||
| if client != nil { | ||
| clientOptions := client.Options() | ||
| if clientOptions.TracePropagationTargets != nil { | ||
| tracePropagationTargets = clientOptions.TracePropagationTargets | ||
| } | ||
| } | ||
| } | ||
|
|
||
| t := &SentryRoundTripper{ | ||
| originalRoundTripper: originalRoundTripper, | ||
| tracePropagationTargets: tracePropagationTargets, | ||
| } | ||
|
|
||
| for _, opt := range opts { | ||
| if opt != nil { | ||
| opt(t) | ||
| } | ||
| } | ||
|
|
||
| return t | ||
| } | ||
|
|
||
| // SentryRoundTripper provides a http.RoundTripper implementation for Sentry Requests module. | ||
| type SentryRoundTripper struct { | ||
| originalRoundTripper http.RoundTripper | ||
|
|
||
| tracePropagationTargets []string | ||
| } | ||
|
|
||
| func (s *SentryRoundTripper) RoundTrip(request *http.Request) (*http.Response, error) { | ||
| // Respect trace propagation targets | ||
| if len(s.tracePropagationTargets) > 0 { | ||
| requestURL := request.URL.String() | ||
| foundMatch := false | ||
| for _, target := range s.tracePropagationTargets { | ||
| if strings.Contains(requestURL, target) { | ||
|
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Bug: Trace Propagation Target Matching VulnerabilityThe |
||
| foundMatch = true | ||
| break | ||
| } | ||
| } | ||
|
|
||
| if !foundMatch { | ||
| return s.originalRoundTripper.RoundTrip(request) | ||
| } | ||
| } | ||
|
|
||
| // Only create the `http.client` span only if there is a parent span. | ||
| parentSpan := sentry.SpanFromContext(request.Context()) | ||
| if parentSpan == nil { | ||
| if hub := sentry.GetHubFromContext(request.Context()); hub != nil { | ||
| request.Header.Add("Baggage", hub.GetBaggage()) | ||
| request.Header.Add("Sentry-Trace", hub.GetTraceparent()) | ||
| } | ||
|
|
||
| return s.originalRoundTripper.RoundTrip(request) | ||
| } | ||
|
|
||
| cleanRequestURL := request.URL.Redacted() | ||
|
|
||
| span := parentSpan.StartChild("http.client", sentry.WithDescription(fmt.Sprintf("%s %s", request.Method, cleanRequestURL))) | ||
| defer span.Finish() | ||
|
|
||
| span.SetData("http.query", request.URL.Query().Encode()) | ||
| span.SetData("http.fragment", request.URL.Fragment) | ||
| span.SetData("http.request.method", request.Method) | ||
| span.SetData("server.address", request.URL.Hostname()) | ||
| span.SetData("server.port", request.URL.Port()) | ||
|
|
||
| // Always add `Baggage` and `Sentry-Trace` headers. | ||
| request.Header.Add("Baggage", span.ToBaggage()) | ||
| request.Header.Add("Sentry-Trace", span.ToSentryTrace()) | ||
|
Comment on lines
+117
to
+119
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. To support the recently added "Tracing without Performance" feature, we should use |
||
|
|
||
| response, err := s.originalRoundTripper.RoundTrip(request) | ||
giortzisg marked this conversation as resolved.
Show resolved
Hide resolved
|
||
| if err != nil { | ||
| span.Status = sentry.SpanStatusInternalError | ||
| } | ||
|
|
||
| if response != nil { | ||
| span.Status = sentry.HTTPtoSpanStatus(response.StatusCode) | ||
| span.SetData("http.response.status_code", response.StatusCode) | ||
| span.SetData("http.response_content_length", response.ContentLength) | ||
| } | ||
|
|
||
| return response, err | ||
| } | ||
|
|
||
| // SentryHTTPClient provides a default HTTP client with SentryRoundTripper included. | ||
| // This can be used directly to perform HTTP request. | ||
| var Client = &http.Client{ | ||
| Transport: NewSentryRoundTripper(http.DefaultTransport), | ||
| } | ||
Uh oh!
There was an error while loading. Please reload this page.