lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-26 00:35:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

bbgo: refactor ConfigurePersistence

Browse Source
This commit is contained in:
c9s 2022-10-05 18:46:26 +08:00
parent 4caa457fbe
commit 673304bcf1
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
1 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
pkg/bbgo/persistence.go
View File

@ -6,6 +6,7 @@ import (
"reflect" "reflect"
"github.com/codingconcepts/env" "github.com/codingconcepts/env"
"github.com/pkg/errors"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
"github.com/c9s/bbgo/pkg/dynamic" "github.com/c9s/bbgo/pkg/dynamic"
@ -73,27 +74,40 @@ func storePersistenceFields(obj interface{}, id string, persistence service.Pers
}) })
} }
func ConfigurePersistence(conf *PersistenceConfig) error { func NewPersistenceServiceFacade(conf *PersistenceConfig) (*service.PersistenceServiceFacade, error) {
facade := &service.PersistenceServiceFacade{
Memory: service.NewMemoryService(),
}
if conf.Redis != nil { if conf.Redis != nil {
if err := env.Set(conf.Redis); err != nil { if err := env.Set(conf.Redis); err != nil {
return err return nil, err
} }
redisPersistence := service.NewRedisPersistenceService(conf.Redis) redisPersistence := service.NewRedisPersistenceService(conf.Redis)
persistenceServiceFacade.Redis = redisPersistence facade.Redis = redisPersistence
} }
if conf.Json != nil { if conf.Json != nil {
if _, err := os.Stat(conf.Json.Directory); os.IsNotExist(err) { if _, err := os.Stat(conf.Json.Directory); os.IsNotExist(err) {
if err2 := os.MkdirAll(conf.Json.Directory, 0777); err2 != nil { if err2 := os.MkdirAll(conf.Json.Directory, 0777); err2 != nil {
log.WithError(err2).Errorf("can not create directory: %s", conf.Json.Directory) return nil, errors.Wrapf(err2, "can not create directory: %s", conf.Json.Directory)
return err2
} }
} }
jsonPersistence := &service.JsonPersistenceService{Directory: conf.Json.Directory} jsonPersistence := &service.JsonPersistenceService{Directory: conf.Json.Directory}
persistenceServiceFacade.Json = jsonPersistence facade.Json = jsonPersistence
} }
return facade, nil
}
func ConfigurePersistence(conf *PersistenceConfig) error {
facade, err := NewPersistenceServiceFacade(conf)
if err != nil {
return err
}
persistenceServiceFacade = facade
return nil return nil
} }