lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-14 02:53:50 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

core: set default store ID

Browse Source
This commit is contained in:
c9s 2021-03-18 17:20:07 +08:00
parent 0e6918a352
commit a95c3b94a0
1 changed files with 11 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
pkg/bbgo/persistence.go
View File

@ -21,49 +21,46 @@ type Persistence struct {
Facade *service.PersistenceServiceFacade `json:"-" yaml:"-"` Facade *service.PersistenceServiceFacade `json:"-" yaml:"-"`
} }
func (p *Persistence) backendService(t string) (service service.PersistenceService, err error) { func (p *Persistence) backendService(t string) (service.PersistenceService, error) {
switch t { switch t {
case "json": case "json":
service = p.Facade.Json return p.Facade.Json, nil
case "redis": case "redis":
service = p.Facade.Redis return p.Facade.Redis, nil
case "memory": case "memory":
service = p.Facade.Memory return p.Facade.Memory, nil
default:
err = fmt.Errorf("unsupported persistent type %s", t)
} }
return service, err return nil, fmt.Errorf("unsupported persistent type %s", t)
} }
func (p *Persistence) Load(val interface{}, subIDs ...string) error { func (p *Persistence) Load(val interface{}, subIDs ...string) error {
service, err := p.backendService(p.PersistenceSelector.Type) ps, err := p.backendService(p.PersistenceSelector.Type)
if err != nil { if err != nil {
return err return err
} }
if p.PersistenceSelector.StoreID == "" { if p.PersistenceSelector.StoreID == "" {
return fmt.Errorf("persistence.store can not be empty") p.PersistenceSelector.StoreID = "default"
} }
store := service.NewStore(p.PersistenceSelector.StoreID, subIDs...) store := ps.NewStore(p.PersistenceSelector.StoreID, subIDs...)
return store.Load(val) return store.Load(val)
} }
func (p *Persistence) Save(val interface{}, subIDs ...string) error { func (p *Persistence) Save(val interface{}, subIDs ...string) error {
service, err := p.backendService(p.PersistenceSelector.Type) ps, err := p.backendService(p.PersistenceSelector.Type)
if err != nil { if err != nil {
return err return err
} }
if p.PersistenceSelector.StoreID == "" { if p.PersistenceSelector.StoreID == "" {
return fmt.Errorf("persistence.store can not be empty") p.PersistenceSelector.StoreID = "default"
} }
store := service.NewStore(p.PersistenceSelector.StoreID, subIDs...) store := ps.NewStore(p.PersistenceSelector.StoreID, subIDs...)
return store.Save(val) return store.Save(val)
} }