secure-tunnel-server: add timeout to AWS IoT connect (#265)

seqsense · May 19, 2021 · 19f9515 · 19f9515
1 parent a217f53
commit 19f9515
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tunnel/cmd/secure-tunnel-server/main.go b/tunnel/cmd/secure-tunnel-server/main.go
@@ -64,7 +64,11 @@ func main() {
 		if err != nil {
 			log.Fatalf("Failed to create MQTT client (%s)", err.Error())
 		}
-		if _, err := cli.Connect(context.Background(),
+
+		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+		defer cancel()
+
+		if _, err := cli.Connect(ctx,
 			fmt.Sprintf("secure-tunnel-server-%d", rand.Int()),
 			mqtt.WithKeepAlive(30),
 		); err != nil {