61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
|
package notify
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"golang.org/x/net/context"
|
||
|
|
"golang.org/x/net/context/ctxhttp"
|
||
|
|
|
||
|
|
"github.com/prometheus/alertmanager/config"
|
||
|
|
"github.com/prometheus/alertmanager/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Webhook struct {
|
||
|
|
URL string
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewWebhook(conf *config.WebhookConfig) *Webhook {
|
||
|
|
return &Webhook{URL: conf.URL}
|
||
|
|
}
|
||
|
|
|
||
|
|
type WebhookMessage struct {
|
||
|
|
Version string `json:"version"`
|
||
|
|
Status types.AlertStatus `json:"status"`
|
||
|
|
Alerts []*types.Alert `json:"alert"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *Webhook) Notify(ctx context.Context, alerts ...*types.Alert) error {
|
||
|
|
msg := &WebhookMessage{
|
||
|
|
Version: "1",
|
||
|
|
Status: types.AlertResolved,
|
||
|
|
Alerts: alerts,
|
||
|
|
}
|
||
|
|
for _, a := range alerts {
|
||
|
|
if !a.Resolved() {
|
||
|
|
msg.Status = types.AlertFiring
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
var buf bytes.Buffer
|
||
|
|
if err := json.NewEncoder(&buf).Encode(msg); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO(fabxc): implement retrying as long as context is not canceled.
|
||
|
|
resp, err := ctxhttp.Post(ctx, http.DefaultClient, w.URL, "application/json", &buf)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
resp.Body.Close()
|
||
|
|
|
||
|
|
if resp.StatusCode/100 != 2 {
|
||
|
|
return fmt.Errorf("unexpected status code %v", resp.StatusCode)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|