Use conditional for running provider online

License: MIT
Signed-off-by: 's avatarMichael Avila <davidmichaelavila@gmail.com>
上级 95877239
...@@ -37,23 +37,24 @@ func SimpleReprovider(reproviderInterval time.Duration) interface{} { ...@@ -37,23 +37,24 @@ func SimpleReprovider(reproviderInterval time.Duration) interface{} {
} }
// SimpleProviderSys creates new provider system // SimpleProviderSys creates new provider system
func SimpleProviderSys(lc fx.Lifecycle, p provider.Provider, r provider.Reprovider) provider.System { func SimpleProviderSys(isOnline bool) interface{} {
sys := provider.NewSystem(p, r) return func(lc fx.Lifecycle, p provider.Provider, r provider.Reprovider) provider.System {
lc.Append(fx.Hook{ sys := provider.NewSystem(p, r)
OnStart: func(ctx context.Context) error {
sys.Run() if isOnline {
return nil lc.Append(fx.Hook{
}, OnStart: func(ctx context.Context) error {
OnStop: func(ctx context.Context) error { sys.Run()
return sys.Close() return nil
}, },
}) OnStop: func(ctx context.Context) error {
return sys return sys.Close()
} },
})
}
// SimpleOfflineProviderSys creates a new offline provider system return sys
func SimpleOfflineProviderSys(p provider.Provider, r provider.Reprovider) provider.System { }
return provider.NewSystem(p, r)
} }
// ONLINE/OFFLINE // ONLINE/OFFLINE
...@@ -66,7 +67,7 @@ func OnlineProviders(useStrategicProviding bool, reprovideStrategy string, repro ...@@ -66,7 +67,7 @@ func OnlineProviders(useStrategicProviding bool, reprovideStrategy string, repro
return fx.Options( return fx.Options(
SimpleProviders(reprovideStrategy, reprovideInterval), SimpleProviders(reprovideStrategy, reprovideInterval),
fx.Provide(SimpleProviderSys), fx.Provide(SimpleProviderSys(true)),
) )
} }
...@@ -78,7 +79,7 @@ func OfflineProviders(useStrategicProviding bool, reprovideStrategy string, repr ...@@ -78,7 +79,7 @@ func OfflineProviders(useStrategicProviding bool, reprovideStrategy string, repr
return fx.Options( return fx.Options(
SimpleProviders(reprovideStrategy, reprovideInterval), SimpleProviders(reprovideStrategy, reprovideInterval),
fx.Provide(SimpleOfflineProviderSys), fx.Provide(SimpleProviderSys(false)),
) )
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论