89 lines
2.9 KiB
Go
89 lines
2.9 KiB
Go
package cli
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/alecthomas/kingpin"
|
|
"github.com/prometheus/client_golang/api"
|
|
|
|
"github.com/prometheus/alertmanager/cli/format"
|
|
"github.com/prometheus/alertmanager/client"
|
|
"github.com/prometheus/alertmanager/pkg/parse"
|
|
)
|
|
|
|
type alertQueryCmd struct {
|
|
expired, silenced bool
|
|
matcherGroups []string
|
|
}
|
|
|
|
func configureAlertCmd(app *kingpin.Application, longHelpText map[string]string) {
|
|
var (
|
|
a = &alertQueryCmd{}
|
|
alertCmd = app.Command("alert", "View and search through current alerts")
|
|
queryCmd = alertCmd.Command("query", "View and search through current alerts").Default()
|
|
)
|
|
queryCmd.Flag("expired", "Show expired alerts as well as active").BoolVar(&a.expired)
|
|
queryCmd.Flag("silenced", "Show silenced alerts").Short('s').BoolVar(&a.silenced)
|
|
queryCmd.Arg("matcher-groups", "Query filter").StringsVar(&a.matcherGroups)
|
|
queryCmd.Action(a.queryAlerts)
|
|
longHelpText["alert"] = `View and search through current alerts.
|
|
|
|
Amtool has a simplified prometheus query syntax, but contains robust support for
|
|
bash variable expansions. The non-option section of arguments constructs a list
|
|
of "Matcher Groups" that will be used to filter your query. The following
|
|
examples will attempt to show this behaviour in action:
|
|
|
|
amtool alert query alertname=foo node=bar
|
|
|
|
This query will match all alerts with the alertname=foo and node=bar label
|
|
value pairs set.
|
|
|
|
amtool alert query foo node=bar
|
|
|
|
If alertname is omitted and the first argument does not contain a '=' or a
|
|
'=~' then it will be assumed to be the value of the alertname pair.
|
|
|
|
amtool alert query '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.`
|
|
longHelpText["alert query"] = longHelpText["alert"]
|
|
}
|
|
|
|
func (a *alertQueryCmd) queryAlerts(element *kingpin.ParseElement, ctx *kingpin.ParseContext) error {
|
|
var filterString = ""
|
|
if len(a.matcherGroups) == 1 {
|
|
// If the parser fails then we likely don't have a (=|=~|!=|!~) so lets
|
|
// assume that the user wants alertname=<arg> and prepend `alertname=`
|
|
// to the front.
|
|
_, err := parse.Matcher(a.matcherGroups[0])
|
|
if err != nil {
|
|
filterString = fmt.Sprintf("{alertname=%s}", a.matcherGroups[0])
|
|
} else {
|
|
filterString = fmt.Sprintf("{%s}", strings.Join(a.matcherGroups, ","))
|
|
}
|
|
} else if len(a.matcherGroups) > 1 {
|
|
filterString = fmt.Sprintf("{%s}", strings.Join(a.matcherGroups, ","))
|
|
}
|
|
|
|
c, err := api.NewClient(api.Config{Address: alertmanagerURL.String()})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
alertAPI := client.NewAlertAPI(c)
|
|
fetchedAlerts, err := alertAPI.List(context.Background(), filterString, a.expired, a.silenced)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
formatter, found := format.Formatters[output]
|
|
if !found {
|
|
return errors.New("unknown output formatter")
|
|
}
|
|
return formatter.FormatAlerts(fetchedAlerts)
|
|
}
|