lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-21 22:43:52 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

add Get method to the persistence service facade

Browse Source
This commit is contained in:
c9s 2021-02-21 16:55:45 +08:00
parent 21b092037e
commit a8516edb98
2 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/bbgo/environment.go
View File

@ -536,12 +536,6 @@ func getSessionSymbols(session *ExchangeSession, defaultSymbols ...string) ([]st
} }
func (environ *Environment) ConfigureNotificationSystem(userConfig *Config) error { func (environ *Environment) ConfigureNotificationSystem(userConfig *Config) error {
// Configure persistence service, by default we will use memory service
var persistence service.PersistenceService = environ.PersistenceServiceFacade.Memory
if environ.PersistenceServiceFacade.Redis != nil {
persistence = environ.PersistenceServiceFacade.Redis
}
environ.Notifiability = Notifiability{ environ.Notifiability = Notifiability{
SymbolChannelRouter: NewPatternChannelRouter(nil), SymbolChannelRouter: NewPatternChannelRouter(nil),
SessionChannelRouter: NewPatternChannelRouter(nil), SessionChannelRouter: NewPatternChannelRouter(nil),
@ -562,6 +556,7 @@ func (environ *Environment) ConfigureNotificationSystem(userConfig *Config) erro
} }
} }
persistence := environ.PersistenceServiceFacade.Get()
telegramBotToken := viper.GetString("telegram-bot-token") telegramBotToken := viper.GetString("telegram-bot-token")
if len(telegramBotToken) > 0 { if len(telegramBotToken) > 0 {
tt := strings.Split(telegramBotToken, ":") tt := strings.Split(telegramBotToken, ":")

14
pkg/service/persistence_facade.go
View File

@ -5,3 +5,17 @@ type PersistenceServiceFacade struct {
Json *JsonPersistenceService Json *JsonPersistenceService
Memory *MemoryService Memory *MemoryService
} }
// Get returns the preferred persistence service by fallbacks
// Redis will be preferred at the first position.
func (facade *PersistenceServiceFacade) Get() PersistenceService {
if facade.Redis != nil {
return facade.Redis
}
if facade.Json != nil {
return facade.Json
}
return facade.Memory
}