with stop

manager.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"os"
 	"sync"
+	"time"
 )
 
 const (
@@ -33,7 +34,15 @@ func NewManager() *Manager {
 type ComponentsAreReady struct {
 }
 
-func (m *Manager) RunNonBlocking(ctx context.Context) error {
+func (m *Manager) RunNonBlocking(ctx context.Context, readyChan chan ComponentsAreReady) error {
+	go func() error {
+		defer func() {
+			ctx, _ := context.WithTimeout(context.Background(), time.Second*5)
+			err := m.shutdown(ctx)
+			if err != nil {
+				panic(err)
+			}
+		}()
 		m.initLifetime()
 		err := m.init(ctx)
 		if err != nil {
@@ -45,15 +54,15 @@ func (m *Manager) RunNonBlocking(ctx context.Context) error {
 			return err
 		}
 
+		readyChan <- ComponentsAreReady{}
+
 		err = m.wait(ctx)
 		if err != nil {
 			return err
 		}
 
-	err = m.shutdown(ctx)
+		return nil
-	if err != nil {
+	}()
-		return err
-	}
 	return nil
 }
 
@@ -157,9 +166,14 @@ func (m *Manager) initLifetime() {
 }
 
 func (m *Manager) wait(ctx context.Context) error {
-	exitCode := <-m.exitChan
+	select {
+	case exitCode := <-m.exitChan:
 		m.exitCode = exitCode
 		return nil
+	case <-ctx.Done():
+		return nil
+	}
+	return nil
 }
 
 func (m *Manager) shutdown(ctx context.Context) error {