Compare commits

..

6 Commits

Author SHA1 Message Date
a59a278140
with callback instead 2022-09-21 22:19:54 +02:00
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
eb66692b47
With actual nonblocking one by one manager 2022-09-21 21:40:44 +02:00
44218ec4ac
with state in funcs 2022-09-10 20:05:10 +02:00
2 changed files with 52 additions and 11 deletions

View File

@ -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
}

View File

@ -4,6 +4,7 @@ import (
"context"
"os"
"sync"
"time"
)
const (
@ -30,10 +31,37 @@ func NewManager() *Manager {
}
}
func (m *Manager) RunNonBlocking(ctx context.Context) error {
go m.Run(ctx)
type ComponentsAreReady struct {
}
type CleanupFunc func(ctx context.Context) error
func (m *Manager) RunNonBlocking(ctx context.Context, readyChan chan ComponentsAreReady) (CleanupFunc, error) {
go func() error {
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 func(ctx context.Context) error {
return m.shutdown(ctx)
}, nil
}
func (m *Manager) Run(ctx context.Context) error {
@ -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()
@ -128,9 +164,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 {