2017-04-20 09:04:17 +00:00
|
|
|
package cli
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"encoding/json"
|
|
|
|
"errors"
|
|
|
|
"fmt"
|
|
|
|
"net/http"
|
|
|
|
"os/user"
|
|
|
|
"path"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"github.com/prometheus/alertmanager/types"
|
2017-11-11 13:51:33 +00:00
|
|
|
"github.com/prometheus/common/model"
|
2017-04-20 09:04:17 +00:00
|
|
|
"github.com/spf13/cobra"
|
|
|
|
flag "github.com/spf13/pflag"
|
|
|
|
"github.com/spf13/viper"
|
|
|
|
)
|
|
|
|
|
|
|
|
type addResponse struct {
|
|
|
|
Status string `json:"status"`
|
2017-11-12 16:43:48 +00:00
|
|
|
Data struct {
|
2017-04-20 09:04:17 +00:00
|
|
|
SilenceID string `json:"silenceId"`
|
|
|
|
} `json:"data,omitempty"`
|
|
|
|
ErrorType string `json:"errorType,omitempty"`
|
|
|
|
Error string `json:"error,omitempty"`
|
|
|
|
}
|
|
|
|
|
|
|
|
var addFlags *flag.FlagSet
|
|
|
|
var addCmd = &cobra.Command{
|
|
|
|
Use: "add",
|
|
|
|
Short: "Add silence",
|
|
|
|
Long: `Add a new alertmanager silence
|
|
|
|
|
|
|
|
Amtool uses a simplified prometheus syntax to represent silences. The
|
|
|
|
non-option section of arguments constructs a list of "Matcher Groups"
|
|
|
|
that will be used to create a number of silences. The following examples
|
|
|
|
will attempt to show this behaviour in action:
|
|
|
|
|
|
|
|
amtool silence add alertname=foo node=bar
|
|
|
|
|
|
|
|
This statement will add a silence that matches alerts with the
|
|
|
|
alertname=foo and node=bar label value pairs set.
|
|
|
|
|
|
|
|
amtool silence add foo node=bar
|
|
|
|
|
|
|
|
If alertname is ommited and the first argument does not contain a '=' or a
|
|
|
|
'=~' then it will be assumed to be the value of the alertname pair.
|
|
|
|
|
|
|
|
amtool silence add 'alertname=~foo.*'
|
|
|
|
|
|
|
|
As well as direct equality, regex matching is also supported. The '=~' syntax
|
|
|
|
(similar to prometheus) is used to represent a regex match. Regex matching
|
|
|
|
can be used in combination with a direct match.
|
|
|
|
`,
|
2017-04-21 12:50:02 +00:00
|
|
|
Run: CommandWrapper(add),
|
2017-04-20 09:04:17 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func init() {
|
2017-10-09 08:10:27 +00:00
|
|
|
var username string
|
|
|
|
|
|
|
|
user, err := user.Current()
|
|
|
|
if err != nil {
|
|
|
|
fmt.Printf("failed to get the current user, specify one with --author: %v\n", err)
|
|
|
|
} else {
|
|
|
|
username = user.Username
|
|
|
|
}
|
|
|
|
|
|
|
|
addCmd.Flags().StringP("author", "a", username, "Username for CreatedBy field")
|
2017-04-20 09:04:17 +00:00
|
|
|
addCmd.Flags().StringP("expires", "e", "1h", "Duration of silence (100h)")
|
|
|
|
addCmd.Flags().String("expire-on", "", "Expire at a certain time (Overwrites expires) RFC3339 format 2006-01-02T15:04:05Z07:00")
|
|
|
|
addCmd.Flags().StringP("comment", "c", "", "A comment to help describe the silence")
|
|
|
|
viper.BindPFlag("author", addCmd.Flags().Lookup("author"))
|
|
|
|
viper.BindPFlag("expires", addCmd.Flags().Lookup("expires"))
|
|
|
|
viper.BindPFlag("comment", addCmd.Flags().Lookup("comment"))
|
|
|
|
viper.SetDefault("comment_required", false)
|
|
|
|
addFlags = addCmd.Flags()
|
|
|
|
}
|
|
|
|
|
|
|
|
func add(cmd *cobra.Command, args []string) error {
|
|
|
|
var err error
|
|
|
|
|
|
|
|
matchers, err := parseMatchers(args)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(matchers) < 1 {
|
2017-11-12 16:43:48 +00:00
|
|
|
return fmt.Errorf("no matchers specified")
|
2017-04-20 09:04:17 +00:00
|
|
|
}
|
|
|
|
|
2017-11-01 22:08:34 +00:00
|
|
|
expireOn, err := addFlags.GetString("expire-on")
|
2017-04-20 09:04:17 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
expires := viper.GetString("expires")
|
|
|
|
var endsAt time.Time
|
|
|
|
|
2017-11-01 22:08:34 +00:00
|
|
|
if expireOn != "" {
|
|
|
|
endsAt, err = time.Parse(time.RFC3339, expireOn)
|
2017-04-20 09:04:17 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
} else {
|
2017-11-11 13:51:33 +00:00
|
|
|
duration, err := model.ParseDuration(expires)
|
2017-04-20 09:04:17 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2017-11-11 13:51:33 +00:00
|
|
|
if duration == 0 {
|
|
|
|
return fmt.Errorf("silence duration must be greater than 0")
|
|
|
|
}
|
|
|
|
endsAt = time.Now().UTC().Add(time.Duration(duration))
|
2017-04-20 09:04:17 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
author := viper.GetString("author")
|
|
|
|
comment := viper.GetString("comment")
|
2017-11-01 22:08:34 +00:00
|
|
|
commentRequired := viper.GetBool("comment_required")
|
2017-04-20 09:04:17 +00:00
|
|
|
|
2017-11-01 22:08:34 +00:00
|
|
|
if commentRequired && comment == "" {
|
2017-11-12 16:43:48 +00:00
|
|
|
return errors.New("comment required by config")
|
2017-04-20 09:04:17 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
typeMatchers, err := TypeMatchers(matchers)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
silence := types.Silence{
|
|
|
|
Matchers: typeMatchers,
|
|
|
|
StartsAt: time.Now().UTC(),
|
|
|
|
EndsAt: endsAt,
|
|
|
|
CreatedBy: author,
|
|
|
|
Comment: comment,
|
|
|
|
}
|
|
|
|
|
2017-12-07 12:12:00 +00:00
|
|
|
silenceId, err := addSilence(&silence)
|
2017-04-20 09:04:17 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2017-12-07 12:12:00 +00:00
|
|
|
|
|
|
|
_, err = fmt.Println(silenceId)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
func addSilence(silence *types.Silence) (string, error) {
|
|
|
|
u, err := GetAlertmanagerURL()
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
2017-04-20 09:04:17 +00:00
|
|
|
u.Path = path.Join(u.Path, "/api/v1/silences")
|
|
|
|
|
|
|
|
buf := bytes.NewBuffer([]byte{})
|
|
|
|
err = json.NewEncoder(buf).Encode(silence)
|
|
|
|
if err != nil {
|
2017-12-07 12:12:00 +00:00
|
|
|
return "", err
|
2017-04-20 09:04:17 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
res, err := http.Post(u.String(), "application/json", buf)
|
|
|
|
if err != nil {
|
2017-12-07 12:12:00 +00:00
|
|
|
return "", err
|
2017-04-20 09:04:17 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
defer res.Body.Close()
|
|
|
|
response := addResponse{}
|
|
|
|
err = json.NewDecoder(res.Body).Decode(&response)
|
|
|
|
if err != nil {
|
2017-12-07 12:12:00 +00:00
|
|
|
return "", fmt.Errorf("unable to parse silence json response from %s", u.String())
|
2017-04-20 09:04:17 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if response.Status == "error" {
|
2017-12-07 12:12:00 +00:00
|
|
|
return "", fmt.Errorf("[%s] %s", response.ErrorType, response.Error)
|
2017-04-20 09:04:17 +00:00
|
|
|
}
|
2017-12-07 12:12:00 +00:00
|
|
|
|
|
|
|
return response.Data.SilenceID, nil
|
2017-04-20 09:04:17 +00:00
|
|
|
}
|