Exit when permanently disconnected from Nats (#13)

jjbubudi · Jul 31, 2019 · 3daeb42 · 3daeb42
1 parent 6047ed5
commit 3daeb42
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/cmd/publisher/publisher.go b/cmd/publisher/publisher.go
@@ -23,7 +23,14 @@ func NewPublisherCommand() *cobra.Command {
 			natsURL := viper.GetString("nats-url")
 			natsClusterID := viper.GetString("nats-cluster-id")
 			natsClientID := viper.GetString("nats-client-id")
-			nats, err := stan.Connect(natsClusterID, natsClientID, stan.NatsURL(natsURL))
+			nats, err := stan.Connect(
+				natsClusterID,
+				natsClientID,
+				stan.NatsURL(natsURL),
+				stan.SetConnectionLostHandler(func(conn stan.Conn, err error) {
+					log.Fatal(err)
+				}),
+			)
 			if err != nil {
 				log.Fatal(err)
 			}

diff --git a/cmd/server/server.go b/cmd/server/server.go
@@ -24,7 +24,14 @@ func NewServerCommand() *cobra.Command {
 			natsClusterID := viper.GetString("nats-cluster-id")
 			natsClientID := viper.GetString("nats-client-id")
 			bindAddress := viper.GetString("bind-address")
-			nats, err := stan.Connect(natsClusterID, natsClientID, stan.NatsURL(natsURL))
+			nats, err := stan.Connect(
+				natsClusterID,
+				natsClientID,
+				stan.NatsURL(natsURL),
+				stan.SetConnectionLostHandler(func(conn stan.Conn, err error) {
+					log.Fatal(err)
+				}),
+			)
 			if err != nil {
 				log.Fatal(err)
 			}