关于go:Go-框架深入理解-gin-框中-Context-的-Request-和-Writer-对象

4次阅读

共计 2669 个字符，预计需要花费 7 分钟才能阅读完成。

大家好，我是渔夫子。

明天跟大家聊一聊 gin 中 Context 对象中的 Request 和 Writer 对象。

在应用 gin 框架时，咱们定义的申请处理器，输出参数总是一个 gin.Context 的指针类型，代表申请的上下文。在处理器的业务逻辑中，通过 Context.Request 能够获取本次申请的参数值；通过 Context.Writer 就能将响应后果输入给客户端了。如下代码所示：

package main

import (
    "github.com/gin-gonic/gin"
    "log"
    "net/http"
)

func main() {r := gin.Default()

    // 注册路由，定义申请处理器函数
    r.GET("/", func(c *gin.Context) {
        c.Param
        c.Writer.Write([]byte("Hello World"))

    })

  // 调用该函数，则禁用日志带色彩输入
    gin.DisableConsoleColor()
  
  // 应用该函数，则强制日志带色彩输入，无论是在终端还是其余输出设备 
    gin.ForceConsoleColor()

    r.Run("127.0.0.1:8080")
}

那么，Context 字段是在什么中央初始化的呢，为什么通过 Context.Request 字段就能读取申请的参数呢，又为什么通过 Context.Writer 字段就能将响应后果输入给客户端呢？接下来咱们就一一解答这 3 个问题。

gin 的 Context 对象实际上是在 Engine 对象的 ServeHTTP 函数中进行初始化 的，如下：

// 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()

    engine.handleHTTPRequest(c)

    engine.pool.Put(c)
}

通过该函数咱们能够看到，在第 3 行通过池化技术获取到一个新的 Context 对象。获取到该 Context 后，通过 engine.handleHTTPRequest 函数传入 Context 参数，进而找到对应的路由处理器，传递给具体的路由处理器。在上例中就是 r.GET 中的对应处理器。

在这个函数中，还会把本次的申请体 req 赋值给 Context.Request 变量。这样，在具体的申请处理器中就能够通过 Context 的各种办法从 Request 中获取参数值了。比方 GetHeader、GetRawData 等。

咱们再来回顾下 go 的 http 的启动和申请解决流程，以便理解 ServeHTTP 的调用机会。engine 对象实际上是实现了 go 的规范包中的 Handler 接口。该接口中定义了 ServeHTTP 办法。在接管到申请后，net/http 的包就会调用 engine 的 ServeHTTP 办法。如下图中的 engine.ServeHTTP 局部：

在上节中，在 Engine 的 ServeHTTP 函数中，将函数的入参 req 赋值给了 Context 的 Request 对象。那么，这个 req 变量对象是从哪里来的呢？

Engine.ServeHTTP函数是在 net/http/server.go 文件的 conn.serve 函数中被调用的。conn代表本次申请的连贯对象。conn.serve函数首先会从 conn 对象中读取出本次的申请参数（蕴含申请头等），并解析到 request 对象中，将该对象再赋值给 Context 中的 Request 字段。这样，通过 Context 的 Param、Form、Header 等函数，就能够从 Request 中读取信息了。

在具体的申请处理函数中，咱们发现所有的输入都是通过 Context.Writer 对象的 Write 函数将数据返回给客户端的。那么，这里的 Context.Writer 对象是什么呢，为什么通过 Context.Writer 就是能后果返回给客户端呢？

要答复这个问题，咱们还须要回到 net/http/server.go 文件的 conn.serve 函数中，response参数是在该函数中调用 ServeHTTP 时传入的参数，而该 response 参数是通过 conn.readRequest 函数返回的。

咱们再通过 readRequest 函数来看下 response 构造体的关键字段，如下：

咱们再来看 Engine.ServeHTTP 函数：
首先，将 response 对象赋值给 Context.writermem 对象。即在 c.writermem.reset(w)函数中执行的。
其次，Context.Writer 是对 Context.writermem 的援用。即 Context.Writer 仍然是指向 response 对象。即在 c.reset()函数中执行的。

所以，在 Context 中跟 response 无关的关键字段如下：

最初，在业务逻辑中应用 Context.Writer.Write 函数输入时，实际上是执行的 response.Write 函数。

咱们再来看下 response.Write 函数的实现：

写入的时候是调用的 response 的w字段写入的。那 w 字段又是什么呢？咱们再回到 server.go 中的对 response 初始化的 readRequest 函数中，如下：

能够看到 w 是response的对象。同时将 w 还赋值给了 w.cw.res。最初，w.w 字段是基于 w.cw 的一个 bufio.Writer 对象。当调用 bufio.Writer 的Flush时，实际上是调用了 bufio.Writer 中的 wr 的Write办法。而 wr 的又是指向了 chunkWriter 对象的 Write 办法。最初该办法是执行了 w.conn.bufw.Write 写入了数据。