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

improve/profitTracker: subscribe kline in strategy Subscribe()

Browse Source
This commit is contained in:
Andy Cheng 2023-07-10 15:26:57 +08:00
parent 80170e0397
commit bcbb27de79
No known key found for this signature in database
GPG Key ID: 936427CF651A9D28
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/report/profit_tracker.go
View File

@ -20,6 +20,10 @@ type ProfitTracker struct {
tradeStats *types.TradeStats
}
func (p *ProfitTracker) Subscribe(session *bbgo.ExchangeSession) {
session.Subscribe(types.KLineChannel, p.Market.Symbol, types.SubscribeOptions{Interval: p.Interval})
}
// InitOld is for backward capability. ps is the ProfitStats of the strategy, Market is the strategy Market
func (p *ProfitTracker) InitOld(market types.Market, ps **types.ProfitStats, ts *types.TradeStats) {
p.Market = market
@ -45,8 +49,6 @@ func (p *ProfitTracker) Init(market types.Market, ts *types.TradeStats) {
}
func (p *ProfitTracker) Bind(session *bbgo.ExchangeSession, tradeCollector *bbgo.TradeCollector) {
session.Subscribe(types.KLineChannel, p.Market.Symbol, types.SubscribeOptions{Interval: p.Interval})
tradeCollector.OnProfit(func(trade types.Trade, profit *types.Profit) {
if profit == nil {
return

5
pkg/strategy/supertrend/strategy.go
View File

@ -130,6 +130,11 @@ func (s *Strategy) Subscribe(session *bbgo.ExchangeSession) {
session.Subscribe(types.KLineChannel, s.Symbol, types.SubscribeOptions{Interval: s.LinearRegression.Interval})
s.ExitMethods.SetAndSubscribe(session, s)
// Profit tracker
if s.ProfitTracker != nil {
s.ProfitTracker.Subscribe(session)
}
}
// Position control