chore: more agent debugging

hm-edu · Oct 26, 2023 · 4aa6cca · 4aa6cca
1 parent b83d924
commit 4aa6cca
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cmd/step-agent/main.go b/cmd/step-agent/main.go
@@ -72,7 +72,7 @@ var agent = cli.Command{
 		}
 
 		// Subscribe to topic
-		client.Subscribe(fmt.Sprintf("%s/jobs", c.String("organization")), 0, func(client mqtt.Client, msg mqtt.Message) {
+		token := client.Subscribe(fmt.Sprintf("%s/jobs", c.String("organization")), 0, func(client mqtt.Client, msg mqtt.Message) {
 			logrus.Infof("received message on topic %s", msg.Topic())
 			logrus.Infof("message: %s", msg.Payload())
 
@@ -125,6 +125,10 @@ var agent = cli.Command{
 
 		})
 
+		if token.WaitTimeout(30*time.Second) && token.Error() != nil {
+			logrus.WithError(token.Error()).Warn("subscribing failed")
+		}
+
 		return nil
 	},
 }