Merge pull request #633 from overmindtech/heartbeart_quality_of_life

(fix) give more information about heartbeat
overmindtech · Sep 28, 2024 · ab5978f · ab5978f
2 parents 737a96c + 6cdfffe
commit ab5978f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/cmd/root.go b/cmd/root.go
@@ -152,6 +152,7 @@ var rootCmd = &cobra.Command{
 			}
 		} else if natsJWT != "" || natsNKeySeed != "" {
 			natsTokenClient, err = createTokenClient(natsJWT, natsNKeySeed)
+			log.Info("Using NATS authentication, no heartbeat will be sent")
 
 			if err != nil {
 				log.WithError(err).Fatal("Error validating NATS authentication info")

diff --git a/proc/proc.go b/proc/proc.go
@@ -553,7 +553,10 @@ func InitializeAwsSourceEngine(ctx context.Context, name string, version string,
 			startupErrorMutex.Lock()
 			startupError = err
 			startupErrorMutex.Unlock()
-			_ = e.SendHeartbeat(ctx) // Send the error immediately
+			brokenHeart := e.SendHeartbeat(ctx) // Send the error immediately
+			if brokenHeart != nil {
+				log.WithError(brokenHeart).Error("Error sending heartbeat")
+			}
 
 			if err != nil {
 				log.WithError(err).Debug("Error initializing sources")