在Golang中两次读取请求正文

soh*_*dia 2 json request go

type ValidationModel struct {
    Name     string `json:"name" valid:"alpha,required~Name is required"`
    Email    string `json:"email" valid:"email~Enter a valid email.,required~Email is required."`
    Password string `json:"password" valid:"required~Password is required"`
}

validationModel := ValidationModel{}

json.NewDecoder(r.Body).Decode(&validationModel)

_, err := govalidator.ValidateStruct(validationModel)
Run Code Online (Sandbox Code Playgroud)

首先,我使用govalidator验证请求正文。

type UserModel struct {
    ID           bson.ObjectId `json:"_id" bson:"_id"`
    Name         string        `json:"name" bson:"name"`
    Email        string        `json:"email" bson:"email"`
    Password     string        `json:"password,omitempty" bson:"-"`
    PasswordHash string        `json:"-" bson:"passwordHash"`
    Salt         string        `json:"-" bson:"salt"`
    Token        string        `json:"token,omitempty" bson:"-"`
}


user := models.UserModel{}
json.NewDecoder(r.Body).Decode(&user)

fmt.Println(user)
Run Code Online (Sandbox Code Playgroud)

在验证请求之后,我再次将请求主体解码为用户struct,但已使用validationModel读取了请求主体一次,因此当我尝试再次将其解码为用户时,它没有提供任何值。

我可以在这里想到两种解决方案:

  1. 将请求正文存储在一个单独的变量中,并使用该变量两次。

  2. 复制用户中的validationModel值。

但是我对实现这些方法一无所知,最好采用哪种方法。还是可以实施其他解决方案?

提前致谢。

poy*_*poy 6

可以使用ioutil.ReadAll()轻松地存储数据:

data, err := ioutil.ReadAll(r.Body)
Run Code Online (Sandbox Code Playgroud)

如果需要data背面为io.Reader(这样r.Body),则可以使用bytes.NewReader():

reader := bytes.NewReader(data)
Run Code Online (Sandbox Code Playgroud)

r.Body实际上,是一个io.ReadCloser,因此,如果需要,可以将ioutil.NopCloser()与bytes.NewReader()结合使用:

reader := ioutil.NopCloser(bytes.NewReader(data))
Run Code Online (Sandbox Code Playgroud)

  • 当 r.Body 非常大,比如 1G Bytes 时,使用 ioutil.ReadAll 不是一个好主意。 (3认同)