3 Commits

Author SHA1 Message Date
4774158f24 with stop 2022-09-21 22:14:22 +02:00
5f89d83094 with defer 2022-09-21 22:08:27 +02:00
f9109382cb components signal 2022-09-21 21:45:48 +02:00

View File

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