在 golang 微服务中，中间件提供了通用功能。可以使用 gin、echo 或手动创建中间件。最常见的中间件包括身份验证、授权和日志记录。通过 gin 或 echo 的内置机制或手动创建中间件，可以轻松插入自定义逻辑，例如验证访问令牌，确保只有授权用户可以访问受保护的端点。

利用 Golang 框架构建微服务的中间件策略

在微服务架构中，中间件扮演着至关重要的角色，为服务组件提供通用功能，例如身份验证、授权、监控和日志记录。Golang 提供了强大的框架，可以轻松地创建和部署用于微服务的中间件。

1. 基于 Gin 的中间件

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

Gin 是一个流行的 Golang Web 框架，它提供了一种简单的方法来创建基于 HTTP 的微服务。其内置了中间件机制，可以在请求处理过程中插入自定义逻辑。

func main() {
    r := gin.New()

    // 添加中间件
    r.Use(gin.Logger())
    r.Use(gin.Recovery())

    r.GET("/hello", func(c *gin.Context) {
        c.JSON(200, gin.H{"message": "Hello, world!"})
    })

    r.Run()
}

2. 基于 Echo 的中间件

Echo 是另一个受广泛使用的 Golang Web 框架，它也提供了中间件支持。与 Gin 类似，Echo 允许您在请求处理管道中插入自定义处理器。

func main() {
    e := echo.New()

    // 添加中间件
    e.Use(middleware.Logger())
    e.Use(middleware.Recover())

    e.GET("/hello", func(c echo.Context) error {
        return c.String(http.StatusOK, "Hello, world!")
    })

    e.Start(":3000")
}

3. 手动创建中间件

如果您希望拥有对中间件实现的完全控制，您可以手动创建自己的中间件。这可以在编写自定义请求处理逻辑的情况下很有用。

type MyMiddleware struct {
    next http.Handler
}

func (m *MyMiddleware) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    // 自定义逻辑
    m.next.ServeHTTP(w, r)
}

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

    // 添加自定义中间件
    mux.Handle("/", &MyMiddleware{next: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        fmt.Fprintln(w, "Hello, world!")
    })})

    http.ListenAndServe(":3000", mux)
}

实战案例：身份验证中间件

假设您有一个用于管理用户的微服务。您可以使用中间件来验证每个传入请求的访问令牌。

func AuthMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 从请求获取访问令牌
        token := r.Header.Get("Authorization")

        // 验证令牌
        claims, err := VerifyToken(token)
        if err != nil {
            http.Error(w, "Invalid token", http.StatusUnauthorized)
            return
        }

        // 将已验证的声明添加到请求上下文中
        ctx := context.WithValue(r.Context(), "user_claims", claims)

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

使用此中间件，您可以确保只有拥有有效访问令牌的用户才能访问受保护的端点。