七叶笔记 » golang编程 » 哭晕！不懂搜索盲目造轮子：Golang打印请求体，大神2行搞定！

哭晕！不懂搜索盲目造轮子：Golang打印请求体，大神2行搞定！

分类: golang编程 | 浏览: 166

引言

在Golang与其他 web服务 (包括发送和接收)的接口调试，一直是一个痛点。将所有的值都放入fmt也挺麻烦的。

故障排除时，傻瓜式地fmt.Printf(” %+v “, req)，输出内容惨不忍睹。

为了简化操作，有必要写个函数，随处可用，于是有了下面的代码。

正文

直接上代码： 

 func formatRequest(r *http.Request) string {

 var request []string

 url := fmt.Sprintf(“%v %v %v”, r.Method, r.URL, r.Proto)
 request = append(request, url)

 request = append(request, fmt.Sprintf(“Host: %v”, r.Host))

 for name,  Header s := range r.Header {
   name = strings.ToLower(name)
   for _, h := range headers {
     request = append(request, fmt.Sprintf(“%v: %v”, name, h))
   }
 }
 
 // 处理POST请求体
 if r.Method == “POST” {
    r.ParseForm()
    request = append(request, “\n”)
    request = append(request, r.Form.Encode())
 } 

  return strings.Join(request, “\n”)
}

输出结果看起来时这样的 

 POST  HTTP/1.1
Host: blabla.example.com
content-type: application/x-www-form-urlencoded
client_id=ssssss&client_secret=sssss&grant_type=authorization_code&redirect_uri=ssssss&scope=https%3A%2F%2Fwww.googleapis.com%2Fauth%2Fuserinfo.email+https%3A%2F%2Fwww.googleapis.com%2Fauth%2Fuserinfo.profile

一个小问题是，如果读一个外部请求的 r.Body，只能去缓冲区。这意味着，请求有可能失败，因为发送的 Content-Length 头将设置为请求的原始长度。造成请求体的实际长度为0。

为了解决这个问题，可以把正文读入缓冲区，然后打印出请求后再写回来: 

 // 缓存请求体
bodyBuffer, _ := ioutil.ReadAll(r.Body)
req.Body = myReader{ bytes .NewBuffer(buf)}
fmt.Printf("--> %s\n\n", formatRequest(req))
req.Body = myReader{bytes.NewBuffer(buf)}

轮子已经好了

无可避免，还是搜索的不够彻底，这个需求的轮子早就造好，放在golang里了。

httputil包有一个预先准备好的DumpRequest方法。输出几乎完全一样，这太赞的，因为它少了对第三方代码的依赖。

结果就是2行搞定，还有2行是错误处理。 

 requestDump, err := httputil.DumpRequest(req, true)
if err != nil {
  fmt.Println(err)
}
fmt.Println(string(requestDump))

req参数是http， true让函数知道是否存在请求体。

写在最后

大神都已经造好轮子了，我们只用找出来装上用就行了。

Happy coding 🙂

相关文章