gin Engine源码分析

gin是什么？

Gin 是一个用 Go (Golang) 编写的 HTTP Web 框架。 它具有类似 Martini 的 API，但性能比 Martini 快 40 倍。如果你需要极好的性能，使用 Gin 吧。

gin 为什么快？

我们先来看一下官方的基准测试 地址

这只是其中的一部分，完整的可以自行进入上述地址进行查看。

我们通过性能测试看得出来 gin 在处理请求的性能上确实比其他的web 框架快上不少。

性能问题主要是使用 httprouter，而httprouter使用 Trie Tree（字典树）作为路径存储结构，具有高性能的查找。

gin.Engine 源码

// Engine is the framework's instance, it contains the muxer, middleware and configuration settings.
// Create an instance of Engine, by using New() or Default()
type Engine struct {
	  RouterGroup       // 用于注册根路由组使用
    ...
    pool             sync.Pool    // 用于 gin.context 对象池
    trees            methodTrees  // 用于保存的请求路径
    ...
}

在 gin 的Engine中，最主要的就是这3个字段。

gin.RouterGroup 源码

// HandlerFunc defines the handler used by gin middleware as return value.
type HandlerFunc func(*Context)

// HandlersChain defines a HandlerFunc array.
type HandlersChain []HandlerFunc

// RouterGroup is used internally to configure router, a RouterGroup is associated with
// a prefix and an array of handlers (middleware).
type RouterGroup struct {
    Handlers HandlersChain
    basePath string
    engine   *Engine
    root     bool
}

在RouterGroup中Handlers存放的是当前分组的所有中间件，也可以说：请求到这个分组，那么Handlers 中的函数都会根据索引从小到大的按顺序执行。

basePath 就是存放这个分组的基础路由路径。可能直接解释过于抽象；我们来分析它的调用链来看：

1. 
2. 
3. 

在gin.New() 函数中我们可以看到 basePath 默认是 /

而新的RouterGroup是通过Engine调用Group函数产生的；也就是我们自己的Group。

那么以后在这个Group注册的路径都会自动加上/user的路径。

为什么会自动添加路径，后面还会分析相关细节。

gin.methodTree 源码

type node struct {
	path      string
	indices   string
	wildChild bool
	nType     nodeType
	priority  uint32
	children  []*node // child nodes, at most 1 :param style node at the end of the array
	handlers  HandlersChain
	fullPath  string
}

type methodTree struct {
    method string
    root   *node
}

type methodTrees []methodTree

对于node的添加和查找我就不展开讲了，如果不懂的可以自行搜索字典树和查看 httprouter 的实现。

我们刚刚在gin.New 函数中可以看到，trees是一个methodTree的切片，并在在初始化给了默认容量为9，是容量不是长度哦。

在addRoute 函数中可以看到trees的添加：

如果开始没有对应方法的根路径node对象，就会创建一个出来，然后再去添加对应的路由注册。

整体的流程分析

在gin框架中，主要可以使用下面几种注册方式：

• 对应的http请求方法的名称（get,post,put...）
• 通过Group分组路由减少路径的填写
• 通过Any 注册所有http方法路由

这些注册在下面其是都是调用的同一个方法，我们接下来就要来翻开她的外衣看看里面是什么样的。

可以看到都是调用的group.handle 方法

我们来看一下这个方法都做了哪些事情：

• 计算路由的路径group.calculateAbsolutePath(relativePath)
• 为当前注册的路径增加中间件函数，并将注册的函数放在最后面
• 添加到全局路由表中
• 返回一个IRoutes接口

combineHandlers 支持最大的函数数量为63-1个

Group 函数

现在来看Group的实现，通过调用这个方法会产生一个新的RouterGroup 并根据参数设置的基础路径，和全局engine实体，并把基础的中间件也复制到其中。

使用分组路由最后也是调用group.handle方法进行注册，只是其中的basePath在分组的时候已经设置好了，加上注册函数时的路径，就是这个请求的路由路径。

需要注意的一点，我之前想过使用·Group·后，它后面的注册没有使用返回值，注册的路由是怎么注册到全局路由表中的，看过源码才明白。

添加路由的时候，是获取的全局engine 实体，所以也是添加到全局路由表中的。

最后需要注意的

在使用注册中间件和注册路由的时候，需要注意他们注册的顺序。

上一点错误的注册方式代码：

package main

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

func main() {
	eng := gin.New()
	eng.POST("login", func(context *gin.Context) {
		// 处理登录信息
	})
	eng.Use(gin.Logger())

	userGroup := eng.Group("/user")
	userGroup.GET("info", func(context *gin.Context) {
		// 参数验证

		// 处理逻辑
	})

	adminGroup := eng.Group("/admin")
	adminGroup.GET("info", func(context *gin.Context) {
		// 参数验证

		// 处理逻辑
	})

	eng.Use(gin.Recovery())
	adminGroup.PUT("info", func(context *gin.Context) {
		// 参数验证

		// 处理逻辑
	})
	eng.GET("logout", func(context *gin.Context) {
		// 处理登录信息
	})
	eng.Run(":8080")
}

运行结果：

可以看到在注册路由之后，再注册中间件，那么前面注册过的路由是没有这个中间件的。

关注订阅号：

GolangNewbie GO菜鸟

学习更多！