如何为golang 框架创建自定义中间件-Golang-PHP中文网

如何为golang 框架创建自定义中间件

WBOY

发布： 2024-06-12 08:18:01

原创

872人浏览过

在 go 框架中创建自定义中间件涉及实现 http.handler 接口的 servehttp 方法，该方法处理请求并传递给下一个处理程序或中间件。常见的中间件用例包括验证、日志记录和身份验证。您可以通过包装自定义中间件和实际处理函数的处理函数来应用中间件，并将其注册到路由。

如何为golang 框架创建自定义中间件

为 Go 框架创建自定义中间件

简介

中间件是一种强大且可扩展的技术，可用于在 Go 框架中增强请求处理流程。它们允许您在请求到达实际处理程序之前或之后采取特定操作，例如验证、日志记录和身份验证。

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

创建自定义中间件

自定义中间件实现 http.Handler 接口，该接口定义了一个 ServeHTTP 方法。ServeHTTP 方法接收一个 http.ResponseWriter 和 http.Request，并负责处理请求并将控制权传递给下一个处理程序或中间件。

以下是一个创建自定义中间件的简单示例：

import (
    "log"
    "net/http"
)

// MiddlewareHandler 是自定义中间件的示例
type MiddlewareHandler struct {
    // 下一个处理程序或中间件
    Next http.Handler
}

// ServeHTTP 方法实现了 http.Handler 接口
func (h *MiddlewareHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    // 自定义中间件的业务逻辑
    log.Println("Custom middleware executed")

    // 将控制权传递给下一个处理程序或中间件
    h.Next.ServeHTTP(w, r)
}

登录后复制

应用自定义中间件

要将自定义中间件应用于 Go 框架，可以使用 http.HandlerFunc 创建一个处理函数，该函数包装您的中间件和实际处理程序。

例如，假设我们有一个处理函数 MyHandler，它处理 /my-route 路由：

func MyHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Hello from MyHandler!")
}

登录后复制

我们可以使用自定义中间件包装 MyHandler：

// WrapWithMiddleware 将自定义中间件应用于处理函数
func WrapWithMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        middleware := &MiddlewareHandler{Next: next}
        middleware.ServeHTTP(w, r)
    })
}

登录后复制

然后，我们可以使用包装处理函数注册路由：

http.Handle("/my-route", WrapWithMiddleware(MyHandler))

登录后复制

实战案例

一个常见的中间件用例是身份验证。我们可以创建一个中间件来验证请求中的 JWT 令牌，并在验证失败时返回错误响应。

import (
    "fmt"
    "http"
    "strings"

    "github.com/dgrijalva/jwt-go"
)

// AuthMiddlewareHandler 是身份验证中间件示例
type AuthMiddlewareHandler struct {
    // 下一个处理程序或中间件
    Next http.Handler
    // 秘密密钥用于验证 JWT 令牌
    SecretKey string
}

// ServeHTTP 方法实现了 http.Handler 接口
func (h *AuthMiddlewareHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    // 从授权标头中提取 JWT 令牌
    tokenString := r.Header.Get("Authorization")
    if tokenString == "" {
        http.Error(w, "Missing Authorization header", http.StatusBadRequest)
        return
    }

    // 验证 JWT 令牌
    claims, err := jwt.ParseWithClaims(strings.Split(tokenString, " ")[1], &jwt.StandardClaims{}, func(token *jwt.Token) (interface{}, error) {
        return []byte(h.SecretKey), nil
    })
    if err != nil || claims.Valid != true {
        http.Error(w, "Invalid JWT token", http.StatusUnauthorized)
        return
    }

    // 将控制权传递给下一个处理程序或中间件
    h.Next.ServeHTTP(w, r)
}

登录后复制

我们可以通过将 AuthMiddlewareHandler 应用于需要身份验证的任何路由来使用此中间件：

// WrapWithAuthMiddleware 将身份验证中间件应用于处理函数
func WrapWithAuthMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        middleware := &AuthMiddlewareHandler{Next: next, SecretKey: "super_secret_key"}
        middleware.ServeHTTP(w, r)
    })
}

http.Handle("/authenticated-route", WrapWithAuthMiddleware(MyHandler))

登录后复制

以上就是如何为golang 框架创建自定义中间件的详细内容，更多请关注php中文网其它相关文章！