Tweak NATS startup and readiness checking

neilalexander · Feb 11, 2025 · 543eeee · 543eeee
1 parent ff1449e
commit 543eeee
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/setup/jetstream/nats.go b/setup/jetstream/nats.go
@@ -67,18 +67,21 @@ func (s *NATSInstance) Prepare(process *process.ProcessContext, cfg *config.JetS
 		if !cfg.NoLog {
 			s.SetLogger(NewLogAdapter(), opts.Debug, opts.Trace)
 		}
-		go func() {
-			process.ComponentStarted()
-			s.Start()
-		}()
+		process.ComponentStarted()
+		go s.Start()
 		go func() {
 			<-process.WaitForShutdown()
 			s.Shutdown()
 			s.WaitForShutdown()
 			process.ComponentFinished()
 		}()
 		if !s.ReadyForConnections(time.Second * 60) {
-			logrus.Fatalln("NATS did not start in time")
+			logrus.Fatalln("NATS did not start in time, shutting down")
+			process.ShutdownDendrite()
+			s.Shutdown()
+			s.WaitForShutdown()
+			process.ComponentFinished()
+			return nil, nil
 		}
 	}