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

feature/profitTracker: add AddTrade()

Browse Source
This commit is contained in:
Andy Cheng 2023-06-16 15:06:58 +08:00
parent d5e194ca80
commit 57cdbb1d77
No known key found for this signature in database
GPG Key ID: 936427CF651A9D28
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/report/profit_tracker.go
View File

@ -34,17 +34,17 @@ func (p *ProfitTracker) Init(market types.Market) {
} }
func (p *ProfitTracker) Bind(tradeCollector *bbgo.TradeCollector, session *bbgo.ExchangeSession) { func (p *ProfitTracker) Bind(tradeCollector *bbgo.TradeCollector, session *bbgo.ExchangeSession) {
// TODO: Register kline close callback
tradeCollector.OnProfit(func(trade types.Trade, profit *types.Profit) { tradeCollector.OnProfit(func(trade types.Trade, profit *types.Profit) {
p.AddProfit(*profit) p.AddProfit(*profit)
}) })
tradeCollector.OnTrade(func(trade types.Trade, profit fixedpoint.Value, netProfit fixedpoint.Value) { tradeCollector.OnTrade(func(trade types.Trade, profit fixedpoint.Value, netProfit fixedpoint.Value) {
p.AddTrade(trade)
}) })
// Rotate profitStats slice
session.MarketDataStream.OnKLineClosed(types.KLineWith(p.market.Symbol, p.Interval, func(kline types.KLine) { session.MarketDataStream.OnKLineClosed(types.KLineWith(p.market.Symbol, p.Interval, func(kline types.KLine) {
p.Rotate()
})) }))
} }
@ -61,3 +61,7 @@ func (p *ProfitTracker) Rotate() {
func (p *ProfitTracker) AddProfit(profit types.Profit) { func (p *ProfitTracker) AddProfit(profit types.Profit) {
(*p.currentProfitStats).AddProfit(profit) (*p.currentProfitStats).AddProfit(profit)
} }
func (p *ProfitTracker) AddTrade(trade types.Trade) {
(*p.currentProfitStats).AddTrade(trade)
}