coosld/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2026-02-25 20:37:53 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Use conditional for running provider online

Browse Source 
License: MIT
Signed-off-by: Michael Avila <davidmichaelavila@gmail.com>
This commit is contained in:
Michael Avila 2019-05-06 16:16:49 -07:00
parent 9587723911
commit 23e1b1db3a
No known key found for this signature in database
GPG Key ID: EC20A5E4E97A5E1E
1 changed files with 19 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
core/node/provider.go
View File

@ -37,23 +37,24 @@ func SimpleReprovider(reproviderInterval time.Duration) interface{} {
}
// SimpleProviderSys creates new provider system
func SimpleProviderSys(lc fx.Lifecycle, p provider.Provider, r provider.Reprovider) provider.System {
sys := provider.NewSystem(p, r)
lc.Append(fx.Hook{
OnStart: func(ctx context.Context) error {
sys.Run()
return nil
},
OnStop: func(ctx context.Context) error {
return sys.Close()
},
})
return sys
}
func SimpleProviderSys(isOnline bool) interface{} {
return func(lc fx.Lifecycle, p provider.Provider, r provider.Reprovider) provider.System {
sys := provider.NewSystem(p, r)
// SimpleOfflineProviderSys creates a new offline provider system
func SimpleOfflineProviderSys(p provider.Provider, r provider.Reprovider) provider.System {
return provider.NewSystem(p, r)
if isOnline {
lc.Append(fx.Hook{
OnStart: func(ctx context.Context) error {
sys.Run()
return nil
},
OnStop: func(ctx context.Context) error {
return sys.Close()
},
})
}
return sys
}
}
// ONLINE/OFFLINE
@ -66,7 +67,7 @@ func OnlineProviders(useStrategicProviding bool, reprovideStrategy string, repro
return fx.Options(
SimpleProviders(reprovideStrategy, reprovideInterval),
fx.Provide(SimpleProviderSys),
fx.Provide(SimpleProviderSys(true)),
)
}
@ -78,7 +79,7 @@ func OfflineProviders(useStrategicProviding bool, reprovideStrategy string, repr
return fx.Options(
SimpleProviders(reprovideStrategy, reprovideInterval),
fx.Provide(SimpleOfflineProviderSys),
fx.Provide(SimpleProviderSys(false)),
)
}