Merge pull request #30 from uselagoon/log-disconnect

feat: log NATS disconnects
uselagoon · Mar 14, 2022 · 4ee3409 · 4ee3409
2 parents 7d8dbad + 99d1cfa
commit 4ee3409
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/cmd/ssh-portal/serve.go b/cmd/ssh-portal/serve.go
@@ -38,7 +38,14 @@ func (cmd *ServeCmd) Run(log *zap.Logger) error {
 	nc, err := nats.Connect(cmd.NATSServer,
 		// exit on connection close
 		nats.ClosedHandler(func(_ *nats.Conn) {
+			log.Error("nats connection closed")
 			stop()
+		}),
+		nats.DisconnectErrHandler(func(_ *nats.Conn, err error) {
+			log.Warn("nats disconnected", zap.Error(err))
+		}),
+		nats.ReconnectHandler(func(nc *nats.Conn) {
+			log.Info("nats reconnected", zap.String("url", nc.ConnectedUrl()))
 		}))
 	if err != nil {
 		return fmt.Errorf("couldn't connect to NATS server: %v", err)

diff --git a/internal/sshportalapi/server.go b/internal/sshportalapi/server.go
@@ -37,8 +37,15 @@ func ServeNATS(ctx context.Context, stop context.CancelFunc, log *zap.Logger,
 	nc, err := nats.Connect(natsURL,
 		// synchronise exiting ServeNATS()
 		nats.ClosedHandler(func(_ *nats.Conn) {
+			log.Error("nats connection closed")
 			stop()
 			wg.Done()
+		}),
+		nats.DisconnectErrHandler(func(_ *nats.Conn, err error) {
+			log.Warn("nats disconnected", zap.Error(err))
+		}),
+		nats.ReconnectHandler(func(nc *nats.Conn) {
+			log.Info("nats reconnected", zap.String("url", nc.ConnectedUrl()))
 		}))
 	if err != nil {
 		return fmt.Errorf("couldn't connect to NATS server: %v", err)