Merge 4b22695 into 2a7d29f

alerts.go

@@ -0,0 +1,245 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"regexp"
+	"strings"
+	"time"
+
+	"github.com/codegangsta/cli"
+	"github.com/fatih/color"
+	mkr "github.com/mackerelio/mackerel-client-go"
+	"github.com/mackerelio/mkr/logger"
+)
+
+var commandAlerts = cli.Command{
+	Name:  "alerts",
+	Usage: "Retrieve/Close alerts",
+	Description: `
+    Retrieve/Close alerts. Without subcommand, show all alerts.
+    Request APIs under "/api/v0/alerts". See http://help-ja.mackerel.io/entry/spec/api/v0 .
+`,
+	Action: doAlertsRetrieve,
+	Subcommands: []cli.Command{
+		{
+			Name:        "list",
+			Usage:       "list alerts",
+			Description: "Show alerts with human-readable format.",
+			Action:      doAlertsList,
+			Flags: []cli.Flag{
+				cli.StringSliceFlag{
+					Name:  "service, s",
+					Value: &cli.StringSlice{},
+					Usage: "Filter alerts by service. Multiple choice allow. ",
+				},
+				cli.StringSliceFlag{
+					Name:  "status, S",
+					Value: &cli.StringSlice{},
+					Usage: "Filter alerts by status of each host. Multiple choice allow. ",
+				},
+				cli.BoolFlag{Name: "color, c", Usage: "Colorize output. default: true"},
+			},
+		},
+		{
+			Name:        "close",
+			Usage:       "close alerts",
+			Description: "Pull monitor rules from Mackerel server and save them to a file. The file can be specified by filepath argument <file>. The default is 'monitors.json'.",
+			Action:      doAlertsClose,
+			Flags: []cli.Flag{
+				cli.StringFlag{Name: "reason, r", Value: "", Usage: "Reason of closing alert."},
+				cli.BoolFlag{Name: "verbose, v", Usage: "Verbose output mode"},
+			},
+		},
+	},
+}
+
+type alertSet struct {
+	Alert   *mkr.Alert
+	Host    *mkr.Host
+	Monitor *mkr.Monitor
+}
+
+func joinMonitorsAndHosts(client *mkr.Client, alerts []*mkr.Alert) []*alertSet {
+	hostsJSON, err := client.FindHosts(&mkr.FindHostsParam{
+		Statuses: []string{"working", "standby", "poweroff", "maintenance"},
+	})
+	logger.DieIf(err)
+
+	hosts := map[string]*mkr.Host{}
+	for _, host := range hostsJSON {
+		hosts[host.ID] = host
+	}
+
+	monitorsJSON, err := client.FindMonitors()
+	logger.DieIf(err)
+
+	monitors := map[string]*mkr.Monitor{}
+	for _, monitor := range monitorsJSON {
+		monitors[monitor.ID] = monitor
+	}
+
+	alertSets := []*alertSet{}
+	for _, alert := range alerts {
+		alertSets = append(
+			alertSets,
+			&alertSet{Alert: alert, Host: hosts[alert.HostID], Monitor: monitors[alert.MonitorID]},
+		)
+	}
+	return alertSets
+}
+
+func formatJoinedAlert(alertSet *alertSet, colorize bool) string {
+	const layout = "2006-01-02 15:04:05"
+
+	host := alertSet.Host
+	monitor := alertSet.Monitor
+	alert := alertSet.Alert
+
+	hostMsg := ""
+	if host != nil {
+		statusMsg := host.Status
+		if host.IsRetired == true {
+			statusMsg = "retired"
+		}
+		if colorize {
+			switch statusMsg {
+			case "working":
+				statusMsg = color.BlueString("working")
+			case "standby":
+				statusMsg = color.GreenString("standby")
+			case "poweroff":
+				statusMsg = "poweroff"
+			case "maintenance":
+				statusMsg = color.YellowString("maintenance")
+			}
+		}
+		hostMsg = fmt.Sprintf("%s %s", host.Name, statusMsg)
+		roleMsgs := []string{}
+		for service, roles := range host.Roles {
+			roleMsgs = append(roleMsgs, fmt.Sprintf("%s:%s", service, strings.Join(roles, ",")))
+		}
+		hostMsg += " [" + strings.Join(roleMsgs, ", ") + "]"
+	}
+
+	monitorMsg := ""
+	if monitor != nil {
+		switch monitor.Type {
+		case "connectivity":
+			monitorMsg = fmt.Sprintf("%s", monitor.Type)
+		case "host":
+			switch alert.Status {
+			case "CRITICAL":
+				monitorMsg = fmt.Sprintf("%s %.2f %s %.2f", monitor.Metric, alert.Value, monitor.Operator, monitor.Critical)
+			case "WARNING":
+				monitorMsg = fmt.Sprintf("%s %.2f %s %.2f", monitor.Metric, alert.Value, monitor.Operator, monitor.Warning)
+			default:
+				monitorMsg = fmt.Sprintf("%s %.2f %s %.2f", monitor.Metric, alert.Value, monitor.Operator, monitor.Critical)
+			}
+		case "external":
+			statusRegexp, _ := regexp.Compile("^[2345][0-9][0-9]$")
+			switch alert.Status {
+			case "CRITICAL":
+				if statusRegexp.MatchString(alert.Message) {
+					monitorMsg = fmt.Sprintf("%s %.2f > %.2f msec, status:%s", monitor.Name, alert.Value, monitor.ResponseTimeCritical, alert.Message)
+				} else {
+					monitorMsg = fmt.Sprintf("%s %.2f msec, %s", monitor.Name, alert.Value, alert.Message)
+				}
+			case "WARNING":
+				if statusRegexp.MatchString(alert.Message) {
+					monitorMsg = fmt.Sprintf("%s %.2f > %.2f msec, status:%s", monitor.Name, alert.Value, monitor.ResponseTimeWarning, alert.Message)
+				} else {
+					monitorMsg = fmt.Sprintf("%s %.2f msec, %s", monitor.Name, alert.Value, alert.Message)
+				}
+			default:
+				monitorMsg = fmt.Sprintf("%s %.2f > %.2f msec, status:%s", monitor.Name, alert.Value, monitor.ResponseTimeCritical, alert.Message)
+			}
+		case "check":
+			monitorMsg = fmt.Sprintf("%s", monitor.Type)
+		default:
+			monitorMsg = fmt.Sprintf("%s", monitor.Type)
+		}
+	}
+	statusMsg := alert.Status
+	if colorize {
+		switch alert.Status {
+		case "CRITICAL":
+			statusMsg = color.RedString("CRITICAL")
+		case "WARNING":
+			statusMsg = color.YellowString("WARNING ")
+		}
+	}
+	return fmt.Sprintf("%s %s %s %s %s", alert.ID, time.Unix(alert.OpenedAt, 0).Format(layout), statusMsg, monitorMsg, hostMsg)
+}
+
+func doAlertsRetrieve(c *cli.Context) {
+	conffile := c.GlobalString("conf")
+	client := newMackerel(conffile)
+
+	alerts, err := client.FindAlerts()
+	logger.DieIf(err)
+	PrettyPrintJSON(alerts)
+	return
+}
+
+func doAlertsList(c *cli.Context) {
+	conffile := c.GlobalString("conf")
+	filterServices := c.StringSlice("service")
+	filterStatuses := c.StringSlice("status")
+	client := newMackerel(conffile)
+
+	alerts, err := client.FindAlerts()
+	logger.DieIf(err)
+	joinedAlerts := joinMonitorsAndHosts(client, alerts)
+
+	for _, joinAlert := range joinedAlerts {
+		if len(filterServices) > 0 {
+			found := false
+			for _, filterService := range filterServices {
+				if joinAlert.Host != nil {
+					if _, ok := joinAlert.Host.Roles[filterService]; ok {
+						found = true
+					}
+				}
+			}
+			if found == false {
+				continue
+			}
+		}
+		if len(filterStatuses) > 0 {
+			found := false
+			for _, filterStatus := range filterStatuses {
+				if joinAlert.Host != nil && joinAlert.Host.Status == filterStatus {
+					found = true
+				}
+			}
+			if found == false {
+				continue
+			}
+		}
+		fmt.Println(formatJoinedAlert(joinAlert, c.Bool("color")))
+	}
+}
+
+func doAlertsClose(c *cli.Context) {
+	conffile := c.GlobalString("conf")
+	isVerbose := c.Bool("verbose")
+	argAlertIDs := c.Args()
+	reason := c.String("reason")
+
+	if len(argAlertIDs) < 1 {
+		cli.ShowCommandHelp(c, "alerts")
+		os.Exit(1)
+	}
+
+	client := newMackerel(conffile)
+	for _, alertID := range argAlertIDs {
+		alert, err := client.CloseAlert(alertID, reason)
+		logger.DieIf(err)
+
+		logger.Log("Alert closed", alertID)
+		if isVerbose == true {
+			PrettyPrintJSON(alert)
+		}
+	}
+}

