fix(lifecycle): handle status before shutdown in manager (#4261)

target · Feb 3, 2025 · f49337f · f49337f
1 parent 241dd50
commit f49337f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/app/lifecycle/manager.go b/app/lifecycle/manager.go
@@ -197,7 +197,12 @@ func (m *Manager) Run(ctx context.Context) error {
 
 	err := m.runFunc(ctx)
 	close(m.runDone)
-	<-m.shutdownDone
+	s = <-m.status
+	m.status <- s
+	if s == StatusShutdown {
+		<-m.shutdownDone
+	}
+
 	return err
 }