Go+Redis缓存设计与优化实现

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"log"
	"time"

	"github.com/go-redis/redis/v8"
)

var ctx = context.Background()

// 商品结构体
type Product struct {
	ID    string `json:"id"`
	Name  string `json:"name"`
	Price float64 `json:"price"`
}

func getProductFromDB(productID string) (*Product, error) {
	// 模拟从数据库获取商品数据
	return &Product{
		ID:    productID,
		Name:  "Example Product",
		Price: 99.99,
	}, nil
}

func getProductFromCache(rdb *redis.Client, productID string) (*Product, error) {
	// 从缓存中获取商品数据
	val, err := rdb.Get(ctx, productID).Result()
	if err == redis.Nil {
		// 缓存未命中，查询数据库
		return nil, nil
	} else if err != nil {
		return nil, err
	}

	var product Product
	err = json.Unmarshal([]byte(val), &product)
	if err != nil {
		return nil, err
	}
	return &product, nil
}

func setProductToCache(rdb *redis.Client, product *Product) error {
	// 将商品数据缓存到 Redis
	productData, err := json.Marshal(product)
	if err != nil {
		return err
	}

	return rdb.Set(ctx, product.ID, productData, 10*time.Minute).Err()
}

func getProduct(rdb *redis.Client, productID string) (*Product, error) {
	// 尝试从缓存中获取商品
	product, err := getProductFromCache(rdb, productID)
	if err != nil {
		return nil, err
	}

	if product == nil {
		// 缓存未命中，查询数据库并将结果缓存
		product, err = getProductFromDB(productID)
		if err != nil {
			return nil, err
		}

		err = setProductToCache(rdb, product)
		if err != nil {
			return nil, err
		}
	}

	return product, nil
}

func main() {
	rdb := redis.NewClient(&redis.Options{
		Addr:     "localhost:6379", // Redis 地址
		Password: "",               // 密码
		DB:       0,                // 默认数据库
	})

	// 获取商品
	productID := "12345"
	product, err := getProduct(rdb, productID)
	if err != nil {
		log.Fatalf("获取商品失败: %v", err)
	}

	fmt.Printf("商品信息: %+v\n", product)
}