commands.go

@@ -25,6 +25,7 @@ var Commands = []cli.Command{
 	commandFetch,
 	commandRetire,
 	commandMonitors,
+	commandAlerts,
 }
 
 var commandStatus = cli.Command{
@@ -354,30 +355,30 @@ func doUpdate(c *cli.Context) {
 	client := newMackerel(conffile)
 
 	for _, hostID := range argHostIDs {
-			if needUpdateHostStatus {
-				err := client.UpdateHostStatus(hostID, optStatus)
-				logger.DieIf(err)
-			}
+		if needUpdateHostStatus {
+			err := client.UpdateHostStatus(hostID, optStatus)
+			logger.DieIf(err)
+		}
 
-			if needUpdateHost {
-				host, err := client.FindHost(hostID)
-				logger.DieIf(err)
-				meta := host.Meta
-				name := ""
-				if optName == "" {
-					name = host.Name
-				} else {
-					name = optName
-				}
-				_, err = client.UpdateHost(hostID, &mkr.UpdateHostParam{
-					Name:          name,
-					RoleFullnames: optRoleFullnames,
-					Meta:          meta,
-				})
-				logger.DieIf(err)
+		if needUpdateHost {
+			host, err := client.FindHost(hostID)
+			logger.DieIf(err)
+			meta := host.Meta
+			name := ""
+			if optName == "" {
+				name = host.Name
+			} else {
+				name = optName
 			}
+			_, err = client.UpdateHost(hostID, &mkr.UpdateHostParam{
+				Name:          name,
+				RoleFullnames: optRoleFullnames,
+				Meta:          meta,
+			})
+			logger.DieIf(err)
+		}
 
-			logger.Log("updated", hostID)
+		logger.Log("updated", hostID)
 	}
 }