深入理解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框架中间件内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Golang多线程排序实现快速高效地处理大规模数据

    Golang多线程排序实现快速高效地处理大规模数据

    Golang多线程排序是一种快速高效地处理大规模数据的方法,通过使用Golang的协程和通道,可以将排序任务分配到多个线程中并行处理,提高了排序的效率和速度,需要详细了解可以参考下文
    2023-05-05
  • Go中如何使用set的方法示例

    Go中如何使用set的方法示例

    这篇文章主要介绍了Go中如何使用set的方法示例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-09-09
  • 详解Go多协程并发环境下的错误处理

    详解Go多协程并发环境下的错误处理

    这篇文章主要介绍了详解Go多协程并发环境下的错误处理,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-08-08
  • 详解Go 并发

    详解Go 并发

    这篇文章主要介绍了Go 并发的相关资料,帮助大家更好的理解和学习go语言,感兴趣的朋友可以了解下
    2020-09-09
  • Go语言正则表达式示例

    Go语言正则表达式示例

    这篇文章主要介绍了Go语言正则表达式,结合实例形式分析了Go语言正则表达式实现字符串的匹配、查找等相关操作技巧,需要的朋友可以参考下
    2017-01-01
  • golang包快速生成base64验证码的方法

    golang包快速生成base64验证码的方法

    这篇文章主要介绍了golang包快速生成base64验证码的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-03-03
  • golang中"var"与":="的区别解析

    golang中"var"与":="的区别解析

    这篇文章主要介绍了golang中“var”与“:=”的区别,使用var关键字是Go最基本的定义变量方式,有时也会使用到:=来定义变量,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2023-04-04
  • Go语言集成开发环境IDE详细安装教程

    Go语言集成开发环境IDE详细安装教程

    VSCode是免费开源的现代化轻量级代码编辑器,支持几乎所有主流的开发语言,内置命令行工具和 Git 版本控制系统,支持插件扩展,这篇文章主要介绍了Go语言集成开发环境IDE详细安装教程,需要的朋友可以参考下
    2021-11-11
  • Gotify搭建你的消息推送系统

    Gotify搭建你的消息推送系统

    这篇文章主要介绍了Gotify搭建你的消息推送系统,今天要分享的是 gotify,是一个用 go 编写的消息服务端,有需要的朋友可以借鉴参考下,希望能够有所帮助
    2024-01-01
  • gin 获取post请求的json body操作

    gin 获取post请求的json body操作

    这篇文章主要介绍了gin 获取post请求的json body操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2021-03-03

最新评论