IntenseWebs/grafana
3
0
Fork 0
mirror of https://github.com/grafana/grafana.git synced 2025-02-25 18:55:37 -06:00
Code Issues Packages Projects Releases Wiki Activity

Chore: Add AwaitHealthy to ModuleEngine and Server (#72215)

Browse Source 
* Chore: Add AwaitHealthy to ModuleEngine and Server

* switch from fmt.Errorf to errors.New

Co-authored-by: Will Browne <wbrowne@users.noreply.github.com>

---------

Co-authored-by: Will Browne <wbrowne@users.noreply.github.com>
This commit is contained in:
Todd Treece
2023-07-24 14:01:07 -04:00
committed by GitHub
parent c7eb7fb58a
commit 4b95f611c2
4 changed files with 27 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/modules/modules.go
View File

@@ -15,6 +15,7 @@ import (
)
type Engine interface {
AwaitHealthy(context.Context) error
Init(context.Context) error
Run(context.Context) error
Shutdown(context.Context) error
@@ -59,6 +60,14 @@ func ProvideService(
}
}
// AwaitHealthy waits for all registered modules to be healthy.
func (m *service) AwaitHealthy(ctx context.Context) error {
if m.serviceManager == nil {
return errors.New("service manager has not been initialized")
}
return m.serviceManager.AwaitHealthy(ctx)
}
// Init initializes all registered modules.
func (m *service) Init(_ context.Context) error {
var err error

14
pkg/modules/util.go
View File

@@ -27,9 +27,17 @@ func (m *MockModuleManager) RegisterInvisibleModule(name string, initFn func() (
}
type MockModuleEngine struct {
InitFunc func(context.Context) error
RunFunc func(context.Context) error
ShutdownFunc func(context.Context) error
AwaitHealthyFunc func(context.Context) error
InitFunc func(context.Context) error
RunFunc func(context.Context) error
ShutdownFunc func(context.Context) error
}
func (m *MockModuleEngine) AwaitHealthy(ctx context.Context) error {
if m.AwaitHealthyFunc != nil {
return m.AwaitHealthyFunc(ctx)
}
return nil
}
func (m *MockModuleEngine) Init(ctx context.Context) error {