gin源码分析

一、高性能
  • 使用sync.pool解决频繁创建的context对象,在百万并发的场景下能大大提供访问性能和减少GC

    go 复制代码
    // ServeHTTP conforms to the http.Handler interface.
    // 每次的http请求都会从sync.pool中获取context,用完之后归还到pool中
    func (engine *Engine) ServeHTTP(w http.ResponseWriter, req *http.Request) {
    	c := engine.pool.Get().(*Context)
    	c.writermem.reset(w) //重置 responsewriter
    	c.Request = req
    	c.reset() //重置使用过的context各个属性
    
    	engine.handleHTTPRequest(c)
    
    	engine.pool.Put(c)
    }
    
    // 这里给pool指定了一个创建新对象的函数,注意不是所有的请求共用一个context,context在高并发场景下可能会分配多个,但是远远小于并发协程数量。 sync.pool.new是可能并发调用的,所以内部的逻辑需要保障线程安全
    func New(opts ...OptionFunc) *Engine {
    	... 
    	engine.RouterGroup.engine = engine
    	engine.pool.New = func() any {
    		return engine.allocateContext(engine.maxParams)
    	}
    	return engine.With(opts...)
    }
    
    //每次http请求都需要分配一个context,这个初始context初始化了两个数组的最大容量
    func (engine *Engine) allocateContext(maxParams uint16) *Context {
    	v := make(Params, 0, maxParams)
    	skippedNodes := make([]skippedNode, 0, engine.maxSections)
    	return &Context{engine: engine, params: &v, skippedNodes: &skippedNodes}
    }
  • 前缀树路由(类似httprouter的路由,提升性能近40倍)

    gin在v1.0版本开始放弃了 github.com/julienschmidt/httprouter,重新实现了一套路由;对比gin新实现的路由基本上采用了httprouter的逻辑,但是和框架结合的更加完整,比如说把httprouter中router的能力提到了engine中。

  • json序列化优化

    gin提供了四种可选的json序列化方式,默认情况下会使用encoding/json

    shell 复制代码
    /github.com/gin-gonic/gin@v1.10.0/internal/json
    	-- go_json.go ("github.com/goccy/go-json")
    	-- json.go  ("encoding/json")
    	-- jsoniter.go ("github.com/json-iterator/go")
    	-- sonic.go ("github.com/bytedance/sonic")

    需要在编译期间指定tag来决定使用哪种序列化工具

    shell 复制代码
    go run -tags={go_json|jsoniter|sonic|(不指定默认encoding)} main.go

    通过一个简单的基准测试看看哪种json序列化效率更高

    go 复制代码
    package gan
    
    import (
    	"encoding/json"
    	sonicjson "github.com/bytedance/sonic"
    	gojson "github.com/goccy/go-json"
    	itjson "github.com/json-iterator/go"
    	"testing"
    )
    
    func BenchmarkJson(b *testing.B) {
    	jsonStr := `{"name":"zhangsan","age":18,"address":"beijing"}`
    	m := &map[string]interface{}{}
    	for i := 0; i < b.N; i++ {
    		json.Unmarshal([]byte(jsonStr), m)
    		json.Marshal(m)
    	}
    }
    
    func BenchmarkGOJson(b *testing.B) {
    	jsonStr := `{"name":"zhangsan","age":18,"address":"beijing"}`
    	m := &map[string]interface{}{}
    	for i := 0; i < b.N; i++ {
    		gojson.Unmarshal([]byte(jsonStr), m)
    		gojson.Marshal(m)
    	}
    }
    
    func BenchmarkItJson(b *testing.B) {
    	m := &map[string]interface{}{}
    	jsonStr := `{"name":"zhangsan","age":18,"address":"beijing"}`
    	for i := 0; i < b.N; i++ {
    		itjson.Unmarshal([]byte(jsonStr), m)
    		itjson.Marshal(m)
    	}
    }
    
    func BenchmarkSonicJson(b *testing.B) {
    	m := &map[string]interface{}{}
    	jsonStr := `{"name":"zhangsan","age":18,"address":"beijing"}`
    	for i := 0; i < b.N; i++ {
    		sonicjson.Unmarshal([]byte(jsonStr), m)
    		sonicjson.Marshal(m)
    	}
    }

    测试结果: sonic > go_json > json_iterator > encoding

    shell 复制代码
    $ go test -bench='Json$' -benchtime=5s -benchmem .
    goos: windows
    goarch: amd64
    pkg: gan
    cpu: Intel(R) Core(TM) i5-10400 CPU @ 2.90GHz
    BenchmarkJson-12                 2632854              2260 ns/op             616 B/op         24 allocs/op
    BenchmarkGOJson-12               5226374              1142 ns/op             248 B/op         11 allocs/op
    BenchmarkItJson-12               4811112              1260 ns/op             400 B/op         19 allocs/op
    BenchmarkSonicJson-12            6616218               913.0 ns/op           333 B/op         10 allocs/op
    PASS
    ok      gan     30.313s
