matterbridge/vendor/gopkg.in/gemnasium/logrus-airbrake-hook.v2/airbrake.go

72 lines
1.5 KiB
Go
Raw Normal View History

2017-03-25 19:45:10 +00:00
package airbrake // import "gopkg.in/gemnasium/logrus-airbrake-hook.v2"
import (
"errors"
"fmt"
"net/http"
"os"
"github.com/Sirupsen/logrus"
"gopkg.in/airbrake/gobrake.v2"
)
// AirbrakeHook to send exceptions to an exception-tracking service compatible
// with the Airbrake API.
type airbrakeHook struct {
Airbrake *gobrake.Notifier
}
func NewHook(projectID int64, apiKey, env string) *airbrakeHook {
airbrake := gobrake.NewNotifier(projectID, apiKey)
airbrake.AddFilter(func(notice *gobrake.Notice) *gobrake.Notice {
if env == "development" {
return nil
}
notice.Context["environment"] = env
return notice
})
hook := &airbrakeHook{
Airbrake: airbrake,
}
return hook
}
func (hook *airbrakeHook) Fire(entry *logrus.Entry) error {
var notifyErr error
err, ok := entry.Data["error"].(error)
if ok {
notifyErr = err
} else {
notifyErr = errors.New(entry.Message)
}
var req *http.Request
for k, v := range entry.Data {
if r, ok := v.(*http.Request); ok {
req = r
delete(entry.Data, k)
break
}
}
notice := hook.Airbrake.Notice(notifyErr, req, 3)
for k, v := range entry.Data {
notice.Context[k] = fmt.Sprintf("%s", v)
}
hook.sendNotice(notice)
return nil
}
func (hook *airbrakeHook) sendNotice(notice *gobrake.Notice) {
if _, err := hook.Airbrake.SendNotice(notice); err != nil {
fmt.Fprintf(os.Stderr, "Failed to send error to Airbrake: %v\n", err)
}
}
func (hook *airbrakeHook) Levels() []logrus.Level {
return []logrus.Level{
logrus.ErrorLevel,
logrus.FatalLevel,
logrus.PanicLevel,
}
}