两种方法实现 Http Request Body 多次读取

大家好， 我是 老麦， 一个运维老兵， 现在专注于 Golang，DevOps，云原生基础设施建设。19y28资讯网——每日最新资讯28at.com

原文链接: https://typonotes.com/posts/2024/01/02/http-request-multiple-times-read/19y28资讯网——每日最新资讯28at.com

最近在使用 gin 的时候， 踩了一个重复读取的 Request.Body 的坑。19y28资讯网——每日最新资讯28at.com

起因是 gin 的 gin.Context{} 提供了 c.Copy() 方法创建副本。这个方法一直在用， 但不知道从什么时候开始， 一直认为这个方法是 深拷贝， 但 并不完全是 (T_T)19y28资讯网——每日最新资讯28at.com

// Copy returns a copy of the current context that can be safely used outside the request's scope.// This has to be used when the context has to be passed to a goroutine.func (c *Context) Copy() *Context { cp := Context{  writermem: c.writermem,  Request:   c.Request, // 指针， 也算引用类型。 没有实现完全复制  Params:    c.Params,  engine:    c.engine, } cp.writermem.ResponseWriter = nil cp.Writer = &cp.writermem cp.index = abortIndex cp.handlers = nil cp.Keys = map[string]interface{}{} // Keys 完全复制 for k, v := range c.Keys {  cp.Keys[k] = v } paramCopy := make([]Param, len(cp.Params)) // 切片， 完全复制 copy(paramCopy, cp.Params)  cp.Params = paramCopy return &cp}

1. gin 通过用一个全局变量保存

在 gin 中, 在读取了 request body 后， 通过 c.Set(BodyBytesKey, body) 放到了 gin.Context 中的 Keys。这是一个 map， 上面说到了。19y28资讯网——每日最新资讯28at.com

因此 在 gin 中通过中间变量实现类似效果。虽然感觉上多次读取 Body ， 但实际 只读取了一次，19y28资讯网——每日最新资讯28at.com

// ShouldBindBodyWith is similar with ShouldBindWith, but it stores the request// body into the context, and reuse when it is called again.//// NOTE: This method reads the body before binding. So you should use// ShouldBindWith for better performance if you need to call only once.func (c *Context) ShouldBindBodyWith(obj any, bb binding.BindingBody) (err error) { var body []byte if cb, ok := c.Get(BodyBytesKey); ok {  if cbb, ok := cb.([]byte); ok {   body = cbb  } } if body == nil {  body, err = io.ReadAll(c.Request.Body)  if err != nil {   return err  }  // 将 Body 中的内容放到 gin.Context 中的 Keys 中  c.Set(BodyBytesKey, body) } return bb.BindBody(body, obj)}

参考文档: https://github.com/gin-gonic/gin/blob/v1.9.1/context.go#L744-L76419y28资讯网——每日最新资讯28at.com

2. 再造一个 Request

另外一种方法， 就是在读取 Body 后， 重建一个 Requset 再把 Body 放进去。19y28资讯网——每日最新资讯28at.com

// 读取老的body, err := ioutil.ReadAll(r.Body)if err != nil {    // ...}url, _ := url.Parse(config.GetGameHost())// 创建新的r2 := r.Clone(r.Context())// 将数据方进去r.Body = ioutil.NopCloser(bytes.NewReader(body))r2.Body = ioutil.NopCloser(bytes.NewReader(body))r.ParseForm()proxy := httputil.NewSingleHostReverseProxy(url)proxy.ServeHTTP(w, r2)

参考文档: https://stackoverflow.com/q/6201714619y28资讯网——每日最新资讯28at.com

注意 http.Request 有一个方法叫 Clone(), 但这也不是一个完全的深拷贝。Body 没有复制。19y28资讯网——每日最新资讯28at.com

// Clone returns a deep copy of r with its context changed to ctx.// The provided ctx must be non-nil.//// For an outgoing client request, the context controls the entire// lifetime of a request and its response: obtaining a connection,// sending the request, and reading the response headers and body.func (r *Request) Clone(ctx context.Context) *Request { if ctx == nil {  panic("nil context") } r2 := new(Request) *r2 = *r r2.ctx = ctx r2.URL = cloneURL(r.URL) if r.Header != nil {  r2.Header = r.Header.Clone() } if r.Trailer != nil {  r2.Trailer = r.Trailer.Clone() } if s := r.TransferEncoding; s != nil {  s2 := make([]string, len(s))  copy(s2, s)  r2.TransferEncoding = s2 } r2.Form = cloneURLValues(r.Form) r2.PostForm = cloneURLValues(r.PostForm) r2.MultipartForm = cloneMultipartForm(r.MultipartForm) return r2}

本文链接：http://www.28at.com/showinfo-26-56416-0.html两种方法实现 Http Request Body 多次读取

声明：本网页内容旨在传播知识，若有侵权等问题请及时与本网联系，我们将在第一时间删除处理。邮件：2376512515@qq.com

上一篇： ExecutorCompletionService详解，你学会了吗？

下一篇： 2023 年十种最佳用户体验交互设计