raven.go 714 Bytes
Newer Older
1 2 3 4 5 6
package main

import (
	"net/http"
	"os"

7
	raven "github.com/getsentry/raven-go"
8

9
	"gitlab.com/gitlab-org/gitlab-workhorse/internal/helper"
10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35
)

func wrapRaven(h http.Handler) http.Handler {
	// Use a custom environment variable (not SENTRY_DSN) to prevent
	// clashes with gitlab-rails.
	sentryDSN := os.Getenv("GITLAB_WORKHORSE_SENTRY_DSN")
	raven.SetDSN(sentryDSN) // sentryDSN may be empty

	if sentryDSN == "" {
		return h
	}

	raven.DefaultClient.SetRelease(Version)

	return http.HandlerFunc(raven.RecoveryHandler(
		func(w http.ResponseWriter, r *http.Request) {
			defer func() {
				if p := recover(); p != nil {
					helper.CleanHeadersForRaven(r)
					panic(p)
				}
			}()

			h.ServeHTTP(w, r)
		}))
}