如何自定义 Golang 框架的中间件？-码农资源网

正文概述

如何自定义 golang 框架的中间件？实现自定义中间件：编写一个接受 http.handler 并返回 http.handler 的函数，该函数将作为中间件的处理函数。实战案例：例如，authmiddleware 将检查请求头中的授权信息，如果 token 无效，则返回 401 状态码。注册中间件：使用 use 方法在框架中注册自定义中间件，它们将在每个请求处理过程中执行。

如何自定义 Golang 框架的中间件？

中间件是一种在请求和响应处理期间执行的软件组件，它允许在不同请求阶段执行特定的操作。在 Golang 框架中，中间件可以通过编写符合 HandlerFunc 类型的函数来实现。

实现自定义中间件

要实现自定义中间件，需要编写一个接受 http.Handler 并返回 http.Handler 的函数。这个函数将成为中间件的处理函数。

func MyMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 在请求处理之前执行自定义逻辑
        log.Println("MyMiddleware before")

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

        // 在请求处理之后执行自定义逻辑
        log.Println("MyMiddleware after")
    })
}

在这个示例中，MyMiddleware 函数将原生的 http.Handler 封装在闭包中，并在请求处理前后记录日志信息。

实战案例

考虑一个需要验证用户身份的中间件。可以使用以下代码实现：

func AuthMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 从请求头中获取授权信息
        token := r.Header.Get("Authorization")

        // 检查 token 有效性
        if !validateToken(token) {
            // 返回 401 未授权错误
            http.Error(w, "Unauthorized", http.StatusUnauthorized)
            return
        }

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

AuthMiddleware 将检查请求头中的授权信息，如果 token 无效，则返回 401 状态码。

注册中间件

在 Golang 框架中注册自定义中间件时，需要使用 Use 方法：

func main() {
    router := http.NewServeMux()

    // 注册 MyMiddleware
    router.Use(MyMiddleware)

    // 注册 AuthMiddleware
    router.Use(AuthMiddleware)

    // 添加路由处理程序
    http.ListenAndServe(":8080", router)
}

通过注册中间件，它们将在每个请求处理过程中执行。