xgap: share sourceKLineHandler

c9s · Nov 15, 2024 · 050d1ca · 050d1ca
1 parent e67a6c5
commit 050d1ca
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/pkg/strategy/xgap/strategy.go b/pkg/strategy/xgap/strategy.go
@@ -159,16 +159,13 @@ func (s *Strategy) CrossRun(ctx context.Context, _ bbgo.OrderExecutionRouter, se
 	})
 
 	// from here, set data binding
-	s.sourceSession.MarketDataStream.OnKLine(func(kline types.KLine) {
+	sourceKLineHandler := func(kline types.KLine) {
 		s.mu.Lock()
 		s.lastSourceKLine = kline
 		s.mu.Unlock()
-	})
-	s.tradingSession.MarketDataStream.OnKLine(func(kline types.KLine) {
-		s.mu.Lock()
-		s.lastTradingKLine = kline
-		s.mu.Unlock()
-	})
+	}
+	s.sourceSession.MarketDataStream.OnKLine(sourceKLineHandler)
+	s.tradingSession.MarketDataStream.OnKLine(sourceKLineHandler)
 
 	if s.SourceExchange != "" && s.SourceSymbol != "" {
 		s.sourceBook = types.NewStreamBook(s.SourceSymbol, sourceSession.ExchangeName)