mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-10 09:11:55 +00:00
126 lines
3.3 KiB
Go
126 lines
3.3 KiB
Go
package bbgo
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/c9s/bbgo/pkg/service"
|
|
)
|
|
|
|
type PersistenceSelector struct {
|
|
// StoreID is the store you want to use.
|
|
StoreID string `json:"store" yaml:"store"`
|
|
|
|
// Type is the persistence type
|
|
Type string `json:"type" yaml:"type"`
|
|
}
|
|
|
|
// Persistence is used for strategy to inject the persistence.
|
|
type Persistence struct {
|
|
PersistenceSelector *PersistenceSelector `json:"persistence,omitempty" yaml:"persistence,omitempty"`
|
|
|
|
Facade *service.PersistenceServiceFacade `json:"-" yaml:"-"`
|
|
}
|
|
|
|
func (p *Persistence) backendService(t string) (service.PersistenceService, error) {
|
|
switch t {
|
|
case "json":
|
|
return p.Facade.Json, nil
|
|
|
|
case "redis":
|
|
if p.Facade.Redis == nil {
|
|
log.Warn("redis persistence is not available, fallback to memory backend")
|
|
return p.Facade.Memory, nil
|
|
}
|
|
return p.Facade.Redis, nil
|
|
|
|
case "memory":
|
|
return p.Facade.Memory, nil
|
|
|
|
}
|
|
|
|
return nil, fmt.Errorf("unsupported persistent type %s", t)
|
|
}
|
|
|
|
func (p *Persistence) Load(val interface{}, subIDs ...string) error {
|
|
ps, err := p.backendService(p.PersistenceSelector.Type)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Debugf("using persistence store %T for loading", ps)
|
|
|
|
if p.PersistenceSelector.StoreID == "" {
|
|
p.PersistenceSelector.StoreID = "default"
|
|
}
|
|
|
|
store := ps.NewStore(p.PersistenceSelector.StoreID, subIDs...)
|
|
return store.Load(val)
|
|
}
|
|
|
|
func (p *Persistence) Save(val interface{}, subIDs ...string) error {
|
|
ps, err := p.backendService(p.PersistenceSelector.Type)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Debugf("using persistence store %T for storing", ps)
|
|
|
|
if p.PersistenceSelector.StoreID == "" {
|
|
p.PersistenceSelector.StoreID = "default"
|
|
}
|
|
|
|
store := ps.NewStore(p.PersistenceSelector.StoreID, subIDs...)
|
|
return store.Save(val)
|
|
}
|
|
|
|
func (p *Persistence) Sync(obj interface{}) error {
|
|
id := callID(obj)
|
|
if len(id) == 0 {
|
|
return nil
|
|
}
|
|
|
|
ps := p.Facade.Get()
|
|
return storePersistenceFields(obj, id, ps)
|
|
}
|
|
|
|
func loadPersistenceFields(obj interface{}, id string, persistence service.PersistenceService) error {
|
|
return iterateFieldsByTag(obj, "persistence", func(tag string, field reflect.StructField, value reflect.Value) error {
|
|
log.Debugf("[loadPersistenceFields] loading value into field %v, tag = %s, original value = %v", field, tag, value)
|
|
|
|
newValueInf := newTypeValueInterface(value.Type())
|
|
// inf := value.Interface()
|
|
store := persistence.NewStore("state", id, tag)
|
|
if err := store.Load(&newValueInf); err != nil {
|
|
if err == service.ErrPersistenceNotExists {
|
|
log.Debugf("[loadPersistenceFields] state key does not exist, id = %v, tag = %s", id, tag)
|
|
return nil
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
newValue := reflect.ValueOf(newValueInf)
|
|
if value.Kind() != reflect.Ptr && newValue.Kind() == reflect.Ptr {
|
|
newValue = newValue.Elem()
|
|
}
|
|
|
|
log.Debugf("[loadPersistenceFields] %v = %v -> %v\n", field, value, newValue)
|
|
|
|
value.Set(newValue)
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func storePersistenceFields(obj interface{}, id string, persistence service.PersistenceService) error {
|
|
return iterateFieldsByTag(obj, "persistence", func(tag string, ft reflect.StructField, fv reflect.Value) error {
|
|
log.Debugf("[storePersistenceFields] storing value from field %v, tag = %s, original value = %v", ft, tag, fv)
|
|
|
|
inf := fv.Interface()
|
|
store := persistence.NewStore("state", id, tag)
|
|
return store.Save(inf)
|
|
})
|
|
}
|