35 lines
995 B
Go
35 lines
995 B
Go
|
package sentry
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/getsentry/sentry-go"
|
||
|
|
)
|
||
|
|
|
||
|
|
type contextSentryNoSample struct{}
|
||
|
|
|
||
|
|
func SentryNoSample(handler func(rw http.ResponseWriter, r *http.Request)) func(http.ResponseWriter, *http.Request) {
|
||
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
ctx := context.WithValue(r.Context(), contextSentryNoSample{}, true)
|
||
|
|
handler(w, r.WithContext(ctx))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func SentryNoSampleMiddleware(h http.Handler) http.Handler {
|
||
|
|
return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
||
|
|
ctx := context.WithValue(r.Context(), contextSentryNoSample{}, true)
|
||
|
|
h.ServeHTTP(rw, r.WithContext(ctx))
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func SamplerFunc(defaultRate float64) sentry.TracesSamplerFunc {
|
||
|
|
return sentry.TracesSamplerFunc(func(ctx sentry.SamplingContext) sentry.Sampled {
|
||
|
|
data, ok := ctx.Span.Context().Value(contextSentryNoSample{}).(bool)
|
||
|
|
if data && ok {
|
||
|
|
return sentry.SampledFalse
|
||
|
|
}
|
||
|
|
return sentry.UniformTracesSampler(defaultRate).Sample(ctx)
|
||
|
|
})
|
||
|
|
}
|