(fix) give more information about heartbeat

overmindtech · Sep 27, 2024 · 6cdfffe · 6cdfffe
1 parent 737a96c
commit 6cdfffe
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")