bbgo: configure persistence facade into the isolation context

Author: c9s

pkg/bbgo/bootstrap.go

@@ -15,7 +15,7 @@ func BootstrapEnvironmentLightweight(ctx context.Context, environ *Environment,
 	}
 
 	if userConfig.Persistence != nil {
-		if err := ConfigurePersistence(userConfig.Persistence); err != nil {
+		if err := ConfigurePersistence(ctx, userConfig.Persistence); err != nil {
 			return errors.Wrap(err, "persistence configure error")
 		}
 	}
@@ -33,7 +33,7 @@ func BootstrapEnvironment(ctx context.Context, environ *Environment, userConfig
 	}
 
 	if userConfig.Persistence != nil {
-		if err := ConfigurePersistence(userConfig.Persistence); err != nil {
+		if err := ConfigurePersistence(ctx, userConfig.Persistence); err != nil {
 			return errors.Wrap(err, "persistence configure error")
 		}
 	}

pkg/bbgo/persistence.go

@@ -102,12 +102,15 @@ func NewPersistenceServiceFacade(conf *PersistenceConfig) (*service.PersistenceS
 	return facade, nil
 }
 
-func ConfigurePersistence(conf *PersistenceConfig) error {
+func ConfigurePersistence(ctx context.Context, conf *PersistenceConfig) error {
 	facade, err := NewPersistenceServiceFacade(conf)
 	if err != nil {
 		return err
 	}
 
+	isolation := GetIsolationFromContext(ctx)
+	isolation.persistenceServiceFacade = facade
+
 	persistenceServiceFacade = facade
 	return nil
 }