all: implement stats tracker

- rename Add() to AddProfit() or AddTrade() so that we can apply
  interface here

Author: c9s

pkg/bbgo/order_executor_general.go

@@ -160,7 +160,7 @@ func (e *GeneralOrderExecutor) BindTradeStats(tradeStats *types.TradeStats) {
 			return
 		}
 
-		tradeStats.Add(profit)
+		tradeStats.AddProfit(profit)
 	})
 }
 
@@ -171,7 +171,7 @@ func (e *GeneralOrderExecutor) BindProfitStats(profitStats *types.ProfitStats) {
 			return
 		}
 
-		profitStats.AddProfit(*profit)
+		profitStats.AddProfit(profit)
 
 		if !e.disableNotify {
 			Notify(profit)

pkg/cmd/backtest.go

@@ -325,7 +325,7 @@ var BacktestCmd = &cobra.Command{
 					if profit == nil {
 						return
 					}
-					tradeStats.Add(profit)
+					tradeStats.AddProfit(profit)
 				})
 				tradeStatsMap[usedSymbol] = tradeStats
 

pkg/report/profit_stats_tracker.go

@@ -84,7 +84,7 @@ func (p *ProfitStatsTracker) Rotate() {
 }
 
 func (p *ProfitStatsTracker) AddProfit(profit types.Profit) {
-	(*p.CurrentProfitStats).AddProfit(profit)
+	(*p.CurrentProfitStats).AddProfit(&profit)
 }
 
 func (p *ProfitStatsTracker) AddTrade(trade types.Trade) {

pkg/report/stats_collector.go

@@ -0,0 +1,83 @@
+package report
+
+import (
+	"github.com/c9s/bbgo/pkg/bbgo"
+	"github.com/c9s/bbgo/pkg/core"
+	"github.com/c9s/bbgo/pkg/types"
+)
+
+type TradeAdder interface {
+	AddTrade(trade *types.Trade)
+}
+
+type ProfitAdder interface {
+	AddProfit(trade *types.Profit)
+}
+
+// StatsCollector is the v2 profit stats tracker
+type StatsCollector struct {
+	Market   types.Market   `json:"market"`
+	Interval types.Interval `json:"interval"`
+	Window   int            `json:"window"`
+
+	CurrentProfitStats     *types.ProfitStats  `json:"profitStats"`
+	AccumulatedProfitStats *types.ProfitStats  `json:"accumulatedProfitStats"`
+	HistoryProfitStats     []types.ProfitStats `json:"historyProfitStats"`
+
+	CurrentTradeStats     *types.TradeStats  `json:"tradeStats"`
+	AccumulatedTradeStats *types.TradeStats  `json:"accumulatedTradeStats"`
+	HistoryTradeStats     []types.TradeStats `json:"historyTradeStats"`
+
+	tradeCollector *core.TradeCollector
+}
+
+func NewStatsCollector(market types.Market, interval types.Interval, window int, tradeCollector *core.TradeCollector) *StatsCollector {
+	return &StatsCollector{
+		Market:                 market,
+		Interval:               interval,
+		Window:                 window,
+		CurrentProfitStats:     types.NewProfitStats(market),
+		CurrentTradeStats:      types.NewTradeStats(market.Symbol),
+		AccumulatedProfitStats: types.NewProfitStats(market),
+		AccumulatedTradeStats:  types.NewTradeStats(market.Symbol),
+		tradeCollector:         tradeCollector,
+	}
+}
+
+func (c *StatsCollector) Subscribe(session *bbgo.ExchangeSession) {
+	session.Subscribe(types.KLineChannel, c.Market.Symbol, types.SubscribeOptions{Interval: c.Interval})
+}
+
+func (c *StatsCollector) Bind(session *bbgo.ExchangeSession) {
+	c.tradeCollector.OnProfit(func(trade types.Trade, profit *types.Profit) {
+		if profit != nil {
+			c.CurrentProfitStats.AddProfit(profit)
+			c.AccumulatedProfitStats.AddProfit(profit)
+		}
+
+		c.CurrentProfitStats.AddTrade(trade)
+		c.AccumulatedProfitStats.AddTrade(trade)
+
+		c.CurrentTradeStats.AddProfit(profit)
+		c.AccumulatedTradeStats.AddProfit(profit)
+	})
+
+	// Rotate profitStats slice
+	session.MarketDataStream.OnKLineClosed(types.KLineWith(c.Market.Symbol, c.Interval, func(k types.KLine) {
+		// p.Rotate()
+	}))
+}
+
+// Rotate the tracker to make a new ProfitStats to record the profits
+func (c *StatsCollector) Rotate() {
+	c.HistoryProfitStats = append(c.HistoryProfitStats, *c.CurrentProfitStats)
+	c.HistoryTradeStats = append(c.HistoryTradeStats, *c.CurrentTradeStats)
+	/*
+		*p.CurrentProfitStats = types.NewProfitStats(p.Market)
+		p.ProfitStatsSlice = append(p.ProfitStatsSlice, *p.CurrentProfitStats)
+		// Truncate
+		if len(p.ProfitStatsSlice) > p.Window {
+			p.ProfitStatsSlice = p.ProfitStatsSlice[len(p.ProfitStatsSlice)-p.Window:]
+		}
+	*/
+}

pkg/risk/riskcontrol/circuit_break.go

@@ -37,5 +37,6 @@ func (c *CircuitBreakRiskControl) IsHalted() bool {
 	log.Infof("[CircuitBreakRiskControl] realized PnL = %f, unrealized PnL = %f\n",
 		c.profitStats.TodayPnL.Float64(),
 		unrealized.Float64())
+
 	return unrealized.Add(c.profitStats.TodayPnL).Compare(c.lossThreshold) <= 0
 }

pkg/strategy/drift/strategy.go

@@ -831,9 +831,9 @@ func (s *Strategy) Run(ctx context.Context, orderExecutor bbgo.OrderExecutor, se
 		if madeProfit {
 			p := s.Position.NewProfit(trade, profit, netProfit)
 			s.Environment.RecordPosition(s.Position, trade, &p)
-			s.TradeStats.Add(&p)
+			s.TradeStats.AddProfit(&p)
 			s.ProfitStats.AddTrade(trade)
-			s.ProfitStats.AddProfit(p)
+			s.ProfitStats.AddProfit(&p)
 			bbgo.Notify(&p)
 			bbgo.Notify(s.ProfitStats)
 		}

pkg/strategy/fmaker/strategy.go

@@ -193,7 +193,7 @@ func (s *Strategy) Run(ctx context.Context, orderExecutor bbgo.OrderExecutor, se
 			p.StrategyInstanceID = instanceID
 			bbgo.Notify(&p)
 
-			s.ProfitStats.AddProfit(p)
+			s.ProfitStats.AddProfit(&p)
 			bbgo.Notify(&s.ProfitStats)
 
 			s.Environment.RecordPosition(s.Position, trade, &p)

pkg/strategy/wall/strategy.go

@@ -292,7 +292,7 @@ func (s *Strategy) Run(ctx context.Context, orderExecutor bbgo.OrderExecutor, se
 			p.StrategyInstanceID = instanceID
 			bbgo.Notify(&p)
 
-			s.ProfitStats.AddProfit(p)
+			s.ProfitStats.AddProfit(&p)
 			bbgo.Notify(&s.ProfitStats)
 
 			s.Environment.RecordPosition(s.Position, trade, &p)

pkg/strategy/xfunding/strategy.go

@@ -1097,7 +1097,7 @@ func (s *Strategy) allocateOrderExecutor(ctx context.Context, session *bbgo.Exch
 
 		if profit, netProfit, madeProfit := s.NeutralPosition.AddTrade(trade); madeProfit {
 			p := s.NeutralPosition.NewProfit(trade, profit, netProfit)
-			s.ProfitStats.AddProfit(p)
+			s.ProfitStats.AddProfit(&p)
 		}
 	})
 	return orderExecutor

pkg/strategy/xmaker/strategy.go

@@ -801,7 +801,7 @@ func (s *Strategy) CrossRun(ctx context.Context, orderExecutionRouter bbgo.Order
 			p.Strategy = ID
 			p.StrategyInstanceID = instanceID
 			bbgo.Notify(&p)
-			s.ProfitStats.AddProfit(p)
+			s.ProfitStats.AddProfit(&p)
 
 			s.Environment.RecordPosition(s.Position, trade, &p)
 		}