Benchmark: Close redis client on each test iteration (-count=10)

go-chi · Aug 8, 2024 · a2a09ff · a2a09ff
1 parent c164784
commit a2a09ff
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/httprateredis.go b/httprateredis.go
@@ -182,6 +182,10 @@ func (c *redisCounter) IsFallbackActivated() bool {
 	return c.fallbackActivated.Load()
 }
 
+func (c *redisCounter) Close() error {
+	return c.client.Close()
+}
+
 func (c *redisCounter) shouldFallback(err error) bool {
 	if err == nil {
 		return false

diff --git a/httprateredis_test.go b/httprateredis_test.go
@@ -26,6 +26,7 @@ func TestRedisCounter(t *testing.T) {
 	if err != nil {
 		t.Fatalf("redis not available: %v", err)
 	}
+	defer limitCounter.Close()
 
 	limitCounter.Config(1000, time.Minute)
 
@@ -166,6 +167,7 @@ func BenchmarkLocalCounter(b *testing.B) {
 	if err != nil {
 		b.Fatalf("redis not available: %v", err)
 	}
+	defer limitCounter.Close()
 
 	limitCounter.Config(1000, time.Minute)