如何使用 Golang 建立 RESTful API 並使用快取？

在使用 Golang 建立 RESTful API 時整合快取：使用 Gin 建立 API；整合 Redis 快取；定義一個快取層來管理快取值；在 Gin 處理函數中使用快取層從 Redis 取得和設定資料。

如何在使用Golang 建立RESTful API 時整合快取

建置RESTful API 時，快取可以大幅提高效能，減少伺服器負載並改善使用者體驗。本文將指導您使用 Golang 建立 RESTful API 並整合快取。

使用Gin 建立RESTful API

首先，使用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 並正在運作。

在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 會先從快取中取得回應，如果快取中不存在值，才會從資料庫中取得資料並將其儲存在快取中。這將顯著減少重複請求的資料庫查詢，從而提高效能。

以上是如何使用 Golang 建立 RESTful API 並使用快取？的詳細內容。更多資訊請關注PHP中文網其他相關文章！