http.request 的 golang 多个 parseBody

golang multiple parseBody for http.request

嘿,我想像下面那样解析 http.resquest 两次。当我第一次解析 Body 时,body 将被关闭。我需要一些 help/hint 处理此问题的最佳方法是什么,我必须创建请求的副本还是有更好的方法?

func myfunc(w http.ResponseWriter, req *http.Request) {
  err := parseBody(req, &type1){
  .....
  }

  err := parseBody(req, &type2){
  .....
  }
}

感谢帮助

当您阅读 request.Body 时,您正在从客户端(例如网络浏览器)读取流。客户端只发送一次请求。如果您想多次解析它,请将整个内容读入缓冲区(例如 []byte),然后根据需要多次解析。请注意许多具有大有效负载的并发请求的潜在内存使用,因为您至少要将完整的有效负载保存在内存中,直到您完全解析完它。

的确,您只能读取一次正文,这没关系,因为要多次解析正文,您不必多次读取它。让我们考虑一个简单的例子:

package main

import (
    "encoding/json"
    "fmt"
    "io/ioutil"
    "net/http"
)

type RequestData1 struct {
    Code   string `json:"code"`
    Status string `json:"status"`
}

type RequestData2 struct {
    Status  string `json:"status"`
    Message string `json:"message"`
}

func main() {
    http.HandleFunc("/post", post)
    http.ListenAndServe(":8080", nil)
}

如果我们使用这段代码:

func post(w http.ResponseWriter, r *http.Request) {
    body1, err := ioutil.ReadAll(r.Body)
    if err != nil {
        panic(err)
    }
    rd1 := RequestData1{}
    err = json.Unmarshal(body1, &rd1)
    if err != nil {
        panic(err)
    }

    body2, err := ioutil.ReadAll(r.Body)
    if err != nil {
        panic(err)
    }
    rd2 := RequestData2{}
    err = json.Unmarshal(body2, &rd2)
    if err != nil {
        panic(err) // panic!!!
    }

    fmt.Printf("rd1: %+v \nrd2: %+v", rd1, rd2)
    w.WriteHeader(http.StatusOK)
    w.Write([]byte(`Look into console.`))
}

我们会恐慌:http: panic serving [::1]:54581: unexpected end of JSON input
但下一个代码:

func post(w http.ResponseWriter, r *http.Request) {
    body, err := ioutil.ReadAll(r.Body)
    if err != nil {
        panic(err)
    }

    rd1 := RequestData1{}
    err = json.Unmarshal(body, &rd1)
    if err != nil {
        panic(err)
    }

    rd2 := RequestData2{}
    err = json.Unmarshal(body, &rd2)
    if err != nil {
        panic(err)
    }

    fmt.Printf("rd1: %+v \nrd2: %+v", rd1, rd2)
    w.WriteHeader(http.StatusOK)
    w.Write([]byte(`Look into console.`))
}

一切正常!您可以通过发出请求来测试它:

curl -X POST 'http://localhost:8080/post' \
-H 'Content-Type: application/json' -d '{"code":"200", "status": "OK", "message": "200 OK"}'

结果将是:

rd1: {Code:200 Status:OK}
rd2: {Status:OK Message:200 OK}