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

bbgo: fix LooseFormatTime.UnmarshalYAML

Browse Source
This commit is contained in:
c9s 2022-01-25 01:18:50 +08:00
parent 8f0e80499b
commit a29198f733
2 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pkg/bbgo/config_test.go
View File

@ -184,7 +184,6 @@ func TestLoadConfig(t *testing.T) {
assert.NotNil(t, config.Backtest.Account) assert.NotNil(t, config.Backtest.Account)
assert.NotNil(t, config.Backtest.Account.Balances) assert.NotNil(t, config.Backtest.Account.Balances)
assert.Len(t, config.Backtest.Account.Balances, 2) assert.Len(t, config.Backtest.Account.Balances, 2)
assert.NotEmpty(t, config.Backtest.StartTime)
}, },
}, },
} }

10
pkg/types/time.go
View File

@ -229,17 +229,17 @@ type LooseFormatTime time.Time
func (t *LooseFormatTime) UnmarshalYAML(unmarshal func(interface{}) error) error { func (t *LooseFormatTime) UnmarshalYAML(unmarshal func(interface{}) error) error {
var str string var str string
if err := unmarshal(&str); err == nil { if err := unmarshal(&str); err != nil {
return t.UnmarshalJSON([]byte(str)) return err
} }
var bin []byte tv, err := util.ParseTimeWithFormats(str, looseTimeFormats)
err := unmarshal(&bin)
if err != nil { if err != nil {
return err return err
} }
return t.UnmarshalJSON(bin) *t = LooseFormatTime(tv)
return nil
} }
func (t *LooseFormatTime) UnmarshalJSON(data []byte) error { func (t *LooseFormatTime) UnmarshalJSON(data []byte) error {