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

程序员小助手

发布于 2020-04-26 13:54:52

3590

发布于 2020-04-26 13:54:52

文章被收录于专栏：程序员小助手

引言

在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, headers := 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 https://blabla.example.com/o/oauth2/token 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 :)

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2020-04-23，如有侵权请联系 cloudcommunity@tencent.com 删除

http

本文分享自程序员小助手微信公众号，前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

http

登录后参与评论

0 条评论

热度