commit
This commit is contained in:
sjk
168
go_backend/cmd/test_redis.go
Normal file
168
go_backend/cmd/test_redis.go
Normal file
@@ -0,0 +1,168 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"ai_xhs/config"
|
||||
"ai_xhs/database"
|
||||
"ai_xhs/utils"
|
||||
"context"
|
||||
"fmt"
|
||||
"log"
|
||||
"time"
|
||||
)
|
||||
|
||||
// TestData 测试数据结构
|
||||
type TestData struct {
|
||||
Name string `json:"name"`
|
||||
Age int `json:"age"`
|
||||
Email string `json:"email"`
|
||||
}
|
||||
|
||||
func main() {
|
||||
// 加载配置
|
||||
if err := config.LoadConfig("dev"); err != nil {
|
||||
log.Fatalf("配置加载失败: %v", err)
|
||||
}
|
||||
|
||||
// 初始化Redis
|
||||
if err := database.InitRedis(); err != nil {
|
||||
log.Fatalf("Redis初始化失败: %v", err)
|
||||
}
|
||||
defer database.CloseRedis()
|
||||
|
||||
ctx := context.Background()
|
||||
|
||||
fmt.Println("\n=== Redis缓存功能测试 ===\n")
|
||||
|
||||
// 1. 测试基本的Set/Get
|
||||
fmt.Println("1. 测试基本Set/Get操作:")
|
||||
testData := TestData{
|
||||
Name: "张三",
|
||||
Age: 25,
|
||||
Email: "zhangsan@example.com",
|
||||
}
|
||||
|
||||
if err := utils.SetCache(ctx, "user:1001", testData, 5*time.Minute); err != nil {
|
||||
log.Fatalf("设置缓存失败: %v", err)
|
||||
}
|
||||
fmt.Println("✓ 缓存设置成功")
|
||||
|
||||
var result TestData
|
||||
if err := utils.GetCache(ctx, "user:1001", &result); err != nil {
|
||||
log.Fatalf("获取缓存失败: %v", err)
|
||||
}
|
||||
fmt.Printf("✓ 缓存获取成功: %+v\n", result)
|
||||
|
||||
// 2. 测试Exists
|
||||
fmt.Println("\n2. 测试缓存是否存在:")
|
||||
exists, err := utils.ExistsCache(ctx, "user:1001")
|
||||
if err != nil {
|
||||
log.Fatalf("检查缓存失败: %v", err)
|
||||
}
|
||||
fmt.Printf("✓ 缓存存在性检查: %v\n", exists)
|
||||
|
||||
// 3. 测试TTL
|
||||
fmt.Println("\n3. 测试获取TTL:")
|
||||
ttl, err := utils.GetTTL(ctx, "user:1001")
|
||||
if err != nil {
|
||||
log.Fatalf("获取TTL失败: %v", err)
|
||||
}
|
||||
fmt.Printf("✓ 缓存TTL: %v\n", ttl)
|
||||
|
||||
// 4. 测试计数器
|
||||
fmt.Println("\n4. 测试计数器操作:")
|
||||
count, err := utils.IncrCache(ctx, "counter:test")
|
||||
if err != nil {
|
||||
log.Fatalf("递增失败: %v", err)
|
||||
}
|
||||
fmt.Printf("✓ 递增后计数: %d\n", count)
|
||||
|
||||
count, err = utils.IncrCache(ctx, "counter:test")
|
||||
if err != nil {
|
||||
log.Fatalf("递增失败: %v", err)
|
||||
}
|
||||
fmt.Printf("✓ 再次递增后计数: %d\n", count)
|
||||
|
||||
// 5. 测试Hash操作
|
||||
fmt.Println("\n5. 测试Hash操作:")
|
||||
if err := utils.HSetCache(ctx, "user:hash:1001", "name", "李四"); err != nil {
|
||||
log.Fatalf("HSet失败: %v", err)
|
||||
}
|
||||
if err := utils.HSetCache(ctx, "user:hash:1001", "age", "30"); err != nil {
|
||||
log.Fatalf("HSet失败: %v", err)
|
||||
}
|
||||
fmt.Println("✓ Hash字段设置成功")
|
||||
|
||||
name, err := utils.HGetCache(ctx, "user:hash:1001", "name")
|
||||
if err != nil {
|
||||
log.Fatalf("HGet失败: %v", err)
|
||||
}
|
||||
fmt.Printf("✓ 获取Hash字段 name: %s\n", name)
|
||||
|
||||
allFields, err := utils.HGetAllCache(ctx, "user:hash:1001")
|
||||
if err != nil {
|
||||
log.Fatalf("HGetAll失败: %v", err)
|
||||
}
|
||||
fmt.Printf("✓ 获取所有Hash字段: %+v\n", allFields)
|
||||
|
||||
// 6. 测试Set操作
|
||||
fmt.Println("\n6. 测试Set操作:")
|
||||
if err := utils.SAddCache(ctx, "tags:1001", "golang", "redis", "mysql"); err != nil {
|
||||
log.Fatalf("SAdd失败: %v", err)
|
||||
}
|
||||
fmt.Println("✓ Set成员添加成功")
|
||||
|
||||
members, err := utils.SMembersCache(ctx, "tags:1001")
|
||||
if err != nil {
|
||||
log.Fatalf("SMembers失败: %v", err)
|
||||
}
|
||||
fmt.Printf("✓ 获取Set所有成员: %v\n", members)
|
||||
|
||||
// 7. 测试ZSet操作
|
||||
fmt.Println("\n7. 测试ZSet操作:")
|
||||
if err := utils.ZAddCache(ctx, "rank:score", 100.5, "user1"); err != nil {
|
||||
log.Fatalf("ZAdd失败: %v", err)
|
||||
}
|
||||
if err := utils.ZAddCache(ctx, "rank:score", 95.0, "user2"); err != nil {
|
||||
log.Fatalf("ZAdd失败: %v", err)
|
||||
}
|
||||
if err := utils.ZAddCache(ctx, "rank:score", 105.5, "user3"); err != nil {
|
||||
log.Fatalf("ZAdd失败: %v", err)
|
||||
}
|
||||
fmt.Println("✓ ZSet成员添加成功")
|
||||
|
||||
rangeResult, err := utils.ZRangeCache(ctx, "rank:score", 0, -1)
|
||||
if err != nil {
|
||||
log.Fatalf("ZRange失败: %v", err)
|
||||
}
|
||||
fmt.Printf("✓ 获取ZSet所有成员(按分数排序): %v\n", rangeResult)
|
||||
|
||||
// 8. 测试删除操作
|
||||
fmt.Println("\n8. 测试删除操作:")
|
||||
if err := utils.DelCache(ctx, "counter:test", "tags:1001", "rank:score"); err != nil {
|
||||
log.Fatalf("删除缓存失败: %v", err)
|
||||
}
|
||||
fmt.Println("✓ 缓存删除成功")
|
||||
|
||||
// 9. 测试SetNX (仅当key不存在时设置)
|
||||
fmt.Println("\n9. 测试SetNX操作:")
|
||||
success, err := utils.SetCacheNX(ctx, "lock:test", "locked", 10*time.Second)
|
||||
if err != nil {
|
||||
log.Fatalf("SetNX失败: %v", err)
|
||||
}
|
||||
fmt.Printf("✓ SetNX首次设置: %v\n", success)
|
||||
|
||||
success, err = utils.SetCacheNX(ctx, "lock:test", "locked", 10*time.Second)
|
||||
if err != nil {
|
||||
log.Fatalf("SetNX失败: %v", err)
|
||||
}
|
||||
fmt.Printf("✓ SetNX重复设置(应该失败): %v\n", success)
|
||||
|
||||
// 清理测试数据
|
||||
fmt.Println("\n10. 清理测试数据:")
|
||||
if err := utils.DelCache(ctx, "user:1001", "user:hash:1001", "lock:test"); err != nil {
|
||||
log.Fatalf("清理失败: %v", err)
|
||||
}
|
||||
fmt.Println("✓ 测试数据清理完成")
|
||||
|
||||
fmt.Println("\n=== 所有测试完成! ===")
|
||||
}
|
||||
Reference in New Issue
Block a user