通过中间件扩展golang框架功能- 技术经验 -卓越飞翔博客

go 中间件是扩展框架功能的机制，允许拦截和修改 http 请求和响应。创建中间件需实现 http.handler 接口。将其注册到框架中可通过 use 函数。常用案例包括限制请求速率和添加认证逻辑。

通过中间件扩展golang框架功能

通过中间件扩展 Go 框架功能

简介

Go 框架（如 Gin、Echo 和 Fiber）提供了强大的功能集，但有时我们需要扩展它们的默认行为或添加自定义逻辑。为此，Go 提供了一种称为中间件的机制。

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

什么是中间件？

中间件是处理 HTTP 请求和响应的函数或结构，在应用程序处理请求之前或之后执行。它们允许我们拦截请求和响应，并根据需要修改它们。

创建中间件

创建中间件非常简单。它是一个实现 http.Handler 接口的函数或结构：

func myMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 自定义逻辑

        // 调用下一个处理程序
        next.ServeHTTP(w, r)
    })
}

注册中间件

要注册中间件，我们可以在框架的配置期间调用 Use 函数：

// Gin 示例

r := gin.New()
r.Use(myMiddleware)

实战案例

限制请求速率

这是一个使用中间件限制请求速率的实战案例：

func rateLimitMiddleware(next http.Handler) http.Handler {
    // 创建令牌桶
    bucket := tokenBucket{
        capacity: 10,
        rate:     1 / time.Second,
    }

    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 获取令牌
        if !bucket.Take() {
            // 超出速率限制，拒绝请求
            http.Error(w, "Too Many Requests", http.StatusTooManyRequests)
            return
        }

        // 调用下一个处理程序
        next.ServeHTTP(w, r)
    })
}

相关推荐