Golang Gin局部和全局中间件使用详解
作者:富士康质检员张全蛋 时间:2023-07-10 03:03:00
中间件
中间件是放在客户端和服务端的中间。
当你的客户端对某个接口发起一个请求,但是在到达接口2之前,这里是有一层中间件的处理。
一般常用的就是处理跨域,比如处理登入的验证token,接口返回的信息比较敏感,说白了就是需要做鉴权。其实也就是防止别人刷你接口,拿到你的数据。
也就是你必须携带正确的token,让中间件去做验证,做完验证之后再去处理这个接口的逻辑。
Gin框架允许开发者在处理请求的过程中,加入用户自己的钩子(Hook)函数(钩子函数其实就是中间件)。
这个钩子函数就叫中间件,中间件适合处理一些公共的业务逻辑 ,比如 登录认证 、 权限校验 、数据分页、记录日志、耗时统计等(比如微服务架构,你要去统计每个服务的metrics,比如请求的耗时,请求的状态码,也可以通过这个中间件去实现)
1. 全局中间件
全局中间件对所有的路由都生效。
如果先注册了路由器,我再去use这个中间件,发现这个中间件没有生效,这个是正常的,因为中间件是在路由之后,那么注册中间件就是失效的, 中间件一定要在路由之前注册。
HandlerFunc这个其实也就是接受上下文对匿名函数。
type HandlerFunc func(*Context)
这里很像一个闭包的用法,我们只能在函数当中定义,但是没有办法去改变其一些变量。它不想让你去改变其一些配置,它只允许你写部分对实现。
import (
"fmt"
"github.com/gin-gonic/gin"
"net/http"
)
//全局中间件,对所有路由生效,必须在路由注册之前去use
func MiddleWare() gin.HandlerFunc {
return func(c *gin.Context) { //返回一个匿名函数
fmt.Println("我是一个中间件")
}
}
func main() {
r := gin.Default()
r.Use(MiddleWare())
r.GET("/hello", func(c *gin.Context) {
fmt.Println("执行了get方法")
c.String(http.StatusOK, "hello")
c.JSON(http.StatusOK, gin.H{
"msg": "success",
"data": "hello",
})
})
r.Run(":8000")
}
[GIN-debug] Listening and serving HTTP on :8000
我是一个中间件
执行了get方法
[GIN] 2023/04/03 - 10:58:24 | 200 | 44.667µs | 127.0.0.1 | GET "/hello"
这里可以看到中间件是先执行,中间件比路由的handler是先执行的。
2. 局部中间件
局部中间件就是对某个路由生效,这个就没有顺序了,因为它是跟着注册路由一起走的。
import (
"fmt"
"github.com/gin-gonic/gin"
)
func MiddleWare() gin.HandlerFunc {
return func(c *gin.Context) {
fmt.Println("这里可以做一些身份验证等")
}
}
func main() {
r := gin.Default()
// 首页无需验证
r.GET("/index", func(c *gin.Context) {
c.JSON(200, gin.H{"msg": "index 页面"})
})
// home页需要用户登录才能查看
r.GET("/home", MiddleWare(), func(c *gin.Context) {
c.JSON(200, gin.H{"msg": "home 页面"})
})
r.Run() }
可以看到在返回日志之前去执行了这个中间件,这就是一个局部中间件对用法。(先去执行中间件再去执行handler,这个局部中间件只对上面的路由生效)
[GIN-debug] Listening and serving HTTP on :8000
[GIN] 2023/04/03 - 11:14:11 | 200 | 56.791µs | 127.0.0.1 | GET "/index"
这里可以做一些身份验证
[GIN] 2023/04/03 - 11:14:17 | 200 | 31.291µs | 127.0.0.1 | GET "/home"
某些中间件需要特殊处理的话,使用局部中间件就好了。
如果中间件有很多公用的方法,那么大部分情况就是全局的中间件,而不是局部的中间件。其实大部分使用场景都是使用全局中间件。
来源:https://blog.csdn.net/qq_34556414/article/details/129922392