Compare commits
7 Commits
Author | SHA1 | Date | |
---|---|---|---|
5f89d83094
|
|||
f9109382cb
|
|||
eb66692b47
|
|||
44218ec4ac
|
|||
a2c53271e4
|
|||
3b3c0e6118
|
|||
78c0c309d7
|
34
builder.go
Normal file
34
builder.go
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
package curre
|
||||||
|
|
||||||
|
import "context"
|
||||||
|
|
||||||
|
type FunctionalComponent struct {
|
||||||
|
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(
|
||||||
|
fc *FunctionalComponent,
|
||||||
|
) Component {
|
||||||
|
return fc
|
||||||
|
}
|
||||||
|
|
||||||
|
func (fc *FunctionalComponent) Init(ctx context.Context) error {
|
||||||
|
if fc.InitFunc != nil {
|
||||||
|
return fc.InitFunc(fc, ctx)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
func (fc *FunctionalComponent) Start(ctx context.Context) error {
|
||||||
|
if fc.StartFunc != nil {
|
||||||
|
return fc.StartFunc(fc, ctx)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
func (fc *FunctionalComponent) Stop(ctx context.Context) error {
|
||||||
|
if fc.StopFunc != nil {
|
||||||
|
return fc.StopFunc(fc, ctx)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
@@ -2,6 +2,6 @@ package curre
|
|||||||
|
|
||||||
import "context"
|
import "context"
|
||||||
|
|
||||||
type Closer interface {
|
type Stopper interface {
|
||||||
Close(ctx context.Context) error
|
Stop(ctx context.Context) error
|
||||||
}
|
}
|
||||||
|
@@ -3,5 +3,5 @@ package curre
|
|||||||
type Component interface {
|
type Component interface {
|
||||||
Initer
|
Initer
|
||||||
Starter
|
Starter
|
||||||
Closer
|
Stopper
|
||||||
}
|
}
|
||||||
|
@@ -28,6 +28,6 @@ func (hs *httpServer) Start(ctx context.Context) error {
|
|||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
func (hs *httpServer) Close(ctx context.Context) error {
|
func (hs *httpServer) Stop(ctx context.Context) error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
42
manager.go
42
manager.go
@@ -30,9 +30,37 @@ func NewManager() *Manager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *Manager) RunNonBlocking(ctx context.Context) error {
|
type ComponentsAreReady struct {
|
||||||
go m.Run(ctx)
|
}
|
||||||
|
|
||||||
|
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
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -98,6 +126,14 @@ func (m *Manager) start(ctx context.Context) error {
|
|||||||
return nil
|
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) {
|
func (m *Manager) startComponent(ctx context.Context, component Component) {
|
||||||
defer func() {
|
defer func() {
|
||||||
err := recover()
|
err := recover()
|
||||||
@@ -139,7 +175,7 @@ func (m *Manager) shutdown(ctx context.Context) error {
|
|||||||
|
|
||||||
go func(ctx context.Context) {
|
go func(ctx context.Context) {
|
||||||
for _, c := range closers {
|
for _, c := range closers {
|
||||||
c.Close(ctx)
|
c.Stop(ctx)
|
||||||
}
|
}
|
||||||
|
|
||||||
shutdownChan <- struct{}{}
|
shutdownChan <- struct{}{}
|
||||||
|
Reference in New Issue
Block a user