bbgo_origin/pkg/bbgo/bootstrap.go

71 lines
1.8 KiB
Go

package bbgo
import (
"context"
"github.com/pkg/errors"
)
// BootstrapEnvironmentLightweight bootstrap the environment in lightweight mode
// - no database configuration
// - no notification
func BootstrapEnvironmentLightweight(ctx context.Context, environ *Environment, userConfig *Config) error {
if err := environ.ConfigureExchangeSessions(userConfig); err != nil {
return errors.Wrap(err, "exchange session configure error")
}
if userConfig.Logging != nil {
environ.SetLogging(userConfig.Logging)
}
if userConfig.Persistence != nil {
if err := ConfigurePersistence(ctx, environ, userConfig.Persistence); err != nil {
return errors.Wrap(err, "persistence configure error")
}
}
if userConfig.Service != nil {
if err := environ.ConfigureService(ctx, userConfig.Service); err != nil {
return err
}
}
return nil
}
func BootstrapEnvironment(ctx context.Context, environ *Environment, userConfig *Config) error {
if err := environ.ConfigureDatabase(ctx); err != nil {
return err
}
if err := environ.ConfigureExchangeSessions(userConfig); err != nil {
return errors.Wrap(err, "exchange session configure error")
}
if userConfig.Logging != nil {
environ.SetLogging(userConfig.Logging)
}
if userConfig.Persistence != nil {
if err := ConfigurePersistence(ctx, environ, userConfig.Persistence); err != nil {
return errors.Wrap(err, "persistence configure error")
}
}
if userConfig.Service != nil {
if err := environ.ConfigureService(ctx, userConfig.Service); err != nil {
return err
}
}
if err := environ.ConfigureNotificationSystem(ctx, userConfig); err != nil {
return errors.Wrap(err, "notification configure error")
}
return nil
}
func BootstrapBacktestEnvironment(ctx context.Context, environ *Environment) error {
return environ.ConfigureDatabase(ctx)
}