bbgo: add DisableStartupBalanceQuery option

pkg/bbgo/config.go

@@ -329,8 +329,13 @@ type ServiceConfig struct {
 type EnvironmentConfig struct {
 	DisableDefaultKLineSubscription bool `json:"disableDefaultKLineSubscription"`
 	DisableHistoryKLinePreload      bool `json:"disableHistoryKLinePreload"`
-	DisableSessionTradeBuffer       bool `json:"disableSessionTradeBuffer"`
+
-	MaxSessionTradeBufferSize       int  `json:"maxSessionTradeBufferSize"`
+	// DisableStartUpBalanceQuery disables the balance query in the startup process
+	// which initializes the session.Account with the QueryAccount method.
+	DisableStartupBalanceQuery bool `json:"disableStartupBalanceQuery"`
+
+	DisableSessionTradeBuffer bool `json:"disableSessionTradeBuffer"`
+	MaxSessionTradeBufferSize int  `json:"maxSessionTradeBufferSize"`
 }
 
 type Config struct {

pkg/bbgo/session.go

@@ -256,15 +256,22 @@ func (session *ExchangeSession) Init(ctx context.Context, environ *Environment)
 			}
 		}
 
-		logger.Infof("querying account balances...")
+		disableStartupBalanceQuery := environ.environmentConfig != nil && environ.environmentConfig.DisableStartupBalanceQuery
-		account, err := session.Exchange.QueryAccount(ctx)
+		if disableStartupBalanceQuery {
-		if err != nil {
+			session.accountMutex.Lock()
-			return err
+			session.Account = types.NewAccount()
-		}
+			session.accountMutex.Unlock()
+		} else {
+			logger.Infof("querying account balances...")
+			account, err := session.Exchange.QueryAccount(ctx)
+			if err != nil {
+				return err
+			}
 
-		session.accountMutex.Lock()
+			session.accountMutex.Lock()
-		session.Account = account
+			session.Account = account
-		session.accountMutex.Unlock()
+			session.accountMutex.Unlock()
+		}
 
 		logger.Infof("account %s balances:\n%s", session.Name, account.Balances().String())
 

pkg/types/account.go

@@ -103,8 +103,26 @@ type IsolatedMarginAccountInfo struct {
 
 func NewAccount() *Account {
 	return &Account{
-		balances: make(BalanceMap),
+		AccountType:        "spot",
+		FuturesInfo:        nil,
+		MarginInfo:         nil,
+		IsolatedMarginInfo: nil,
+		MarginLevel:        fixedpoint.Zero,
+		MarginTolerance:    fixedpoint.Zero,
+		BorrowEnabled:      false,
+		TransferEnabled:    false,
+		MarginRatio:        fixedpoint.Zero,
+		LiquidationPrice:   fixedpoint.Zero,
+		LiquidationRate:    fixedpoint.Zero,
+		MakerFeeRate:       fixedpoint.Zero,
+		TakerFeeRate:       fixedpoint.Zero,
+		TotalAccountValue:  fixedpoint.Zero,
+		CanDeposit:         false,
+		CanTrade:           false,
+		CanWithdraw:        false,
+		balances:           make(BalanceMap),
 	}
+
 }
 
 // Balances lock the balances and returned the copied balances