From 2907d7633374cc9287beac7f982eb641e03af928 Mon Sep 17 00:00:00 2001
From: Scott Leggett <scott@sl.id.au>
Date: Wed, 16 Mar 2022 10:45:28 +0800
Subject: [PATCH] feat: add name to NATS connections

This facilitates debugging.
---
 cmd/ssh-portal/serve.go         | 1 +
 internal/sshportalapi/server.go | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/cmd/ssh-portal/serve.go b/cmd/ssh-portal/serve.go
index 3caf076f..ab6af791 100644
--- a/cmd/ssh-portal/serve.go
+++ b/cmd/ssh-portal/serve.go
@@ -36,6 +36,7 @@ func (cmd *ServeCmd) Run(log *zap.Logger) error {
 	defer stop()
 	// get nats server connection
 	nconn, err := nats.Connect(cmd.NATSServer,
+		nats.Name("ssh-portal"),
 		// exit on connection close
 		nats.ClosedHandler(func(_ *nats.Conn) {
 			log.Error("nats connection closed")
diff --git a/internal/sshportalapi/server.go b/internal/sshportalapi/server.go
index bd7f96ae..1038ead1 100644
--- a/internal/sshportalapi/server.go
+++ b/internal/sshportalapi/server.go
@@ -35,9 +35,10 @@ func ServeNATS(ctx context.Context, stop context.CancelFunc, log *zap.Logger,
 	wg.Add(1)
 	// connect to NATS server
 	nconn, err := nats.Connect(natsURL,
+		nats.Name("ssh-portal-api"),
+		// synchronise exiting ServeNATS()
 		nats.ClosedHandler(func(_ *nats.Conn) {
 			log.Error("nats connection closed")
-			// synchronise exiting ServeNATS()
 			stop()
 			wg.Done()
 		}),