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

bbgo: add SetTradeScanTime method

Browse Source
This commit is contained in:
c9s 2020-10-20 13:11:04 +08:00
parent 3b3df77ec3
commit a08aebaa17
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pkg/bbgo/environment.go
View File

@ -22,7 +22,8 @@ type Environment struct {
TradeService *service.TradeService TradeService *service.TradeService
TradeSync *service.TradeSync TradeSync *service.TradeSync
sessions map[string]*ExchangeSession tradeScanTime time.Time
sessions map[string]*ExchangeSession
} }
func NewDefaultEnvironment(db *sqlx.DB) *Environment { func NewDefaultEnvironment(db *sqlx.DB) *Environment {
@ -49,7 +50,8 @@ func NewEnvironment(db *sqlx.DB) *Environment {
TradeSync: &service.TradeSync{ TradeSync: &service.TradeSync{
Service: tradeService, Service: tradeService,
}, },
sessions: make(map[string]*ExchangeSession), tradeScanTime: time.Now().AddDate(0, 0, -7), // sync from 7 days ago
sessions: make(map[string]*ExchangeSession),
} }
} }
@ -82,9 +84,14 @@ func (environ *Environment) Init(ctx context.Context) (err error) {
return nil return nil
} }
// SetTradeScanTime overrides the default trade scan time (-7 days)
func (environ *Environment) SetTradeScanTime(t time.Time) *Environment {
environ.tradeScanTime = t
return environ
}
func (environ *Environment) Connect(ctx context.Context) error { func (environ *Environment) Connect(ctx context.Context) error {
var err error var err error
var startTime = time.Now().AddDate(0, 0, -7) // sync from 7 days ago
for n := range environ.sessions { for n := range environ.sessions {
// avoid using the placeholder variable for the session because we use that in the callbacks // avoid using the placeholder variable for the session because we use that in the callbacks
@ -103,7 +110,7 @@ func (environ *Environment) Connect(ctx context.Context) error {
// trade sync and market data store depends on subscribed symbols so we have to do this here. // trade sync and market data store depends on subscribed symbols so we have to do this here.
for symbol := range loadedSymbols { for symbol := range loadedSymbols {
log.Infof("syncing trades from %s for symbol %s...", session.Exchange.Name(), symbol) log.Infof("syncing trades from %s for symbol %s...", session.Exchange.Name(), symbol)
if err := environ.TradeSync.Sync(ctx, session.Exchange, symbol, startTime); err != nil { if err := environ.TradeSync.Sync(ctx, session.Exchange, symbol, environ.tradeScanTime); err != nil {
return err return err
} }