Compare commits
4 Commits
Author | SHA1 | Date | |
---|---|---|---|
5f89d83094
|
|||
f9109382cb
|
|||
eb66692b47
|
|||
44218ec4ac
|
12
builder.go
12
builder.go
@@ -3,9 +3,9 @@ package curre
|
||||
import "context"
|
||||
|
||||
type FunctionalComponent struct {
|
||||
InitFunc func(ctx context.Context) error
|
||||
StartFunc func(ctx context.Context) error
|
||||
StopFunc func(ctx context.Context) error
|
||||
InitFunc func(fc *FunctionalComponent, ctx context.Context) error
|
||||
StartFunc func(fc *FunctionalComponent, ctx context.Context) error
|
||||
StopFunc func(fc *FunctionalComponent, ctx context.Context) error
|
||||
}
|
||||
|
||||
func NewFunctionalComponent(
|
||||
@@ -16,19 +16,19 @@ func NewFunctionalComponent(
|
||||
|
||||
func (fc *FunctionalComponent) Init(ctx context.Context) error {
|
||||
if fc.InitFunc != nil {
|
||||
return fc.InitFunc(ctx)
|
||||
return fc.InitFunc(fc, ctx)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
func (fc *FunctionalComponent) Start(ctx context.Context) error {
|
||||
if fc.StartFunc != nil {
|
||||
return fc.StartFunc(ctx)
|
||||
return fc.StartFunc(fc, ctx)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
func (fc *FunctionalComponent) Stop(ctx context.Context) error {
|
||||
if fc.StopFunc != nil {
|
||||
return fc.StopFunc(ctx)
|
||||
return fc.StopFunc(fc, ctx)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
40
manager.go
40
manager.go
@@ -30,9 +30,37 @@ func NewManager() *Manager {
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Manager) RunNonBlocking(ctx context.Context) error {
|
||||
go m.Run(ctx)
|
||||
type ComponentsAreReady struct {
|
||||
}
|
||||
|
||||
func (m *Manager) RunNonBlocking(ctx context.Context, readyChan chan ComponentsAreReady) error {
|
||||
go func() error {
|
||||
defer func(ctx context.Context) {
|
||||
err := m.shutdown(ctx)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}(ctx)
|
||||
m.initLifetime()
|
||||
err := m.init(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = m.startBlocking(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
readyChan <- ComponentsAreReady{}
|
||||
|
||||
err = m.wait(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}()
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -98,6 +126,14 @@ func (m *Manager) start(ctx context.Context) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Manager) startBlocking(ctx context.Context) error {
|
||||
for _, c := range m.components {
|
||||
m.startComponent(ctx, c)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Manager) startComponent(ctx context.Context, component Component) {
|
||||
defer func() {
|
||||
err := recover()
|
||||
|
Reference in New Issue
Block a user