您现在的位置是:首页 >其他 >Golang Gin 局部、全局 中间件使用网站首页其他

Golang Gin 局部、全局 中间件使用

富士康质检员张全蛋 2023-06-15 20:00:02
简介Golang Gin 局部、全局 中间件使用

中间件


中间件是放在客户端和服务端的中间。 

当你的客户端对某个接口发起一个请求,但是在到达接口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"

某些中间件需要特殊处理的话,使用局部中间件就好了。

如果中间件有很多公用的方法,那么大部分情况就是全局的中间件,而不是局部的中间件。其实大部分使用场景都是使用全局中间件。

风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。