FIX: fix memory leak

c9s · Sep 4, 2024 · 0e68cb1 · 0e68cb1
1 parent 50262f2
commit 0e68cb1
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/pkg/bbgo/marketdatastore.go b/pkg/bbgo/marketdatastore.go
@@ -5,6 +5,8 @@ import "github.com/c9s/bbgo/pkg/types"
 const MaxNumOfKLines = 5_000
 const MaxNumOfKLinesTruncate = 100
 
+const CapacityOfKLineWindowLimit = 5_000
+
 // MarketDataStore receives and maintain the public market data of a single symbol
 //go:generate callbackgen -type MarketDataStore
 type MarketDataStore struct {
@@ -57,8 +59,13 @@ func (store *MarketDataStore) AddKLine(k types.KLine) {
 	}
 	window.Add(k)
 
-	if len(*window) > MaxNumOfKLines {
-		*window = (*window)[MaxNumOfKLinesTruncate-1:]
+	lenOfWindow := len(*window)
+	capOfWindow := cap(*window)
+
+	if lenOfWindow == capOfWindow && capOfWindow > CapacityOfKLineWindowLimit {
+		truncatedWindow := (*window)[(CapacityOfKLineWindowLimit / 2):]
+		copy(*window, truncatedWindow)
+		*window = (*window)[:len(truncatedWindow)]
 	}
 
 	store.EmitKLineClosed(k)