with callback instead

This commit is contained in:
Kasper Juul Hermansen 2022-09-21 22:19:54 +02:00
parent 4774158f24
commit a59a278140
Signed by: kjuulh
GPG Key ID: 57B6E1465221F912
1 changed files with 7 additions and 9 deletions

View File

@ -34,15 +34,10 @@ func NewManager() *Manager {
type ComponentsAreReady struct {
}
func (m *Manager) RunNonBlocking(ctx context.Context, readyChan chan ComponentsAreReady) error {
type CleanupFunc func(ctx context.Context) error
func (m *Manager) RunNonBlocking(ctx context.Context, readyChan chan ComponentsAreReady) (CleanupFunc, 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 {
@ -63,7 +58,10 @@ func (m *Manager) RunNonBlocking(ctx context.Context, readyChan chan ComponentsA
return nil
}()
return nil
return func(ctx context.Context) error {
return m.shutdown(ctx)
}, nil
}
func (m *Manager) Run(ctx context.Context) error {