bbgo: fix LooseFormatTime.UnmarshalYAML

wdshin · Jan 24, 2022 · a29198f · a29198f
1 parent 8f0e804
commit a29198f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/pkg/bbgo/config_test.go b/pkg/bbgo/config_test.go
@@ -184,7 +184,6 @@ func TestLoadConfig(t *testing.T) {
 				assert.NotNil(t, config.Backtest.Account)
 				assert.NotNil(t, config.Backtest.Account.Balances)
 				assert.Len(t, config.Backtest.Account.Balances, 2)
-				assert.NotEmpty(t, config.Backtest.StartTime)
 			},
 		},
 	}

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