lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-09-20 08:11:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

show wallet balance and move kline cache

Browse Source
This commit is contained in:
c9s 2020-07-18 10:01:10 +08:00
parent 60efce96b6
commit c94c85d51f
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
bbgo/trader.go
View File

@ -71,11 +71,16 @@ func (trader *KLineRegressionTrader) RunStrategy(ctx context.Context, strategy S
fee := 0.0
feeCurrency := ""
trader.Context.Lock()
if order.Side == types.SideTypeBuy {
fee = price * volume * 0.001
feeCurrency = "USDT"
quote := trader.Context.Balances[trader.Context.Market.QuoteCurrency]
if quote.Available < volume * price {
log.Fatalf("quote balance not enough: %+v", quote)
}
quote.Available -= volume * price
trader.Context.Balances[trader.Context.Market.QuoteCurrency] = quote
@ -87,14 +92,19 @@ func (trader *KLineRegressionTrader) RunStrategy(ctx context.Context, strategy S
fee = volume * 0.001
feeCurrency = "BTC"
base := trader.Context.Balances[trader.Context.Market.BaseCurrency]
if base.Available < volume {
log.Fatalf("base balance not enough: %+v", base)
}
base.Available -= volume
trader.Context.Balances[trader.Context.Market.BaseCurrency] = base
quote := trader.Context.Balances[trader.Context.Market.QuoteCurrency]
quote.Available += volume * price
trader.Context.Balances[trader.Context.Market.QuoteCurrency] = quote
base := trader.Context.Balances[trader.Context.Market.BaseCurrency]
base.Available -= volume
trader.Context.Balances[trader.Context.Market.BaseCurrency] = base
}
trader.Context.Unlock()
trade := types.Trade{
ID: tradeID,
@ -119,6 +129,11 @@ func (trader *KLineRegressionTrader) RunStrategy(ctx context.Context, strategy S
report := trader.ProfitAndLossCalculator.Calculate()
report.Print()
log.Infof("wallet balance:")
for _, balance := range trader.Context.Balances {
log.Infof(" %s: %f", balance.Currency, balance.Available)
}
return done, nil
}