二、基于前缀树的路由设计
  • go语言中原生net/http包在负载路由下的缺陷

    动态路由:缺少例如hello/:name,hello/*这类的规则。

    鉴权:没有分组/统一鉴权的能力,需要在每个路由映射的handler中实现。

  • http请求怎么进入的gin的处理逻辑中去的?

    1. gin框架中调用了net/http包,监听address,请求都会进入顶层路由,也就是engine结构实现的ServeHTTP函数中

      go 复制代码
      func (engine *Engine) Run(addr ...string) (err error) {
          ......
      	debugPrint("Listening and serving HTTP on %s\n", address)
      	err = http.ListenAndServe(address, engine.Handler())
      	return
      }
      func (engine *Engine) ServeHTTP(w http.ResponseWriter, req *http.Request) {
          ......
      	engine.handleHTTPRequest(c)
          ......
      }
    2. 在engine.handleHTTPRequest© 中,根据request中的path,在路由表中获取路由对应的handler然执行

      go 复制代码
      func (engine *Engine) handleHTTPRequest(c *Context) {
          httpMethod := c.Request.Method
      	rPath := c.Request.URL.Path
          ......
      	// Find root of the tree for the given HTTP method
      	t := engine.trees
      	for i, tl := 0, len(t); i < tl; i++ {
      		if t[i].method != httpMethod {
      			continue
      		}
      		root := t[i].root
      		// Find route in tree
      		value := root.getValue(rPath, c.params, c.skippedNodes, unescape)
      		if value.params != nil {
      			c.Params = *value.params
      		}
      		if value.handlers != nil {
      			c.handlers = value.handlers
      			c.fullPath = value.fullPath
      			c.Next() //最终执行的handler
      			c.writermem.WriteHeaderNow()
      			return
      		}
      		......
      	}
      }
  • 路由的注册和匹配

    1. Engine继承了RouterGroup的能力,当我们注册handler时,或者在分组路由下去注册handler时,其实都是调用的RouterGroup的能力

      go 复制代码
      type Engine struct {
         RouterGroup
         ......
      }
    2. RouterGroup中保留了一个Engine指针,最终的路由是注册到了Engine中的路由表中,这个路由表没有设计在RouterGroup中,保证了RouterGroup的职责单一性,也就是只做路由分组。

      go 复制代码
      func (group *RouterGroup) handle(httpMethod, relativePath string, handlers HandlersChain) IRoutes {
          ......
      	group.engine.addRoute(httpMethod, absolutePath, handlers)
      	return group.returnObj()
      }
    3. Engine中保存了一个methodTree对象,这个就是路由表,是一个树状结构

      go 复制代码
      type methodTree struct {
      	method string
      	root   *node
      }
      
      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 methodTrees []methodTree
三、中间件执行流程
  • 中间件执行链路

    1. 中间件保存在GroupRouter中

      go 复制代码
      func (group *RouterGroup) Use(middleware ...HandlerFunc) IRoutes {
         group.Handlers = append(group.Handlers, middleware...)
         return group.returnObj()
      }
    2. 每声明一次分组,都会将handler进行一次合并

      go 复制代码
      func (group *RouterGroup) Group(relativePath string, handlers ...HandlerFunc) *RouterGroup {
      	return &RouterGroup{
      		Handlers: group.combineHandlers(handlers), //合并中间处理链路
      		basePath: group.calculateAbsolutePath(relativePath),
      		engine:   group.engine,
      	}
      }
    3. 以上只是将中间件合并成了一条链路,最终的业务逻辑会在执行GET.POST等请求方法时,拼接到执行链路末端

      go 复制代码
      func (group *RouterGroup) handle(httpMethod, relativePath string, handlers HandlersChain) IRoutes {
      	absolutePath := group.calculateAbsolutePath(relativePath)
      	handlers = group.combineHandlers(handlers) //这里参数的handlers是业务逻辑,合并到链路末尾。
      	group.engine.addRoute(httpMethod, absolutePath, handlers)
      	return group.returnObj()
      }
    4. 最终效果

  • 前置处理和后置处理

    1. 假设需要对一段业务逻辑采集它的执行耗时,一般我们需要在执行逻辑前声明一个时间点,执行完后再用当前时间减去执行前的时间得到耗时。

    2. 在gin中的实现方式是通过context的next方法去实现的,结合上面的流程,在中间件中每声明一次next,执行链会先去执行下一个handler,最终的效果应该是ABBA方式的执行。

      go 复制代码
      func (c *Context) Next() {
         c.index++
         for c.index < int8(len(c.handlers)) {
            c.handlers[c.index](c)
            c.index++
         }
      }
其他:

Gin的核心特性

  • 高性能:Gin的HTTP请求处理速度极快,能够支持大量的并发连接(gin并不是基准测试中最快的框架,但是相对较快,简单好用,用户体量大)
  • 简单易用:Gin的API设计直观,使得开发者可以快速上手并构建应用。
  • 中间件支持:Gin允许开发者通过中间件来扩展其功能,这为构建复杂的Web应用提供了可能。
  • 路由和参数绑定:Gin提供了强大的路由功能,支持参数绑定,使得URL处理更加灵活。
  • 数据渲染:Gin支持多种数据渲染方式,包括JSON、XML、HTML等,方便开发者根据需求选择合适的输出格式。

参考:https://geektutu.com/post/gee-day4.html

相关推荐
007php00710 小时前
GoZero 上传文件File到阿里云 OSS 报错及优化方案
服务器·开发语言·数据库·python·阿里云·架构·golang
高 朗11 小时前
【GO基础学习】基础语法(2)切片slice
开发语言·学习·golang·slice
IT书架12 小时前
golang面试题
开发语言·后端·golang
醒过来摸鱼18 小时前
【Golang】协程
开发语言·后端·golang
灼华十一20 小时前
算法编程题-排序
数据结构·算法·golang·排序算法
宋发元20 小时前
Go语言使用 kafka-go 消费 Kafka 消息教程
golang·kafka·linq
宋发元21 小时前
Go消费kafka中kafkaReader.FetchMessage(ctx)和kafkaReader.ReadMessage(ctx)的区别
golang·kafka·linq
祁许1 天前
【Golang】手搓DES加密
开发语言·golang·密码学
凡人的AI工具箱1 天前
15分钟学 Go 实战项目六 :统计分析工具项目(30000字完整例子)
开发语言·数据库·人工智能·后端·golang
王大锤43911 天前
golang通用后台管理系统10(退出登录,注销token)
golang