go gin(2)中间件

一、中间件

所谓的中间件，是一个请求到达服务器具体服务前或者服务处理后返回前，加了一层处理逻辑。当然，这层逻辑如果在每个处理逻辑前加上这段重复的代码肯定是不好的，所以将这段通用的代码抽离出来，以插件化的形式进行对接，这种方式既保证了系统功能的完整性，又有效的将具体的业务逻辑与系统的功能解耦，并且可以达到灵活配置的目的。

我们实际应用当中比如，在客户端请求到达时，我们先要对用户是否登录、用户权限进行验证，或者请求处理后，将此次请求写入日志。所以中间件有前置操作和后置操作。

中间件的具体角色位置如下图：

 二、gin中间件

1.gin在启动Engine后，它就包含了中间件的方法，看如下源码：

engine.User(Logger(),Recovery())  

2.实际的开发应用中，我们根据不同的业务有不同的中间件操作。具体有分如下四种：

1）、全局中间件，所有的请求都需要用到的中间件

2）、局部中间件，有些请求需要用到中间件，有些不需要要，比如：有的请求需要登录，有些不需要

3）、前置中间件，有些操作需要在请求前执行，比如（登录验证）

4）、后置中间件，有些操作需要在处理完功能后执行，比如（记录系统日志----操作是否成功）

gin中间件自定义，它其实就是一个函数，返回一个gin.Handlefunc方法，具体示例如下

package main

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

func main() {
	engine := gin.Default()
	engine.Use(RequestInfos()) //全局中间件
	engine.POST("/post", func(context *gin.Context) {
		context.Writer.WriteString("返回post请求（中间件示例）")
	})
	engine.GET("/get", func(context *gin.Context) {
		context.Writer.WriteString("返回get请求（中间件示例）")
	})

	engine.Run(":8000")
}

//RequestInfos 请求数据打印的中间件方法
func RequestInfos() gin.HandlerFunc {
	return func(context *gin.Context) {
		path := context.FullPath()
		method := context.Request.Method
		fmt.Println("请求path：",string(path))
		fmt.Println("请求method：",method)

		context.Next()  //此处调用后，将先执行功能逻辑

		fmt.Println("状态码：",context.Writer.Status()) //返回状态码（功能处理后再执行）
	}
}

执行结果：

 

要注意：1.engine.Use(RequestInfos()) 全局中间件用Use方法参数放入中间件函数作为参数，所有的handler方法都会用到

2.在中间件函数中，通过调用context.Next() 先去执行 handler（主功能），执行主功能后，在执行中间件后面未执行完的中间件逻辑

那么我们还有些中间件需要局部调用的呢，代码如下 

package main

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

func main() {
	engine := gin.Default()
	engine.Use(RequestInfos()) //全局中间件
	engine.POST("/post", func(context *gin.Context) {
		context.Writer.WriteString("返回post请求（中间件示例）")
	})
	engine.GET("/get",AnyRequest(), func(context *gin.Context) {
		context.Writer.WriteString("返回get请求（中间件示例）")
	})
	engine.GET("/get2", func(context *gin.Context) {
		context.Writer.WriteString("返回get请求222（中间件示例）")
	})

	engine.Run(":8000")
}

//AnyRequest 局部中间件函数
func AnyRequest() gin.HandlerFunc {
	return func(context *gin.Context) {
		context.Writer.WriteString("调用了局部中间件")
	}
}

//RequestInfos 请求数据打印的中间件方法
func RequestInfos() gin.HandlerFunc {
	return func(context *gin.Context) {
		path := context.FullPath()
		method := context.Request.Method
		fmt.Println("请求path：",string(path))
		fmt.Println("请求method：",method)

		context.Next()  //此处调用后，将先执行功能逻辑

		fmt.Println("状态码：",context.Writer.Status())
	}
}

注意：定义了一个局部的中间件函数，有两个Handler get方法，一个用了局部中间件，一个没有用，使用局部中间件，就是在Handler参数列表中加入中间件函数

 同时我们可以去查看 源码

engine.POST() 、engine.GET() 、engine.Handle() 他们最后的一个参数都是可变参数，都是HandlerFunc。那么刚好也印证了，中间件就是一个HandlerFunc