使用分块传输编码处理请求

时间:2017-12-22 20:25:15

标签: go httprequest chunked-encoding chunked transfer-encoding

golang的net / http包是否支持使用分块传输编码的请求?到目前为止,我已经能够使用Hijacker接口(https://golang.org/src/net/http/server.go?s=6173:6875#L156)至少不关闭连接并接收完整的chunked请求,但还没有解析块并怀疑我可能会走错路径

https://golang.org/src/net/http/httputil/httputil.go?s=688:732#L10开始,我看到有一个分块阅读器,但似乎是供内部使用。

基本上,我正在尝试接受带有“分块”传输编码的HTTP PUT,并将其“即时”发送到后端服务器(即不在缓存中缓冲完整请求)。我无法控制上游请求。有没有推荐的方法来处理这样的请求,或者Hijacker是这样做的吗?

1 个答案:

答案 0 :(得分:4)

net / http客户端和服务器透明地读取和写入分块的主体。

要接受分块请求并将其发送到另一个HTTP服务器,请将服务器请求主体作为客户端请求主体传递。现在将身体转发到另一台服务器作为PUT:

 func handler(w http.ResponseWriter, r *http.Request) {
    creq, err := http.NewRequest("PUT", url, r.Body)
    if err != nil {
       // handle error
    }
    if ct := r.Header.Get("Content-Type"); ct != "" {
        creq.Header.Set("Content-Type", ct)
    }
    cresp, err := http.DefaultClient.Do(creq)
    if err != nil {
        // handle error
    }
    ... do something with cresp.
 }

如果要复制到文件,请将请求正文io.Copy复制到文件中。

 func handler(w http.ResponseWriter, r *http.Request) {
     f, err := os.Create("fname")
     if err != nil {
         // handle error
     }
     _, err := io.Copy(f, r.Body)
     if err != nil {
          // handle error
     }
     ...
  }

这些片段会“动态”复制身体。