From 182200994d3ca0adbc6782b6a1b611e0c2f369c6 Mon Sep 17 00:00:00 2001
From: gezimbll <gezim.blliku@itsyscom.com>
Date: Fri, 20 Jun 2025 14:56:48 +0200
Subject: [PATCH] added benchmark for GetKeysForPrefix

---
 engine/storage_redis_test.go | 69 ++++++++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 engine/storage_redis_test.go

diff --git a/engine/storage_redis_test.go b/engine/storage_redis_test.go
new file mode 100644
index 000000000..e0d2f9120
--- /dev/null
+++ b/engine/storage_redis_test.go
@@ -0,0 +1,69 @@
+//go:build performance
+
+/*
+Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
+Copyright (C) ITsysCOM GmbH
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>
+*/
+
+package engine
+
+import (
+	"fmt"
+	"strconv"
+	"testing"
+	"time"
+)
+
+func BenchmarkRedisScan(b *testing.B) {
+	rs, err := NewRedisStorage("127.0.0.1:6379", 10, "cgrates", "", "json", 10, 20,
+		"", false, 5*time.Second, 0, 0, 0, 0, 0, 0, false, "", "", "")
+	if err != nil {
+		b.Fatalf("Failed to create Redis storage: %v", err)
+	}
+	defer rs.Close()
+	chargerProfile := &ChargerProfile{
+		ID:           "TestA_CHARGER1",
+		Tenant:       "cgrates.org",
+		FilterIDs:    []string{"*string:~*req.TestCase:AdminSAPIs"},
+		Weight:       10,
+		RunID:        "run1",
+		AttributeIDs: []string{"ATTR_ TEST1"},
+	}
+	id := "ChargerP"
+	var prfID string
+	for i := 0; i <= 20; i++ {
+		if i%10 == 0 {
+			if (i/10)%2 == 0 {
+				prfID = "TestA:"
+			} else {
+				prfID = "TestB:"
+			}
+		}
+		prfID = prfID[:6] + strconv.Itoa(i) + id
+		chargerProfile.ID = prfID
+		rs.SetChargerProfileDrv(chargerProfile)
+	}
+	prfx := []string{"TestA", "TestB", "Test"}
+	for _, v := range prfx {
+		b.Run(fmt.Sprintf("test case: prefix = %q", v), func(b *testing.B) {
+			for b.Loop() {
+				rs.GetKeysForPrefix(v)
+			}
+		})
+	}
+	rs.Flush("")
+
+}