Matt Bostock
83a820d91e
Rework the Airbrake hook to: a) change the interface so that the Airbrake credentials are stored in an unexported struct, `airbrakeHook`, which is instantiated using the `NewHook()` method b) send log entries where no 'error' field is set to Airbrake, using the `entry.Message` string as the message sent to Airbrake but continue to allow the passing of error types using the 'error' field Update the tests accordingly, assuring that the correct message is received by the Airbrake server. Also update the examples in the README, which would not have worked with the previous implementation of the Airbrake hook.
55 lines
1.0 KiB
Go
55 lines
1.0 KiB
Go
package airbrake
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
"github.com/tobi/airbrake-go"
|
|
)
|
|
|
|
// AirbrakeHook to send exceptions to an exception-tracking service compatible
|
|
// with the Airbrake API.
|
|
type airbrakeHook struct {
|
|
APIKey string
|
|
Endpoint string
|
|
Environment string
|
|
}
|
|
|
|
func NewHook(endpoint, apiKey, env string) *airbrakeHook {
|
|
return &airbrakeHook{
|
|
APIKey: apiKey,
|
|
Endpoint: endpoint,
|
|
Environment: env,
|
|
}
|
|
}
|
|
|
|
func (hook *airbrakeHook) Fire(entry *logrus.Entry) error {
|
|
airbrake.ApiKey = hook.APIKey
|
|
airbrake.Endpoint = hook.Endpoint
|
|
airbrake.Environment = hook.Environment
|
|
|
|
var notifyErr error
|
|
err, ok := entry.Data["error"].(error)
|
|
if ok {
|
|
notifyErr = err
|
|
} else {
|
|
notifyErr = errors.New(entry.Message)
|
|
}
|
|
|
|
airErr := airbrake.Notify(notifyErr)
|
|
if airErr != nil {
|
|
return fmt.Errorf("Failed to send error to Airbrake: %s", airErr)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (hook *airbrakeHook) Levels() []logrus.Level {
|
|
return []logrus.Level{
|
|
logrus.ErrorLevel,
|
|
logrus.FatalLevel,
|
|
logrus.PanicLevel,
|
|
}
|
|
}
|