bbgo: add context to LoadState

This commit is contained in:
c9s 2023-04-25 18:29:07 +08:00
parent 45a27ffe06
commit a9b0270390
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
4 changed files with 6 additions and 8 deletions

View File

@ -101,7 +101,7 @@ func main() {
return return
} }
if err := trader.LoadState(); err != nil { if err := trader.LoadState(ctx); err != nil {
log.WithError(err).Error("failed to load strategy states") log.WithError(err).Error("failed to load strategy states")
return return
} }

View File

@ -109,7 +109,7 @@ func main() {
return return
} }
if err := trader.LoadState(); err != nil { if err := trader.LoadState(ctx); err != nil {
log.WithError(err).Error("failed to load strategy states") log.WithError(err).Error("failed to load strategy states")
return return
} }

View File

@ -365,16 +365,14 @@ func (trader *Trader) Run(ctx context.Context) error {
return trader.environment.Connect(ctx) return trader.environment.Connect(ctx)
} }
func (trader *Trader) LoadState() error { func (trader *Trader) LoadState(ctx context.Context) error {
if trader.environment.BacktestService != nil { if trader.environment.BacktestService != nil {
return nil return nil
} }
if persistenceServiceFacade == nil { isolation := GetIsolationFromContext(ctx)
return nil
}
ps := persistenceServiceFacade.Get() ps := isolation.persistenceServiceFacade.Get()
log.Infof("loading strategies states...") log.Infof("loading strategies states...")

View File

@ -163,7 +163,7 @@ func runConfig(basectx context.Context, cmd *cobra.Command, userConfig *bbgo.Con
return err return err
} }
if err := trader.LoadState(); err != nil { if err := trader.LoadState(tradingCtx); err != nil {
return err return err
} }