autobuy: improve balance checking

c9s · Dec 16, 2024 · 93bb7a3 · 93bb7a3
1 parent f259f0b
commit 93bb7a3
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/pkg/strategy/autobuy/strategy.go b/pkg/strategy/autobuy/strategy.go
@@ -146,16 +146,14 @@ func (s *Strategy) cancelOrders(ctx context.Context) {
 func (s *Strategy) autobuy(ctx context.Context) {
 	s.cancelOrders(ctx)
 
-	baseBalance, ok := s.Session.GetAccount().Balance(s.Market.BaseCurrency)
-	if !ok {
-		baseBalance = types.NewZeroBalance(s.Market.BaseCurrency)
-	}
+	account := s.Session.GetAccount()
+	baseBalance, _ := account.Balance(s.Market.BaseCurrency)
 
 	log.Infof("balance: %s", baseBalance.String())
 
-	quoteBalance, ok := s.Session.GetAccount().Balance(s.Market.QuoteCurrency)
+	quoteBalance, ok := account.Balance(s.Market.QuoteCurrency)
 	if !ok {
-		log.Errorf("%s balance not found", s.Market.QuoteCurrency)
+		log.Errorf("quote balance is zero")
 		return
 	}