Set system on incoming alerts

Author: zachomedia

Gopkg.lock

@@ -38,6 +38,7 @@ var group string
 var delay int
 var alertsAddress string
 var fetchTopic string
+var system string
 
 // consumeCmd represents the consume command
 var consumeCmd = &cobra.Command{
@@ -65,6 +66,7 @@ var consumeCmd = &cobra.Command{
 			Delay:         delay,
 			AlertsService: alertsService,
 			FetchTopic:    fetchTopic,
+			System:        system,
 		}
 
 		ctx, cancel := context.WithCancel(context.Background())
@@ -102,4 +104,6 @@ func init() {
 	consumeCmd.MarkFlagRequired("alerts-address")
 
 	consumeCmd.Flags().StringVarP(&fetchTopic, "fetch-topic", "f", "", "Fetch topic")
+
+	consumeCmd.Flags().StringVarP(&system, "system", "s", "naads", "System name")
 }

cmd/consume.go

@@ -72,6 +72,9 @@ func collectAlert(ctx context.Context, conf *Config) func(ctx goka.Context, msg
 			// TODO: Handle
 		}
 
+		// Add the system
+		alert.System = conf.System
+
 		// Save the alert, if it's good
 		if (alert.Status == cap.Alert_ACTUAL || alert.Status == cap.Alert_EXCERCISE || alert.Status == cap.Alert_TEST) && (alert.MessageType == cap.Alert_ALERT || alert.MessageType == cap.Alert_UPDATE || alert.MessageType == cap.Alert_CANCEL) {
 			if _, err := conf.AlertsService.Add(ctx, &alert); err != nil {

pkg/processor/alerts.go

@@ -16,4 +16,6 @@ type Config struct {
 	FetchTopic    string
 	AlertsTopic   string
 	FetchURLs     []string
+
+	System string
 }