golang 中间件使用

顿悟树下你和我2024-02-03 8:55

1、自定义中间件函数

Go 复制代码

func CustomMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 在请求之前执行的逻辑
		fmt.Println("执行自定义中间件 - 请求之前")

		// 调用链中的下一个处理程序
		c.Next()

		// 在请求之后执行的逻辑
		fmt.Println("执行自定义中间件 - 请求之后")
	}
}

2、使用中间件

Go 复制代码

func main() {
	// 创建一个Gin引擎
	router := gin.Default()

	// 使用自定义中间件
	router.Use(CustomMiddleware())

	// 定义路由
	router.GET("/hello", func(c *gin.Context) {
		c.JSON(200, gin.H{
			"message": "Hello, Gin Middleware!",
		})
	})

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

在上述示例中，CustomMiddleware函数返回一个gin.HandlerFunc，它是一个用于Gin中间件的类型。在中间件函数中，你可以在处理请求之前和之后执行特定的逻辑。在示例中，我们简单地在控制台打印一些信息。

在main函数中，通过router.Use方法将自定义中间件添加到Gin引擎。中间件的执行顺序与它们添加到引擎的顺序相同。

当访问/hello路由时，中间件会在处理请求之前和之后执行。

请注意，Gin中间件的设计允许你执行一些预处理或后处理逻辑，例如身份验证、日志记录等。