初始化缓存管理器

manager/cache_benchmark_test.go

@@ -0,0 +1,129 @@
+package manager_test
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"testing"
+	"time"
+
+	"code.yun.ink/pkg/cache_manager/manager"
+
+	"github.com/redis/go-redis/v9"
+)
+
+// 本地缓存压测
+type testStruct struct {
+	A int
+	B string
+}
+
+func getRedisClient() *redis.Client {
+	// 支持通过环境变量覆盖 redis 地址
+	addr := os.Getenv("TEST_REDIS_ADDR")
+	if addr == "" {
+		addr = "10.40.92.54:30379"
+	}
+	client := redis.NewClient(&redis.Options{
+		Addr:     addr,
+		Password: "123000",
+		DB:       0,
+	})
+	return client
+}
+
+func BenchmarkCacheLocal_SetGet(b *testing.B) {
+	cache := manager.NewCacheLocal(time.Minute)
+
+	ctx := context.Background()
+	defer cache.Stop()
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		key := "key" + fmt.Sprintf("%d", i)
+		val := testStruct{A: i, B: "val"}
+		_ = cache.Set(ctx, key, val, time.Minute, []int64{int64(i)})
+		var out testStruct
+		_ = cache.Get(ctx, key, &out)
+	}
+}
+
+func BenchmarkCacheLocal_Parallel(b *testing.B) {
+	cache := manager.NewCacheLocal(time.Minute)
+	ctx := context.Background()
+	defer cache.Stop()
+	b.RunParallel(func(pb *testing.PB) {
+		i := 0
+		for pb.Next() {
+			key := "key" + fmt.Sprintf("%d", i)
+			val := testStruct{A: i, B: "val"}
+			_ = cache.Set(ctx, key, val, time.Minute, []int64{int64(i)})
+			var out testStruct
+			_ = cache.Get(ctx, key, &out)
+			i++
+		}
+	})
+}
+
+// Redis Hash 缓存压测
+func BenchmarkCacheRedisHash_SetGet(b *testing.B) {
+	client := getRedisClient()
+	cache := manager.NewCacheRedisHash(client, "bench_hash", time.Hour)
+	ctx := context.Background()
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		key := "key" + fmt.Sprintf("%d", i)
+		val := testStruct{A: i, B: "val"}
+		_ = cache.Set(ctx, key, val, time.Hour, []int64{int64(i)})
+		var out testStruct
+		_ = cache.Get(ctx, key, &out)
+	}
+}
+
+func BenchmarkCacheRedisHash_Parallel(b *testing.B) {
+	client := getRedisClient()
+	cache := manager.NewCacheRedisHash(client, "bench_hash", time.Hour)
+	ctx := context.Background()
+	b.RunParallel(func(pb *testing.PB) {
+		i := 0
+		for pb.Next() {
+			key := "key" + fmt.Sprintf("%d", i)
+			val := testStruct{A: i, B: "val"}
+			_ = cache.Set(ctx, key, val, time.Hour, []int64{int64(i)})
+			var out testStruct
+			_ = cache.Get(ctx, key, &out)
+			i++
+		}
+	})
+}
+
+// Redis String 缓存压测
+func BenchmarkCacheRedis_SetGet(b *testing.B) {
+	client := getRedisClient()
+	ctx := context.Background()
+	cache := manager.NewCacheRedis(ctx, client, "bench_str")
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		key := "key" + fmt.Sprintf("%d", i)
+		val := testStruct{A: i, B: "val"}
+		_ = cache.Set(ctx, key, val, time.Hour, []int64{int64(i)})
+		var out testStruct
+		_ = cache.Get(ctx, key, &out)
+	}
+}
+
+func BenchmarkCacheRedis_Parallel(b *testing.B) {
+	client := getRedisClient()
+	ctx := context.Background()
+	cache := manager.NewCacheRedis(ctx, client, "bench_str")
+	b.RunParallel(func(pb *testing.PB) {
+		i := 0
+		for pb.Next() {
+			key := "key" + fmt.Sprintf("%d", i)
+			val := testStruct{A: i, B: "val"}
+			_ = cache.Set(ctx, key, val, time.Hour, []int64{int64(i)})
+			var out testStruct
+			_ = cache.Get(ctx, key, &out)
+			i++
+		}
+	})
+}