深入理解Go Gin框架中间件的实现原理

 更新时间:2023年09月25日 09:41:12   作者:水洗牛仔裤  
在Go Gin框架中,中间件是一种在请求处理过程中插入的功能模块,它可以用于处理请求的前置和后置逻辑,例如认证、日志记录、错误处理等,本文将给大家介绍一下Go Gin框架中间件的实现原理,需要的朋友可以参考下

中间件的实现原理

中间件的实现有:链式调用, 拦截器,注册中心等方式,Gin框架使用的是 链式调用洋葱模型 。

链式调用

链式调用,顾名思义就是像一个链子一样连在一起,一个一个的别调用。

Gin框架中的是一个切片:

// HandlerFunc将gin中间件使用的处理程序定义为返回值。
type HandlerFunc func(*Context)
//HandlersChain定义HandlerFunc切片。
type HandlersChain []HandlerFunc

一个路由需要经历的所有中间件函数以及最终的处理函数都会被按照添加的顺序添加到函数链中并储存在路由树的节点上。

怎么调用中间件

在Gin中实现了http包的Handler接口的ServerHTTP方法,(调用流程在Gin启动和接收请求的文章中介绍)。经过ServerHTTP方法,初始化了Context结构体,给其中index这个字段设为-1,调用c.Next时会调用到中间件函数。

ServerHTTP

// ServeHTTP实现http.Handler接口。
func (engine *Engine) ServeHTTP(w http.ResponseWriter, req *http.Request) {
	c := engine.pool.Get().(*Context)
	c.writermem.reset(w)
	c.Request = req
	c.reset()//初始化Context
	engine.handleHTTPRequest(c)//在这里边调用了c.Next(),
	engine.pool.Put(c)
}

c.Next

// Next should be used only inside middleware.
// It executes the pending handlers in the chain inside the calling handler.
// See example in GitHub.
func (c *Context) Next() {
	c.index++
	for c.index < int8(len(c.handlers)) {
		c.handlers[c.index](c)//这个就是函数链,调用第index的函数,参数为c。
		c.index++
	}
}

c.Abort

在中间件中还有一个Abort函数,作用是终止执行。

// Abort阻止调用挂起的处理程序。请注意,这不会停止当前处理程序。
//假设你有一个授权中间件,它验证当前请求是否被授权。
//如果授权失败(例如:密码不匹配),
//则调用Abort以确保不调用此请求的其余处理程序。
const abortIndex int8 = math.MaxInt8 >> 1
func (c *Context) Abort() {
	c.index = abortIndex
}

洋葱模型

就像这个洋葱一样,链式函数是一层一层的调用的。

我们看一下Gin框架官方的两个中间件,他们很好的体现了洋葱模型。

func Default() *Engine {
	debugPrintWARNINGDefault()
	engine := New()
	engine.Use(Logger(), Recovery())
	return engine
}

Logger

func Logger() HandlerFunc {
	return LoggerWithConfig(LoggerConfig{})
}
func LoggerWithConfig(conf LoggerConfig) HandlerFunc {
	//省略初始化日志。
	return func(c *Context) {
	...
		c.Next()
		//省略将日志格式化输出
		}
	}
}

通过前面我们知道,调用c.Next函数的时候会调用下一个函数链中下一个函数,这时候LoggerWithConfig函数还没有执行完,就开始执行下一个函数了,当这个c.Next执行完成后,才会去执行下面的语句了。这个过程是不是就像是一层包一层,类似于洋葱一样。

Recovery

func Recovery() HandlerFunc {
	return RecoveryWithWriter(DefaultErrorWriter)
}
func RecoveryWithWriter(out io.Writer, recovery ...RecoveryFunc) HandlerFunc {
	if len(recovery) > 0 {
		return CustomRecoveryWithWriter(out, recovery[0])
	}
	return CustomRecoveryWithWriter(out, defaultHandleRecovery)
}
func CustomRecoveryWithWriter(out io.Writer, handle RecoveryFunc) HandlerFunc {
	...
	return func(c *Context) {
		defer func() {
			//基本功能是去捕获异常的
		}()
		c.Next()
	}
}

Recovery的做作用是捕获异常的,defer后的函数的作用是去捕获c.Next函数后的异常,防止程序崩溃。具体原理可以去看看defer,painc和recovery的运行机制。

到此这篇关于深入理解Go Gin框架中间件的实现原理的文章就介绍到这了,更多相关Go Gin框架中间件内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Go用两个协程交替打印100以内的奇偶数的方法详解

    Go用两个协程交替打印100以内的奇偶数的方法详解

    这篇文章主要给大家详细介绍了Go用两个协程交替打印100以内的奇偶数的示例代码,文中给大家介绍了两个实现方法,使用无缓冲的channel和设置GOMAXPROCS=1,介绍的非常详细,需要的朋友可以参考下
    2023-08-08
  • 通过手机案例理解Go设计模式之装饰器模式的功能属性

    通过手机案例理解Go设计模式之装饰器模式的功能属性

    这篇文章主要为大家介绍了Go设计模式之装饰器模式的功能属性,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-05-05
  • go语言对文件按照指定块大小进行分割的方法

    go语言对文件按照指定块大小进行分割的方法

    这篇文章主要介绍了go语言对文件按照指定块大小进行分割的方法,实例分析了Go语言文件操作的技巧,需要的朋友可以参考下
    2015-03-03
  • Golang错误处理方式异常与error

    Golang错误处理方式异常与error

    我们在使用Golang时,不可避免会遇到异常情况的处理,与Java、Python等语言不同的是,Go中并没有try...catch...这样的语句块,这个时候我们如何才能更好的处理异常呢?本文来教你正确方法
    2023-01-01
  • Golang中gin框架绑定解析json数据的两种方法

    Golang中gin框架绑定解析json数据的两种方法

    本文介绍 Golang 的 gin 框架接收json数据并解析的2种方法,文中通过代码示例介绍的非常详细,对大家的学习或工作有一定的帮助,需要的朋友可以参考下
    2023-12-12
  • 详解Go语言如何实现字符串切片反转函数

    详解Go语言如何实现字符串切片反转函数

    Go 语言不像其他语言如 Python,有着内置的 reverse() 函数,本文将先学习一下Python中对于列表的反转方法,然后再学习如果在Go语言中实现相同的功能,感兴趣的小伙伴快跟随小编一起来学习一下
    2022-10-10
  • Go库text与template包使用示例详解

    Go库text与template包使用示例详解

    这篇文章主要为大家介绍了Go库text与template包使用示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-12-12
  • Golang实现自己的Redis(有序集合跳表)实例探究

    Golang实现自己的Redis(有序集合跳表)实例探究

    这篇文章主要为大家介绍了Golang实现自己的Redis(有序集合跳表)实例探究,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2024-01-01
  • 详解golang中的闭包与defer

    详解golang中的闭包与defer

    闭包一个函数与其相关的引用环境组合的一个实体,其实可以理解为面向对象中类中的属性与方法,这篇文章主要介绍了golang的闭包与defer,需要的朋友可以参考下
    2022-09-09
  • GO语言对数组切片去重的实现

    GO语言对数组切片去重的实现

    本文主要介绍了GO语言对数组切片去重的实现,主要介绍了几种方法,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2022-04-04

最新评论