GO框架学习(1)-gin源码阅读

写在开头

非原创,网络收集,用于自己知识整理

1.IRoutes接口

官网gin的demo

    r := gin.Default()
    r.GET("/ping", func(ctx *gin.Context) {
        ctx.JSON(200, gin.H{
            "message": "pong",
        })
    })

使用gin的Default和New方法创建的实例是Engine,它也是框架的核心实例,包括了中间件、配置和路由等功能
vscode下ctrl+f12发现Engine是IRoutes接口的实现

type IRoutes interface {
    Use(...HandlerFunc) IRoutes

    Handle(string, string, ...HandlerFunc) IRoutes
    Any(string, ...HandlerFunc) IRoutes
    GET(string, ...HandlerFunc) IRoutes
    POST(string, ...HandlerFunc) IRoutes
    DELETE(string, ...HandlerFunc) IRoutes
    PATCH(string, ...HandlerFunc) IRoutes
    PUT(string, ...HandlerFunc) IRoutes
    OPTIONS(string, ...HandlerFunc) IRoutes
    HEAD(string, ...HandlerFunc) IRoutes
    Match([]string, string, ...HandlerFunc) IRoutes

    StaticFile(string, string) IRoutes
    StaticFileFS(string, string, http.FileSystem) IRoutes
    Static(string, string) IRoutes
    StaticFS(string, http.FileSystem) IRoutes
}

主要分为3个功能:

  1. Use方法提供了用户接入自定义逻辑的能力,即插件或中间件机制
  2. Handle等方法是提供注册路由的抽象
  3. Static开头的方法是处理静态文件的接口

不同于传统MVC框架(例如PHP的ThinkPHP和GO的Beengo),Gin没有Controller的抽象,也就是说,对于gin作者,他的设计理念是--MVC应该是用户组织web项目的模式,而不是框架设计者要考虑的

2.路由树设计

Engine本身也实现了net/http.Handler接口,即本身就可以作为一个Handler传递到http包中用于启动服务器


image.png

2.1添加路由

Engine->addRoute
方法签名如下

func (engine *Engine) addRoute(method, path string, handlers HandlersChain)

addroute根据要添加的方法获取对应的路由树(会先进行合法性检查)

// Engine.addRoute()
assert1(path[0] == '/', "path must begin with '/'")
root := engine.trees.get(method)

trees是存放methodTree实例的数组,因为请求方法最多有9个,所以使用数组足够

//methodTrees
type methodTree struct {
    method string
    root   *node
}

type methodTrees []methodTree

func (trees methodTrees) get(method string) *node

由此可见,我们每次添加的一个路由就是一个node,这些node根据访问方法GET/POST等被组织在一颗methodTree路由树上,各种不同的路由树组成了一个路由森林methodTrees

type node struct {
    path      string
    indices   string
    wildChild bool
    nType     nodeType
    priority  uint32
    children  []*node
    handlers  HandlersChain
    fullPath  string
}

回到Engine.addRoute中,在找到路由树后,就会根据路径添加对应的路由

//Engine.addRoute
root.addRoute(path, handlers)

这里是调用node.addRoute的逻辑,也是gin的前缀树核心逻辑所在

2.1.1什么是前缀树(Tire)

可以先尝试完成leet这道题
前缀树(Tire)是哈希树的变种,通常叫做单词查找树,能利用字符串的公共前缀减少查询时间,最大限度减少不必要的字符串比较,所以前缀树常被用于搜索引擎系统用于文本词频统计,前缀树拥有以下特点:

  1. 跟节点不包含字符(如上面的root只保存"/"路径),其他节点包含字符
  2. 每一层节点内容不同
  3. 从跟节点到某一个节点,路径上经过的字符连接起来,为该节点对应的字符串
  4. 每个节点的子节点,通常有一个标志位,用来标识单词的结束

以新华词典查找“心想事成”中“心想”两字来模拟tire树查找流程
image.png

在查找的过程中,我们根据首字母 x,找到 x 当中的 xi 这一共同部分,然后再根据不同的字母找到所对应的剩余部分。放到前缀树查找上,案例中的“心”对应 xi -> n,而“想”则对应 xi -> ang

2.1.2紧凑前缀树

GIN 中的前缀树相比普通的前缀树减少了查询的层级,比如说上方我们想查找的“心想”其中 xi 做为共有的部分,其实可以被分配在同一层同一个节点当中而不是分为两部分:
image.png

这样的就是紧凑前缀树,同理如果我们有如下四个路由,他们所形成的紧凑前缀树就会是这个样子:

r.GET("/", handle1)
r.GET("/product", handle2)
r.GET("/product/:id", handle3)
r.GET("/product/:name", handle4)
image.png

在有了这些数据结构知识储备的前提下,我们再回头看node.addRoute代码

