Gin 框架中的中间件和路由分组

中间件的使用

在使用 Gin 框架进行 Web 开发时，中间件是一个非常重要的概念。中间件可以在请求处理的前后执行特定的逻辑，例如日志记录、身份验证等。以下是如何在 Gin 中使用中间件的示例：

package main

import (
    "fmt"
    "github.com/gin-gonic/gin"
    "net/http"
)

// GlobalMiddleware 全局中间件
func GlobalMiddleware() gin.HandlerFunc {
    return func(c *gin.Context) {
        fmt.Println("全局中间件：处理每个请求")
        c.Next()
    }
}

// CORSMiddleware 跨域中间件
func CORSMiddleware() gin.HandlerFunc {
    return func(c *gin.Context) {
        c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
        c.Writer.Header().Set("Access-Control-Allow-Credentials", "true")
        c.Writer.Header().Set("Access-Control-Allow-Headers", "Content-Type, Content-Length, Accept-Encoding, Authorization, Origin, Cache-Control, X-Requested-With")
        c.Writer.Header().Set("Access-Control-Allow-Methods", "POST, OPTIONS, GET, PUT, DELETE")

        if c.Request.Method == "OPTIONS" {
            c.AbortWithStatus(http.StatusNoContent)
            return
        }

        c.Next()
    }
}

// LocalMiddleware 局部中间件
func LocalMiddleware() gin.HandlerFunc {
    return func(c *gin.Context) {
        fmt.Println("局部中间件：仅在特定路由触发")
        c.Next()
    }
}

func main() {
    router := gin.Default()

    // 应用全局中间件
    router.Use(GlobalMiddleware())

    // 应用跨域中间件
    router.Use(CORSMiddleware())

    // 无中间件的路由
    router.GET("/", func(c *gin.Context) {
        c.JSON(http.StatusOK, gin.H{"message": "无中间件"})
    })

    // 应用局部中间件的路由
    router.GET("/local", LocalMiddleware(), func(c *gin.Context) {
        c.JSON(http.StatusOK, gin.H{"message": "局部中间件"})
    })

    // 启动服务器
    router.Run(":8080")
}

在上述代码中：

GlobalMiddleware 是全局中间件，会在每个请求处理前执行。
CORSMiddleware 处理跨域请求，设置了相关的响应头。
LocalMiddleware 是局部中间件，仅在特定路由中使用。

通过 router.Use() 方法，可以将中间件应用于全局。😊

路由分组

Gin 支持对路由进行分组，方便管理和维护。以下是一个路由分组的示例：

package main

import (
    "github.com/gin-gonic/gin"
    "net/http"
)

func main() {
    router := gin.Default()

    // 创建用户相关的路由分组
    userGroup := router.Group("/user")
    {
        userGroup.POST("/", addUser)         // 添加用户
        userGroup.DELETE("/:id", deleteUser) // 删除用户
        userGroup.PUT("/:id", updateUser)    // 更新用户
    }

    // 启动服务器
    router.Run(":8080")
}

// 添加用户的处理函数
func addUser(c *gin.Context) {
    c.JSON(http.StatusOK, gin.H{"message": "用户添加成功"})
}

// 删除用户的处理函数
func deleteUser(c *gin.Context) {
    userID := c.Param("id")
    c.JSON(http.StatusOK, gin.H{"message": "成功删除用户", "userID": userID})
}

// 更新用户的处理函数
func updateUser(c *gin.Context) {
    userID := c.Param("id")
    c.JSON(http.StatusOK, gin.H{"message": "用户更新成功", "userID": userID})
}

在这个示例中，我创建了一个以 /user 为前缀的路由组 userGroup，将用户相关的操作都放在这个组内，便于管理。

注意事项

中间件的执行顺序与注册顺序一致，需注意全局中间件和局部中间件的先后关系。
在处理跨域请求时，需正确设置响应头，特别是 Access-Control-Allow-Origin 和 Access-Control-Allow-Methods。

最佳实践

合理使用中间件，将通用功能抽象出来，提升代码复用性。
对功能相似的路由进行分组，清晰明了。
变量名应易于理解，提升代码可读性。

中间件的使用​

路由分组​

注意事项​

最佳实践​

中间件的使用

路由分组

注意事项

最佳实践