From 52523d0a73714660c1cb1991118b7a8a3459dbbe Mon Sep 17 00:00:00 2001
From: DanB <danb@cgrates.org>
Date: Tue, 24 Sep 2024 16:32:50 +0200
Subject: [PATCH] TrendS - cleanup method

---
 engine/libtrends.go | 35 ++++++++++++++++++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/engine/libtrends.go b/engine/libtrends.go
index 8ed69a52f..f355fb805 100644
--- a/engine/libtrends.go
+++ b/engine/libtrends.go
@@ -105,7 +105,40 @@ type Trend struct {
 	mCounts map[string]int       // number of times a metric is present in Metrics
 	mTotals map[string]float64   // cached sum, used for average calculations
 
-	tP *TrendProfile // cache here the settings
+}
+
+// cleanup will clean stale data out of
+func (t *Trend) cleanup(ttl time.Duration, qLength int) (altered bool) {
+	expTime := time.Now().Add(-ttl)
+	var expIdx *int
+	for i, rT := range t.RunTimes {
+		if rT.After(expTime) {
+			continue
+		}
+		expIdx = &i
+		delete(t.Metrics, rT)
+	}
+	if expIdx != nil {
+		if len(t.RunTimes)-1 == *expIdx {
+			t.RunTimes = make([]time.Time, 0)
+		} else {
+			t.RunTimes = t.RunTimes[*expIdx+1:]
+		}
+		altered = true
+	}
+	diffLen := len(t.RunTimes) - qLength
+	if qLength > 0 && diffLen > 0 {
+		var rmTms []time.Time
+		rmTms, t.RunTimes = t.RunTimes[:diffLen], t.RunTimes[diffLen:]
+		for _, rmTm := range rmTms {
+			delete(t.Metrics, rmTm)
+		}
+		altered = true
+	}
+	if altered {
+		t.computeIndexes()
+	}
+	return
 }
 
 // computeIndexes should be called after each retrieval from DB