//node.addRoute
    fullPath := path
    n.priority++

    // Empty tree
    if len(n.path) == 0 && len(n.children) == 0 {
        n.insertChild(path, fullPath, handlers)
        n.nType = root
        return
    }

node.fullpath存储完整路径,添加完成后,经过此节点的路由条数node.prority会+1(node.prority标识后续节点数量,满足tire特点4),如果该树为空树,即跟节点"/",则执行插入一个子节点逻辑

for {
//node.addRoute
i := longestCommonPrefix(path, n.path)

//longestCommonPrefix逻辑
func longestCommonPrefix(a, b string) int {
    i := 0
    max := min(len(a), len(b))
    for i < max && a[i] == b[i] {
        i++
    }
    return i
}

找到最长公共前缀的长度,path[i]==n.path[i],longestCommonPrefix的逻辑也很简单,依次移动下标对比
假设当前存在前缀信息n.path="hello",现有"he"结点进入,则当前节点需要拆分成n.path="he"和n.path="llo"(需要满足前缀树特点3),拆分逻辑如下

//node.addRoute
       if i < len(n.path) {
            child := node{
                                // 除去公共前缀部分,剩余的内容作为子节点
                path:      n.path[i:],
                wildChild: n.wildChild,
                indices:   n.indices,
                children:  n.children,
                handlers:  n.handlers,
                priority:  n.priority - 1,
                fullPath:  n.fullPath,
            }

            n.children = []*node{&child}
            // []byte for proper unicode char conversion, see #65
            n.indices = bytesconv.BytesToString([]byte{n.path[i]})
            n.path = path[:i]
            n.handlers = nil
            n.wildChild = false
            n.fullPath = fullPath[:parentFullPathIndex+i]
        }

当公共前缀比已有path长度小时,拆分节点
上面"llo"被作为子节点放入n.children中,之后更新自己的n.path=path[:i]即"he"
node.indices是string类型,每个indices字符对应一个孩子节点的path的首字母
再来个例子加深印象
gin中注册路由如下

  r.GET("/ping", func(c *gin.Context) {})
  
  r.GET("/pii", func(c *gin.Context) {})

  r.GET("/pin", func(c *gin.Context) {})

前缀树变化


image.png

当公共前缀长度小于要注册的路由时,说明需要新建子节点(因为tire特点3要求我们路径经过字符连接起来,是对应的路由)

//node.addRoute
        if i < len(path) {
    //提取剩下的字符串
            path = path[i:]
            c := path[0]

            // '/' after param
            if n.nType == param && c == '/' && len(n.children) == 1 {
                parentFullPathIndex += len(n.path)
                n = n.children[0]
                n.priority++
                continue walk
            }

这上面是处理nType==param即参数路由"/:id"的逻辑

//node.addRoute
            for i, max := 0, len(n.indices); i < max; i++ {
                if c == n.indices[i] {
                    parentFullPathIndex += len(n.path)
                    i = n.incrementChildPrio(i)
                    n = n.children[i]
                    continue walk
                }
            }

因为n.indices存储所有子节点的首字母,如果相等,就说明还有公共部分,回到walk处继续处理干净
之后出现的字符再没有公共前缀

            if c != ':' && c != '*' && n.nType != catchAll {
                // []byte for proper unicode char conversion, see #65
                n.indices += bytesconv.BytesToString([]byte{c})
                child := &node{
                    fullPath: fullPath,
                }
                n.addChild(child)
                n.incrementChildPrio(len(n.indices) - 1)
                n = child

上面是处理非通配符节点、非参数匹配节点和非正则匹配节点逻辑:

  1. 新增首字母到node.indices中
  2. 插入新增节点
  3. 处理新增节点

其他类型节点的处理逻辑代码就不再贴出,只需要注意当当前节点是通配符节点时,若再插入通配符的节点,即"/*/* "会发生覆盖
图示i<len(path)静态匹配逻辑

  r.GET("/pi", func(c *gin.Context) {})

  r.GET("/pin", func(c *gin.Context) {})

  r.GET("/pii", func(c *gin.Context) {})
image.png

添加静态路由的主要逻辑阅读完毕,剩余类型的路由主要集中在node.insertChild逻辑中

3.寻找路由

了解上面前缀树的特点3后我们自然就想到了根据n.path去寻找n.fullpath,这里具体代码就不再贴出知道逻辑就行了,主要逻辑代码在node.getValue函数中

参考

1.终于搞懂了gin中的路由树
2.关于gin的路由树

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,816评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,729评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,300评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,780评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,890评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,084评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,151评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,912评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,355评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,666评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,809评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,504评论 4 334
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,150评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,882评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,121评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,628评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,724评论 2 351

推荐阅读更多精彩内容