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

schedule: add MinBaseBalance config

Browse Source
This commit is contained in:
c9s 2023-06-07 16:36:38 +08:00
parent e0e27e75bb
commit 0f141c7f79
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
1 changed files with 30 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
pkg/strategy/schedule/strategy.go
View File

@ -38,6 +38,7 @@ type Strategy struct {
bbgo.QuantityOrAmount bbgo.QuantityOrAmount
MinBaseBalance fixedpoint.Value `json:"minBaseBalance"`
MaxBaseBalance fixedpoint.Value `json:"maxBaseBalance"` MaxBaseBalance fixedpoint.Value `json:"maxBaseBalance"`
BelowMovingAverage *bbgo.MovingAverageSettings `json:"belowMovingAverage,omitempty"` BelowMovingAverage *bbgo.MovingAverageSettings `json:"belowMovingAverage,omitempty"`
@ -163,23 +164,35 @@ func (s *Strategy) Run(ctx context.Context, orderExecutor bbgo.OrderExecutor, se
// calculate quote quantity for balance checking // calculate quote quantity for balance checking
quoteQuantity := quantity.Mul(closePrice) quoteQuantity := quantity.Mul(closePrice)
quoteBalance, ok := session.GetAccount().Balance(s.Market.QuoteCurrency)
if !ok {
log.Errorf("can not place scheduled %s order, quote balance %s is empty", s.Symbol, s.Market.QuoteCurrency)
return
}
baseBalance, ok := session.GetAccount().Balance(s.Market.BaseCurrency)
if !ok {
log.Errorf("can not place scheduled %s order, base balance %s is empty", s.Symbol, s.Market.BaseCurrency)
return
}
totalBase := baseBalance.Total()
// execute orders // execute orders
switch side { switch side {
case types.SideTypeBuy: case types.SideTypeBuy:
if !s.MaxBaseBalance.IsZero() { if !s.MaxBaseBalance.IsZero() {
if baseBalance, ok := session.GetAccount().Balance(s.Market.BaseCurrency); ok { if totalBase.Add(quantity).Compare(s.MaxBaseBalance) >= 0 {
total := baseBalance.Total() quantity = s.MaxBaseBalance.Sub(totalBase)
if total.Add(quantity).Compare(s.MaxBaseBalance) >= 0 { quoteQuantity = quantity.Mul(closePrice)
quantity = s.MaxBaseBalance.Sub(total)
quoteQuantity = quantity.Mul(closePrice)
}
} }
} }
quoteBalance, ok := session.GetAccount().Balance(s.Market.QuoteCurrency) // if min base balance is defined
if !ok { if !s.MinBaseBalance.IsZero() && s.MinBaseBalance.Compare(totalBase) > 0 {
log.Errorf("can not place scheduled %s order, quote balance %s is empty", s.Symbol, s.Market.QuoteCurrency) quantity = fixedpoint.Max(quantity, s.MinBaseBalance.Sub(totalBase))
return quantity = fixedpoint.Max(quantity, s.Market.MinQuantity)
} }
if quoteBalance.Available.Compare(quoteQuantity) < 0 { if quoteBalance.Available.Compare(quoteQuantity) < 0 {
@ -188,13 +201,15 @@ func (s *Strategy) Run(ctx context.Context, orderExecutor bbgo.OrderExecutor, se
} }
case types.SideTypeSell: case types.SideTypeSell:
baseBalance, ok := session.GetAccount().Balance(s.Market.BaseCurrency) quantity = fixedpoint.Min(quantity, baseBalance.Available)
if !ok {
log.Errorf("can not place scheduled %s order, base balance %s is empty", s.Symbol, s.Market.BaseCurrency) // skip sell if we hit the minBaseBalance line
return if !s.MinBaseBalance.IsZero() {
if totalBase.Sub(quantity).Compare(s.MinBaseBalance) < 0 {
return
}
} }
quantity = fixedpoint.Min(quantity, baseBalance.Available)
quoteQuantity = quantity.Mul(closePrice) quoteQuantity = quantity.Mul(closePrice)
} }