Fix state loader to only use ES state loader with ES output

andrewvc · andrewvc · commit 6ecac3e4d49b · 2022-09-08T22:23:00.000-05:00
diff --git a/heartbeat/beater/heartbeat.go b/heartbeat/beater/heartbeat.go
@@ -67,16 +67,17 @@ func New(b *beat.Beat, rawConfig *conf.C) (beat.Beater, error) {
 		return nil, fmt.Errorf("error reading config file: %w", err)
 	}
 
-	// Connect to ES and setup the State loader
-	esc, err := getESClient(b.Config.Output.Config())
-	if err != nil {
-		return nil, err
-	}
-	var stateLoader monitorstate.StateLoader
-	if esc != nil {
-		stateLoader = monitorstate.MakeESLoader(esc, "synthetics-*,heartbeat-*", parsedConfig.Location)
-	} else {
-		stateLoader = monitorstate.NilStateLoader
+	stateLoader := monitorstate.NilStateLoader
+
+	if b.Config.Output.Name() == "elasticsearch" {
+		// Connect to ES and setup the State loader
+		esc, err := getESClient(b.Config.Output.Config())
+		if err != nil {
+			return nil, err
+		}
+		if esc != nil {
+			stateLoader = monitorstate.MakeESLoader(esc, "synthetics-*,heartbeat-*", parsedConfig.Location)
+		}
 	}
 
 	limit := parsedConfig.Scheduler.Limit
