30 lines
739 B
Go
30 lines
739 B
Go
|
package web
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/getsentry/sentry-go"
|
||
|
|
)
|
||
|
|
|
||
|
|
type tracingTransport struct {
|
||
|
|
inner http.RoundTripper
|
||
|
|
ctx context.Context
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewTracingTransport(ctx context.Context, inner http.RoundTripper) *tracingTransport {
|
||
|
|
return &tracingTransport{inner, ctx}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (tt *tracingTransport) RoundTrip(r *http.Request) (*http.Response, error) {
|
||
|
|
span := sentry.StartSpan(tt.ctx, "authentik.go.http_request")
|
||
|
|
r.Header.Set("sentry-trace", span.ToSentryTrace())
|
||
|
|
span.Description = fmt.Sprintf("%s %s", r.Method, r.URL.String())
|
||
|
|
span.SetTag("url", r.URL.String())
|
||
|
|
span.SetTag("method", r.Method)
|
||
|
|
defer span.Finish()
|
||
|
|
res, err := tt.inner.RoundTrip(r.WithContext(span.Context()))
|
||
|
|
return res, err
|
||
|
|
}
|