如何使用 Golang 构建 RESTful API 并使用缓存？

在使用 golang 构建 restful api 时集成缓存：使用 gin 创建 api；集成 redis 缓存；定义一个缓存层来管理缓存值；在 gin 处理函数中使用缓存层从 redis 获取和设置数据。

如何在使用 Golang 构建 RESTful API 时集成缓存

构建 RESTful API 时，缓存可以大大提高性能，减少服务器负载并改善用户体验。本文将指导您使用 Golang 构建 RESTful API 并集成缓存。

使用 Gin 创建 RESTful API

立即学习“go语言免费学习笔记（深入）”；

首先，使用 Gin 框架创建新的 Golang RESTful API：

import (
    "github.com/gin-gonic/gin"
)

func main() {
    r := gin.Default()
    r.GET("/api/v1/users", GetUsers)
    r.Run()
}

集成 Redis 缓存

要使用缓存，我们将使用 Redis，一种高度可用的键值存储。确保已安装 Redis 并正在运行。

狼群淘客 免费开源淘宝客程序

狼群淘客系统基于canphp框架进行开发，MVC结构、数据库碎片式缓存机制，使网站支持更大的负载量，结合淘宝开放平台API实现的一个淘宝客购物导航系统采用php+mysql实现，任何人都可以免费下载使用 。狼群淘客的任何代码都是不加密的，你不用担心会有任何写死的PID，不用担心你的劳动成果被窃取。

下载

在 Golang 中，我们可以使用 redigo 库连接到 Redis：

import (
    "time"

    "github.com/gomodule/redigo/redis"
)

// redisPool is a global variable to manage the Redis connection pool.
var redisPool *redis.Pool

func init() {
    redisPool = &redis.Pool{
        MaxIdle:   10,
        MaxActive: 10,
        IdleTimeout: 30 * time.Second,
        Dial: func() (redis.Conn, error) {
            return redis.Dial("tcp", "localhost:6379")
        },
    }
}

缓存层

现在，让我们定义一个新的层用于管理缓存。此层将提供获取、设置和删除缓存值的函数。

import (
    "time"

    "github.com/gomodule/redigo/redis"
)

type Cache struct {
    pool *redis.Pool
}

func NewCache(pool *redis.Pool) *Cache {
    return &Cache{
        pool: pool,
    }
}

func (c *Cache) Get(key string) ([]byte, error) {
    conn := c.pool.Get()
    defer conn.Close()

    return redis.Bytes(conn.Do("GET", key))
}

func (c *Cache) Set(key string, value string) error {
    conn := c.pool.Get()
    defer conn.Close()

    _, err := conn.Do("SET", key, value)
    return err
}

func (c *Cache) Delete(key string) error {
    conn := c.pool.Get()
    defer conn.Close()

    _, err := conn.Do("DEL", key)
    return err
}

使用缓存

在 Gin 处理函数中，我们可以使用缓存层来缓存 API 响应。以下是修改后的 GetUsers 处理函数：

func GetUsers(c *gin.Context) {
    // 获取 Redis 缓存
    key := "users"
    cachedBytes, err := cache.Get(key)
    if err != nil {
        // 缓存中没有值，从数据库中获取
        users, err := getUserFromDB()
        if err != nil {
            c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
            return
        }

        // 将 users 序列化为 JSON 并存储在缓存中
        userJSON, err := json.Marshal(users)
        if err != nil {
            c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
            return
        }

        err = cache.Set(key, string(userJSON))
        if err != nil {
            c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
            return
        }

        c.JSON(http.StatusOK, users)
    } else {
        // 从缓存中获取值，反序列化回 []User 类型
        var users []User
        err := json.Unmarshal(cachedBytes, &users)
        if err != nil {
            c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
            return
        }

        c.JSON(http.StatusOK, users)
    }
}

// getUserFromDB 从数据库中获取用户
func getUserFromDB() ([]User, error) {
    // 假设这是从数据库获取用户数据的模拟函数
    return []User{{ID: 1, Name: "John"}}, nil
}

现在，API 会首先从缓存中获取响应，如果缓存中不存在值，才会从数据库中获取数据并将其存储在缓存中。这将显着减少重复请求的数据库查询，从而提高性能。