深入理解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五种原子性操作的用法,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2021-09-09
  • golang API请求队列的实现

    golang API请求队列的实现

    本文主要介绍了golang API请求队列的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2022-04-04
  • 详解Go中Set的实现方式

    详解Go中Set的实现方式

    这篇文章主要介绍了详解Go中Set的实现方式,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-09-09
  • Golang使用gorm实现分页功能的示例代码

    Golang使用gorm实现分页功能的示例代码

    在提供列表接口时一般要用到分页,对于存储在某些数据库中的数据进行分页起来非常的方便,下文给出一个通过gorm进行分页并通过http返回数据的例子,感兴趣的小伙帮跟着小编一起来看看吧
    2024-10-10
  • Golang interface{}的具体使用

    Golang interface{}的具体使用

    interface{}是Go中可以表示任意类型的空接口,本文主要介绍了Golang interface{}的具体使用,具有一定的参考价值,感兴趣的可以了解一下
    2025-05-05
  • Windows下Goland的环境搭建过程详解

    Windows下Goland的环境搭建过程详解

    这篇文章主要介绍了Windows下Goland的环境搭建过程,本文通过图文并茂的形式给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-10-10
  • 分享6个Go处理字符串的技巧小结

    分享6个Go处理字符串的技巧小结

    这篇文章主要介绍了分享6个Go处理字符串的技巧小结,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-12-12
  • Gin框架整合Swagger生成接口文档完整指南

    Gin框架整合Swagger生成接口文档完整指南

    Swagger是一个基于OpenAPI规范的API文档生成工具,它能够自动生成、描述、调试和可视化RESTful API文档,下面小编就为大家详细讲讲Gin框架整合Swagger生成接口文档的完整方法吧
    2025-09-09
  • golang为什么要统一错误处理

    golang为什么要统一错误处理

    这篇文章主要介绍了golang为什么要统一错误处理,统一错误处理的目的是为了前端开发接收到后端的statuscode,之后便于前端逻辑上开发以及开发,下文具体操作过程需要的小伙伴可以参考一下
    2022-04-04
  • golang croncli 定时器命令详解

    golang croncli 定时器命令详解

    定时器是执行任务时的常用功能,配置系统的定时任务太麻烦,所以就想用golang简单实现一个定时器命令,包括定时器命令格式、定时执行命令的相关知识,感兴趣的朋友跟随小编一起看看吧
    2022-03-03

最新评论