gin 源码探究

1
2
3
4
5
6
7
8
9

// 这里是 gin.Run 的逻辑
func (engine *Engine) Run(addr ...string) (err error) {
	defer func() { debugPrintError(err) }()

	address := resolveAddress(addr)
	debugPrint("Listening and serving HTTP on %s\n", address)
	err = http.ListenAndServe(address, engine)
	return
}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14

func HelloServer(w http.ResponseWriter, req *http.Request) {
	io.WriteString(w, "hello, world!\n")
}

func run() {
    // 点进去就发现，它使用 DefaultServeMux 注册了路由地址
	http.HandleFunc("/hello", HelloServer)
    // 这里直接可以通过函数名知道功能
    // 其中两个主要功能就是
    // 1. 初始化一个 net.Listener 对象，并 Accept 监听
    // 2. 当有新的 socket 链接时，包装一下这个 socket 链接，再启动一个 goroutine 对 connection 进行处理
    // 2.1 对 connection 的处理，主要是解析发来的报文，并封装成 request、response 对象，调用 DefaultServeMux 进行处理
	http.ListenAndServe("", nil)
}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20

type ServeMux struct {
	mu    sync.RWMutex
	m     map[string]muxEntry
	es    []muxEntry // slice of entries sorted from longest to shortest.
	hosts bool       // whether any patterns contain hostnames
}

// ServeHTTP dispatches the request to the handler whose
// pattern most closely matches the request URL.
func (mux *ServeMux) ServeHTTP(w ResponseWriter, r *Request) {
	if r.RequestURI == "*" {
		if r.ProtoAtLeast(1, 1) {
			w.Header().Set("Connection", "close")
		}
		w.WriteHeader(StatusBadRequest)
		return
	}
	h, _ := mux.Handler(r) // 这里匹配了路由，找到了我们注册的函数进行了调用
	h.ServeHTTP(w, r)
}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12

// ServeHTTP conforms to the http.Handler interface.
func (engine *Engine) ServeHTTP(w http.ResponseWriter, req *http.Request) {
	c := engine.pool.Get().(*Context)
	c.writermem.reset(w)
	c.Request = req
	c.reset()

    // 匹配路由，调用 handler 函数链进行处理
	engine.handleHTTPRequest(c)

	engine.pool.Put(c)
}