General library improvements.

cache_compression_bench_test.go

@@ -0,0 +1,172 @@
+package gql
+
+import (
+	"testing"
+	"time"
+
+	cache "github.com/lukaszraczylo/go-simple-graphql/cache"
+)
+
+// Benchmark cache compression with different data sizes
+func BenchmarkCacheCompression(b *testing.B) {
+	c := cache.New(5 * time.Second) // Use shorter TTL for tests
+	defer c.Stop()
+
+	// Small data (under 1KB threshold)
+	smallData := make([]byte, 512)
+	for i := range smallData {
+		smallData[i] = byte(i % 256)
+	}
+
+	// Medium data (over 1KB threshold)
+	mediumData := make([]byte, 2048)
+	for i := range mediumData {
+		mediumData[i] = byte(i % 256)
+	}
+
+	// Large data (much larger than threshold)
+	largeData := make([]byte, 8192)
+	for i := range largeData {
+		largeData[i] = byte(i % 256)
+	}
+
+	// Highly compressible data
+	compressibleData := make([]byte, 4096)
+	for i := range compressibleData {
+		compressibleData[i] = 'A' // Repeating character compresses well
+	}
+
+	b.Run("SmallData_Set", func(b *testing.B) {
+		b.ResetTimer()
+		for i := 0; i < b.N; i++ {
+			c.Set("small", smallData, time.Minute)
+		}
+	})
+
+	b.Run("SmallData_Get", func(b *testing.B) {
+		c.Set("small", smallData, time.Minute)
+		b.ResetTimer()
+		for i := 0; i < b.N; i++ {
+			_, _ = c.Get("small")
+		}
+	})
+
+	b.Run("MediumData_Set", func(b *testing.B) {
+		b.ResetTimer()
+		for i := 0; i < b.N; i++ {
+			c.Set("medium", mediumData, time.Minute)
+		}
+	})
+
+	b.Run("MediumData_Get", func(b *testing.B) {
+		c.Set("medium", mediumData, time.Minute)
+		b.ResetTimer()
+		for i := 0; i < b.N; i++ {
+			_, _ = c.Get("medium")
+		}
+	})
+
+	b.Run("LargeData_Set", func(b *testing.B) {
+		b.ResetTimer()
+		for i := 0; i < b.N; i++ {
+			c.Set("large", largeData, time.Minute)
+		}
+	})
+
+	b.Run("LargeData_Get", func(b *testing.B) {
+		c.Set("large", largeData, time.Minute)
+		b.ResetTimer()
+		for i := 0; i < b.N; i++ {
+			_, _ = c.Get("large")
+		}
+	})
+
+	b.Run("CompressibleData_Set", func(b *testing.B) {
+		b.ResetTimer()
+		for i := 0; i < b.N; i++ {
+			c.Set("compressible", compressibleData, time.Minute)
+		}
+	})
+
+	b.Run("CompressibleData_Get", func(b *testing.B) {
+		c.Set("compressible", compressibleData, time.Minute)
+		b.ResetTimer()
+		for i := 0; i < b.N; i++ {
+			_, _ = c.Get("compressible")
+		}
+	})
+}
+
+// Benchmark cache memory usage with compression
+func BenchmarkCacheMemoryUsage(b *testing.B) {
+	c := cache.New(5 * time.Second) // Use shorter TTL for tests
+	defer c.Stop()
+
+	// Test data that compresses well
+	testData := make([]byte, 4096)
+	for i := range testData {
+		testData[i] = byte(i%10 + '0') // Repeating pattern
+	}
+
+	b.Run("WithCompression", func(b *testing.B) {
+		b.ResetTimer()
+		for i := 0; i < b.N; i++ {
+			key := "key" + string(rune(i%1000))
+			c.Set(key, testData, time.Minute)
+		}
+	})
+}
+
+// Test compression effectiveness
+func TestCompressionEffectiveness(t *testing.T) {
+	c := cache.New(5 * time.Second) // Use shorter TTL for tests
+	defer c.Stop()
+
+	// Highly compressible data
+	compressibleData := make([]byte, 4096)
+	for i := range compressibleData {
+		compressibleData[i] = 'A'
+	}
+
+	// Random data (less compressible)
+	randomData := make([]byte, 4096)
+	for i := range randomData {
+		randomData[i] = byte(i % 256)
+	}
+
+	// Small data (under threshold)
+	smallData := make([]byte, 512)
+	for i := range smallData {
+		smallData[i] = byte(i % 256)
+	}
+
+	// Test compressible data
+	c.Set("compressible", compressibleData, time.Minute)
+	retrieved, ok := c.Get("compressible")
+	if !ok {
+		t.Error("Failed to retrieve compressible data")
+	}
+	if len(retrieved) != len(compressibleData) {
+		t.Errorf("Data length mismatch: expected %d, got %d", len(compressibleData), len(retrieved))
+	}
+
+	// Test random data
+	c.Set("random", randomData, time.Minute)
+	retrieved, ok = c.Get("random")
+	if !ok {
+		t.Error("Failed to retrieve random data")
+	}
+	if len(retrieved) != len(randomData) {
+		t.Errorf("Data length mismatch: expected %d, got %d", len(randomData), len(retrieved))
+	}
+
+	// Test small data (should not be compressed)
+	c.Set("small", smallData, time.Minute)
+	retrieved, ok = c.Get("small")
+	if !ok {
+		t.Error("Failed to retrieve small data")
+	}
+	if len(retrieved) != len(smallData) {
+		t.Errorf("Data length mismatch: expected %d, got %d", len(smallData), len(retrieved))
+	}
+}