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

cmd: apply config to sync

Browse Source
This commit is contained in:
c9s 2022-01-27 00:17:11 +08:00
parent 98170b2d8e
commit ab07768a6d
1 changed files with 31 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
pkg/cmd/sync.go
View File

@ -58,17 +58,7 @@ var SyncCmd = &cobra.Command{
return err return err
} }
var (
// default sync start time
startTime = time.Now().AddDate(-1, 0, 0)
)
if len(since) > 0 {
startTime, err = time.ParseInLocation("2006-01-02", since, time.Local)
if err != nil {
return err
}
}
sessionName, err := cmd.Flags().GetString("session") sessionName, err := cmd.Flags().GetString("session")
if err != nil { if err != nil {
@ -80,22 +70,49 @@ var SyncCmd = &cobra.Command{
return err return err
} }
environ.SetSyncStartTime(startTime) var (
// default sync start time
defaultSyncStartTime = time.Now().AddDate(-1, 0, 0)
)
var syncStartTime = defaultSyncStartTime
if userConfig.Sync != nil && userConfig.Sync.Since != nil {
syncStartTime = *userConfig.Sync.Since
}
if len(since) > 0 {
syncStartTime, err = time.ParseInLocation("2006-01-02", since, time.Local)
if err != nil {
return err
}
}
environ.SetSyncStartTime(syncStartTime)
// syncSymbols is the symbol list to sync
var syncSymbols []string
if userConfig.Sync != nil && len(userConfig.Sync.Symbols) > 0 {
syncSymbols = userConfig.Sync.Symbols
}
var defaultSymbols []string
if len(symbol) > 0 { if len(symbol) > 0 {
defaultSymbols = []string{symbol} syncSymbols = []string{symbol}
} }
var selectedSessions []string var selectedSessions []string
if userConfig.Sync != nil && len(userConfig.Sync.Sessions) > 0 {
selectedSessions = userConfig.Sync.Sessions
}
if len(sessionName) > 0 { if len(sessionName) > 0 {
selectedSessions = []string{sessionName} selectedSessions = []string{sessionName}
} }
sessions := environ.SelectSessions(selectedSessions...) sessions := environ.SelectSessions(selectedSessions...)
for _, session := range sessions { for _, session := range sessions {
if err := environ.SyncSession(ctx, session, defaultSymbols...); err != nil { if err := environ.SyncSession(ctx, session, syncSymbols...); err != nil {
return err return err
} }