mirror of https://github.com/muety/wakapi.git
32 lines
830 B
Go
32 lines
830 B
Go
package middlewares
|
|
|
|
import (
|
|
"context"
|
|
"github.com/getsentry/sentry-go"
|
|
sentryhttp "github.com/getsentry/sentry-go/http"
|
|
"net/http"
|
|
)
|
|
|
|
// SentryMiddleware is a wrapper around sentryhttp to include user information to traces
|
|
type SentryMiddleware struct {
|
|
handler http.Handler
|
|
}
|
|
|
|
func NewSentryMiddleware() func(http.Handler) http.Handler {
|
|
return func(h http.Handler) http.Handler {
|
|
return sentryhttp.New(sentryhttp.Options{
|
|
Repanic: true,
|
|
}).Handle(&SentryMiddleware{handler: h})
|
|
}
|
|
}
|
|
|
|
func (h *SentryMiddleware) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
ctx := context.WithValue(r.Context(), "-", "-")
|
|
h.handler.ServeHTTP(w, r.WithContext(ctx))
|
|
if hub := sentry.GetHubFromContext(ctx); hub != nil {
|
|
if user := GetPrincipal(r); user != nil {
|
|
hub.Scope().SetUser(sentry.User{ID: user.ID})
|
|
}
|
|
}
|
|
}
|