diff --git a/cli/silence.go b/cli/silence.go index d7d2a6ac92..494126f3eb 100644 --- a/cli/silence.go +++ b/cli/silence.go @@ -31,4 +31,5 @@ func init() { silenceCmd.AddCommand(addCmd) silenceCmd.AddCommand(expireCmd) silenceCmd.AddCommand(queryCmd) + silenceCmd.AddCommand(importCmd) } diff --git a/cli/silence_add.go b/cli/silence_add.go index 90146099a1..6d784979ba 100644 --- a/cli/silence_add.go +++ b/cli/silence_add.go @@ -134,34 +134,43 @@ func add(cmd *cobra.Command, args []string) error { Comment: comment, } - u, err := GetAlertmanagerURL() + silenceId, err := addSilence(&silence) if err != nil { return err } + + _, err = fmt.Println(silenceId) + return err +} + +func addSilence(silence *types.Silence) (string, error) { + u, err := GetAlertmanagerURL() + if err != nil { + return "", err + } u.Path = path.Join(u.Path, "/api/v1/silences") buf := bytes.NewBuffer([]byte{}) err = json.NewEncoder(buf).Encode(silence) if err != nil { - return err + return "", err } res, err := http.Post(u.String(), "application/json", buf) if err != nil { - return err + return "", err } defer res.Body.Close() response := addResponse{} err = json.NewDecoder(res.Body).Decode(&response) if err != nil { - return fmt.Errorf("unable to parse silence json response from %s", u.String()) + return "", fmt.Errorf("unable to parse silence json response from %s", u.String()) } if response.Status == "error" { - fmt.Printf("[%s] %s\n", response.ErrorType, response.Error) - } else { - fmt.Println(response.Data.SilenceID) + return "", fmt.Errorf("[%s] %s", response.ErrorType, response.Error) } - return nil + + return response.Data.SilenceID, nil } diff --git a/cli/silence_import.go b/cli/silence_import.go new file mode 100644 index 0000000000..ce41d35b73 --- /dev/null +++ b/cli/silence_import.go @@ -0,0 +1,127 @@ +package cli + +import ( + "encoding/json" + "fmt" + "os" + + "github.com/pkg/errors" + "github.com/prometheus/alertmanager/types" + "github.com/prometheus/common/log" + "github.com/spf13/cobra" + flag "github.com/spf13/pflag" +) + +var importFlags *flag.FlagSet +var importCmd = &cobra.Command{ + Use: "import [JSON file]", + Short: "Import silences", + Long: `Import alertmanager silences from JSON file or stdin + + This command can be used to bulk import silences from a JSON file + created by query command. For example: + + amtool silence query -o json foo > foo.json + amtool silence import foo.json + + JSON data can also come from stdin if no param is specified. + `, + Args: cobra.MaximumNArgs(1), + Run: CommandWrapper(bulkImport), +} + +func init() { + importCmd.Flags().BoolP("force", "f", false, "Force adding new silences even if it already exists") + importCmd.Flags().IntP("worker", "w", 8, "Number of concurrent workers to use for import") + importFlags = importCmd.Flags() +} + +func addSilenceWorker(silences <-chan *types.Silence, errs chan<- error) { + for s := range silences { + silenceId, err := addSilence(s) + if err != nil && err.Error() == "[bad_data] not found" { + // silence doesn't exists yet, retry to create as a new one + s.ID = "" + silenceId, err = addSilence(s) + } + + if err != nil { + log.Errorf("error adding silence %v: %v", s.ID, err) + } else { + fmt.Println(silenceId) + } + errs <- err + } +} + +func bulkImport(cmd *cobra.Command, args []string) error { + force, err := importFlags.GetBool("force") + if err != nil { + return err + } + + input := os.Stdin + if len(args) == 1 { + input, err = os.Open(args[0]) + if err != nil { + return err + } + defer input.Close() + } + + dec := json.NewDecoder(input) + + // read open square bracket + _, err = dec.Token() + if err != nil { + return errors.Wrap(err, "couldn't unmarshal input data, is it JSON?") + } + + silences := make(chan *types.Silence, 100) + errs := make(chan error, 100) + workers, err := importFlags.GetInt("worker") + if err != nil { + return err + } + for w := 0; w < workers; w++ { + go addSilenceWorker(silences, errs) + } + + count := 0 + for dec.More() { + var s types.Silence + err := dec.Decode(&s) + if err != nil { + return errors.Wrap(err, "couldn't unmarshal input data, is it JSON?") + } + + if force { + // reset the silence ID so Alertmanager will always create new silence + s.ID = "" + } + + silences <- &s + count++ + } + close(silences) + + // read closing bracket + _, err = dec.Token() + if err != nil { + return errors.Wrap(err, "invalid JSON") + } + + errCount := 0 + for i := 0; i < count; i++ { + err = <-errs + if err != nil { + errCount++ + } + } + + if errCount > 0 { + return fmt.Errorf("couldn't import %v out of %v silences", errCount, count) + } + + return nil +}