✨ feat(sentinel): add historical kline querying

config/sentinel.yaml

@@ -17,3 +17,4 @@ exchangeStrategies:
     symbol: BTCUSDT
     interval: 1m
     scoreThreshold: 0.6
+    klineLimit: 43200

pkg/strategy/sentinel/strategy.go

@@ -2,10 +2,12 @@ package sentinel
 
 import (
 	"context"
+	"fmt"
 	"time"
 
 	"github.com/c9s/bbgo/pkg/bbgo"
 	"github.com/c9s/bbgo/pkg/datatype/floats"
+	"github.com/c9s/bbgo/pkg/exchange/batch"
 	"github.com/c9s/bbgo/pkg/types"
 	"github.com/narumiruna/go-iforest/pkg/iforest"
 	log "github.com/sirupsen/logrus"
@@ -31,6 +33,7 @@ type Strategy struct {
 
 	notificationRateLimiter *rate.Limiter
 	retrainingRateLimiter   *rate.Limiter
+	klines                  []types.KLine
 }
 
 func (s *Strategy) ID() string {
@@ -68,18 +71,22 @@ func (s *Strategy) Subscribe(session *bbgo.ExchangeSession) {
 }
 
 func (s *Strategy) Run(ctx context.Context, orderExecutor bbgo.OrderExecutor, session *bbgo.ExchangeSession) error {
+	err := s.queryHistoricalKlines(ctx, session)
+	if err != nil {
+		return err
+	}
+
 	session.MarketDataStream.OnKLine(types.KLineWith(s.Symbol, s.Interval, func(kline types.KLine) {
 		if !s.isMarketAvailable(session, s.Symbol) {
 			return
 		}
 
-		klines, err := s.queryKLines(ctx, session)
-		if err != nil {
-			log.Errorf("Unable to query klines: %v", err)
-			return
-		}
+		s.appendKline(kline)
+		s.klines = s.klines[len(s.klines)-s.KLineLimit:]
 
-		volumes := s.extractVolumes(klines)
+		log.Infof("kline length %d", len(s.klines))
+
+		volumes := s.extractVolumes(s.klines)
 		samples := s.generateSamples(volumes)
 
 		if s.shouldSkipIsolationForest(volumes, samples) {
@@ -99,13 +106,23 @@ func (s *Strategy) isMarketAvailable(session *bbgo.ExchangeSession, symbol strin
 	return ok
 }
 
-func (s *Strategy) queryKLines(ctx context.Context, session *bbgo.ExchangeSession) ([]types.KLine, error) {
+func (s *Strategy) queryHistoricalKlines(ctx context.Context, session *bbgo.ExchangeSession) error {
+	batchQuery := batch.KLineBatchQuery{Exchange: session.Exchange}
 	endTime := time.Now()
-	options := types.KLineQueryOptions{
-		Limit:   s.KLineLimit,
-		EndTime: &endTime,
+	startTime := endTime.Add(-time.Duration(s.KLineLimit) * s.Interval.Duration())
+	klineC, errC := batchQuery.Query(ctx, s.Symbol, s.Interval, startTime, endTime)
+	for {
+		select {
+		case <-ctx.Done():
+			return ctx.Err()
+
+		case kline, ok := <-klineC:
+			if !ok {
+				return <-errC
+			}
+			s.klines = append(s.klines, kline)
+		}
 	}
-	return session.Exchange.QueryKLines(ctx, s.Symbol, s.Interval, options)
 }
 
 func (s *Strategy) extractVolumes(klines []types.KLine) floats.Slice {
@@ -159,3 +176,22 @@ func (s *Strategy) notifyOnIsolationForestScore(scores []float64, kline types.KL
 		}
 	}
 }
+
+func (s *Strategy) appendKline(kline types.KLine) error {
+	if len(s.klines) == 0 {
+		return fmt.Errorf("klines is empty")
+	}
+
+	lastKline := s.klines[len(s.klines)-1]
+	if lastKline.Symbol != kline.Symbol {
+		return fmt.Errorf("last kline symbol %s is not equal to current kline symbol %s", lastKline.Symbol, kline.Symbol)
+	}
+
+	if lastKline.EndTime.After(kline.EndTime.Time()) {
+		return fmt.Errorf("last kline end time %s is after current kline end time %s", lastKline.EndTime, kline.EndTime)
+	}
+
+	s.klines = append(s.klines, kline)
+
+	return nil
+}