lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-22 23:05:15 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Use fixedpoint type to calculate the balance

Browse Source
This commit is contained in:
Larry850806 2021-02-10 16:00:19 +08:00
parent 6d399647cf
commit 9f6d9028fa
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
pkg/strategy/bollgrid/strategy.go
View File

@ -95,9 +95,9 @@ func (s *Strategy) Subscribe(session *bbgo.ExchangeSession) {
func (s *Strategy) generateGridBuyOrders(session *bbgo.ExchangeSession) ([]types.SubmitOrder, error) { func (s *Strategy) generateGridBuyOrders(session *bbgo.ExchangeSession) ([]types.SubmitOrder, error) {
balances := session.Account.Balances() balances := session.Account.Balances()
quoteBalance := balances[s.Market.QuoteCurrency].Available.Float64() quoteBalance := balances[s.Market.QuoteCurrency].Available
if quoteBalance <= 0 { if quoteBalance <= 0 {
return nil, fmt.Errorf("quote balance %s is zero: %+v", s.Market.QuoteCurrency, quoteBalance) return nil, fmt.Errorf("quote balance %s is zero: %+v", s.Market.QuoteCurrency, quoteBalance.Float64())
} }
upBand, downBand := s.boll.LastUpBand(), s.boll.LastDownBand() upBand, downBand := s.boll.LastUpBand(), s.boll.LastDownBand()
@ -142,11 +142,12 @@ func (s *Strategy) generateGridBuyOrders(session *bbgo.ExchangeSession) ([]types
Price: price, Price: price,
TimeInForce: "GTC", TimeInForce: "GTC",
} }
if quoteBalance < order.Quantity * price { quotaQuantity := fixedpoint.NewFromFloat(order.Quantity).MulFloat64(price)
log.Infof("quote balance %f is not enough, stop generating buy orders", quoteBalance) if quoteBalance < quotaQuantity {
log.Infof("quote balance %f is not enough, stop generating buy orders", quoteBalance.Float64())
break break
} }
quoteBalance -= order.Quantity * price quoteBalance = quotaQuantity.Sub(quotaQuantity)
log.Infof("submitting order: %s", order.String()) log.Infof("submitting order: %s", order.String())
orders = append(orders, order) orders = append(orders, order)
} }
@ -155,9 +156,9 @@ func (s *Strategy) generateGridBuyOrders(session *bbgo.ExchangeSession) ([]types
func (s *Strategy) generateGridSellOrders(session *bbgo.ExchangeSession) ([]types.SubmitOrder, error) { func (s *Strategy) generateGridSellOrders(session *bbgo.ExchangeSession) ([]types.SubmitOrder, error) {
balances := session.Account.Balances() balances := session.Account.Balances()
baseBalance := balances[s.Market.BaseCurrency].Available.Float64() baseBalance := balances[s.Market.BaseCurrency].Available
if baseBalance <= 0 { if baseBalance <= 0 {
return nil, fmt.Errorf("base balance %s is zero: %+v", s.Market.BaseCurrency, baseBalance) return nil, fmt.Errorf("base balance %s is zero: %+v", s.Market.BaseCurrency, baseBalance.Float64())
} }
upBand, downBand := s.boll.LastUpBand(), s.boll.LastDownBand() upBand, downBand := s.boll.LastUpBand(), s.boll.LastDownBand()
@ -202,11 +203,12 @@ func (s *Strategy) generateGridSellOrders(session *bbgo.ExchangeSession) ([]type
Price: price, Price: price,
TimeInForce: "GTC", TimeInForce: "GTC",
} }
if baseBalance < order.Quantity { baseQuantity := fixedpoint.NewFromFloat(order.Quantity)
log.Infof("base balance %f is not enough, stop generating sell orders", baseBalance) if baseBalance < baseQuantity {
log.Infof("base balance %f is not enough, stop generating sell orders", baseBalance.Float64())
break break
} }
baseBalance -= order.Quantity baseBalance = baseBalance.Sub(baseQuantity)
log.Infof("submitting order: %s", order.String()) log.Infof("submitting order: %s", order.String())
orders = append(orders, order) orders = append(orders, order)
} }