golang框架如何使用中间件？-Golang-PHP中文网

golang框架如何使用中间件？

PHPz

发布： 2024-06-10 17:45:01

原创

635人浏览过

go 框架中的中间件是增强应用程序功能的工具，允许在请求处理过程中拦截和修改请求和响应。使用步骤如下：创建中间件函数，处理请求和响应，并返回一个 http 处理程序。在应用程序的路由器中注册中间件，可以为单个路由注册多个中间件。实战案例：为 api 添加身份验证，使用中间件来验证请求中的 jwt 令牌，并从 token 中提取用户信息添加到上下文中。

golang框架如何使用中间件？

Go 框架中的中间件指南

中间件是一种强大的工具，可用于增强 Go 应用程序的功能，它允许您在请求处理过程中拦截和修改请求和响应。这是如何使用它们：

1. 创建中间件函数

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

创建一个函数来处理请求和响应。这个函数应接受 http.Handler 并返回 http.Handler，例如：

func AuthMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 验证请求中的令牌
        err := verifyToken(r)
        if err != nil {
            http.Error(w, "Unauthorized", http.StatusUnauthorized)
            return
        }
        // 调用下一个处理程序
        next.ServeHTTP(w, r)
    })
}

登录后复制

2. 注册中间件

Zend Framework 2.4.3 完整版本

Zend框架2是一个开源框架,使用PHP 5.3 +开发web应用程序和服务。Zend框架2使用100%面向对象代码和利用大多数PHP 5.3的新特性,即名称空间、延迟静态绑定,lambda函数和闭包。 Zend框架2的组成结构是独一无二的;每个组件被设计与其他部件数的依赖关系。 ZF2遵循SOLID面向对象的设计原则。这样的松耦合结构可以让开发人员使用他们想要的任何部件。我们称之为“松耦合”

344

查看详情

在应用程序的路由器中注册中间件。您可以在单个路由上注册多个中间件，例如：

func main() {
    router := mux.NewRouter()
    router.Use(AuthMiddleware) // 全局应用
    router.HandleFunc("/users", UserHandler).Methods("GET")
    // ... 其余路由
}

登录后复制

3. 实战案例

需求： 为 API 添加身份验证，要求用户在访问受保护的端点之前进行身份验证。

解决方案： 使用中间件来验证请求中的 JWT 令牌：

func AuthMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 从请求中提取令牌
        tokenHeader := r.Header.Get("Authorization")
        if tokenHeader == "" {
            http.Error(w, "Unauthorized", http.StatusUnauthorized)
            return
        }

        // 验证令牌
        claims, err := verifyJWT(tokenHeader)
        if err != nil || !claims.Valid() {
            http.Error(w, "Unauthorized", http.StatusUnauthorized)
            return
        }

        // 添加用户 ID 到上下文
        ctx := context.WithValue(r.Context(), "user_id", claims.Subject)
        r = r.WithContext(ctx)

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

登录后复制

注意：