go-rejson: Redis中JSON数据结构存储与操作指南

go-rejson是一个用于在Redis中存储和操作JSON数据的Golang库，它基于Redis的ReJSON模块。下面我将详细介绍如何使用go-rejson进行JSON数据的存储和操作。

安装

首先需要安装go-rejson库和Redis服务器（确保Redis已加载ReJSON模块）：

go get github.com/nitishm/go-rejson/v4

基本使用

1. 初始化连接

package main

import (
	"context"
	"fmt"
	"log"

	"github.com/go-redis/redis/v8"
	"github.com/nitishm/go-rejson/v4"
)

func main() {
	// 创建Redis客户端
	rdb := redis.NewClient(&redis.Options{
		Addr: "localhost:6379",
	})

	// 创建ReJSON处理器
	rh := rejson.NewReJSONHandler()
	rh.SetGoRedisClient(rdb)

	ctx := context.Background()

	// 测试连接
	if _, err := rdb.Ping(ctx).Result(); err != nil {
		log.Fatalf("Failed to connect to Redis: %v", err)
	}
}

2. 存储JSON数据

type User struct {
	Name    string `json:"name"`
	Age     int    `json:"age"`
	Address struct {
		City    string `json:"city"`
		Country string `json:"country"`
	} `json:"address"`
}

func setJSON(rh *rejson.Handler) {
	ctx := context.Background()
	
	user := User{
		Name: "John Doe",
		Age:  30,
		Address: struct {
			City    string `json:"city"`
			Country string `json:"country"`
		}{
			City:    "New York",
			Country: "USA",
		},
	}

	// 设置JSON对象到Redis
	res, err := rh.JSONSet("user:123", ".", user)
	if err != nil {
		log.Fatalf("Failed to JSONSet: %v", err)
	}
	
	if res.(string) == "OK" {
		fmt.Println("Success: user:123")
	}
}

3. 获取JSON数据

func getJSON(rh *rejson.Handler) {
	ctx := context.Background()
	
	var user User
	// 获取整个JSON对象
	_, err := rh.JSONGet("user:123", ".", &user)
	if err != nil {
		log.Fatalf("Failed to JSONGet: %v", err)
	}
	
	fmt.Printf("User: %+v\n", user)
	
	// 获取特定字段
	var name string
	_, err = rh.JSONGet("user:123", ".name", &name)
	if err != nil {
		log.Fatalf("Failed to JSONGet name: %v", err)
	}
	fmt.Printf("Name: %s\n", name)
}

4. 更新JSON数据

func updateJSON(rh *rejson.Handler) {
	ctx := context.Background()
	
	// 更新单个字段
	res, err := rh.JSONSet("user:123", ".age", 31)
	if err != nil {
		log.Fatalf("Failed to update age: %v", err)
	}
	
	if res.(string) == "OK" {
		fmt.Println("Successfully updated age")
	}
	
	// 更新嵌套字段
	res, err = rh.JSONSet("user:123", ".address.city", "Boston")
	if err != nil {
		log.Fatalf("Failed to update city: %v", err)
	}
	
	if res.(string) == "OK" {
		fmt.Println("Successfully updated city")
	}
}

5. 数组操作

func arrayOperations(rh *rejson.Handler) {
	ctx := context.Background()
	
	// 创建包含数组的JSON
	type Post struct {
		Title   string   `json:"title"`
		Tags    []string `json:"tags"`
		Likes   int      `json:"likes"`
	}
	
	post := Post{
		Title: "Go Redis JSON",
		Tags:  []string{"golang", "redis", "json"},
		Likes: 0,
	}
	
	_, err := rh.JSONSet("post:1", ".", post)
	if err != nil {
		log.Fatalf("Failed to JSONSet post: %v", err)
	}
	
	// 向数组添加元素
	_, err = rh.JSONArrAppend("post:1", ".tags", "database")
	if err != nil {
		log.Fatalf("Failed to append to array: %v", err)
	}
	
	// 获取数组长度
	length, err := rh.JSONArrLen("post:1", ".tags")
	if err != nil {
		log.Fatalf("Failed to get array length: %v", err)
	}
	fmt.Printf("Tags array length: %d\n", length)
	
	// 获取数组元素
	var tags []string
	_, err = rh.JSONGet("post:1", ".tags", &tags)
	if err != nil {
		log.Fatalf("Failed to get tags: %v", err)
	}
	fmt.Printf("Tags: %v\n", tags)
}

6. 删除操作

func deleteOperations(rh *rejson.Handler) {
	ctx := context.Background()
	
	// 删除特定字段
	_, err := rh.JSONDel("user:123", ".address.city")
	if err != nil {
		log.Fatalf("Failed to delete city: %v", err)
	}
	
	// 删除整个JSON对象
	_, err = rh.JSONDel("post:1", ".")
	if err != nil {
		log.Fatalf("Failed to delete post: %v", err)
	}
}

高级特性

1. 批量操作

func batchOperations(rh *rejson.Handler) {
	ctx := context.Background()
	
	// 使用MULTI/EXEC进行批量操作
	pipe := rdb.TxPipeline()
	
	rh.JSONSetWithOptions("user:123", ".", User{Name: "Batch User", Age: 25}, rejson.SetOptions{})
	rh.JSONSetWithOptions("user:123", ".age", 26, rejson.SetOptions{})
	
	_, err := pipe.Exec(ctx)
	if err != nil {
		log.Fatalf("Failed batch operations: %v", err)
	}
}

2. 设置过期时间

func setWithExpire(rh *rejson.Handler) {
	ctx := context.Background()
	
	tempData := map[string]interface{}{
		"token": "abc123",
		"valid": true,
	}
	
	// 设置JSON并添加过期时间
	_, err := rh.JSONSet("temp:session", ".", tempData)
	if err != nil {
		log.Fatalf("Failed to set temp data: %v", err)
	}
	
	// 设置60秒过期
	if err := rdb.Expire(ctx, "temp:session", 60*time.Second).Err(); err != nil {
		log.Fatalf("Failed to set expire: %v", err)
	}
}

最佳实践

1. 键命名规范：使用冒号分隔的命名空间（如user:123, post:456）
2. 合理使用路径：.表示根路径，.field表示特定字段
3. 错误处理：始终检查返回的错误
4. 连接池：重用Redis连接以提高性能
5. 数据结构设计：根据查询模式设计JSON结构

go-rejson提供了强大的功能来操作Redis中的JSON数据，结合Redis的高性能和JSON的灵活性，非常适合需要复杂数据结构和快速访问的场景。