Question

我需要验证我的http请求是否具有两个参数，即Start和End。当前，我设置了一个默认值，该默认值不应作为任何一个参数出现，并与其他无效值一起检查它。但是，这感觉像是黑客。正确的方法应该是什么？

这是我的代码：

type Request struct {
    Start int `json: "start"`
    End int `json: "end"`
}


func HandlePost(w http.ResponseWriter, r *http.Request) {
    body , _ := ioutil.ReadAll(r.Body)
    reqData := Request{Start: -1, End: -1} // < whats the correct way to do this
    json.Unmarshal(body, &reqData)  

    if reqData.Start < 0 && reqData.End < 0 {
        w.WriteHeader(http.StatusBadRequest)
        return
    }
    // rest of the logic
}

Answer 1

您可以使用https://github.com/asaskevich/govalidator作为验证请求的基本方法。但是，如果您需要更复杂的功能，则需要编写自己的自定义验证器函数。例如

type Request struct {
    Start int `json: "start"`
    End int `json: "end"`
}

func (a *Request) validate() url.Values {
    err := url.Values{}

    // Write your own validation rules
    if a.Start < 0 {
        err.Add("Start", "Start cannot be less than 0");
    }

    return err;
}

func handler(w http.ResponseWriter, r *http.Request) {
    requestBody := &Request{}

    defer r.Body.Close()
    if err := json.NewDecoder(r.Body).Decode(requestBody); err != nil {
        panic(err)
    }

    if errors := requestBody.validate(); len(errors) > 0 {
        err := map[string]interface{}{"validationError": errors}
        w.Header().Set("Content-type", "application/json")
        w.WriteHeader(http.StatusBadRequest)
        json.NewEncoder(w).Encode(err)
    }

    fmt.Fprint(w, "success request scenario!")
}

Answer 2

您可以使用https://github.com/buger/jsonparser getInt。如果json缺少预期的密钥，则会出现错误。

我建议使用基准测试，而不是由代码美感或其他任何预感所决定

Answer 3

这是使用struct标记和指针验证结构的另一种方法。请注意，如果传递的有效值是0，则此解决方案将不起作用。 omitempty认为0值为空。如果您希望这样做，则将0视为有效，请删除指针并修改IsValid方法

package main

import (
    "encoding/json"
    "fmt"
)

type Request struct {
    Start *int `json: "start,omitempty"`
    End   *int `json: "end,omitempty"`
}

func (r Request) IsValid() (bool, error) {
    if r.Start == nil {
        return false, fmt.Errorf("start is missing")
    }

    if r.End == nil {
        return false, fmt.Errorf("end is missing")
    }

    return true, nil
}

var (
    invalidStartb = `{"end": 1}`
    invalidEndb   = `{"start": 1}`
    valid         = `{"start": 1, "end": 1}`
)

func main() {
    var r Request

    _ = json.Unmarshal([]byte(invalidStartb), &r)
    fmt.Println(r.IsValid())

    r = Request{}
    _ = json.Unmarshal([]byte(invalidEndb), &r)
    fmt.Println(r.IsValid())

    r = Request{}
    _ = json.Unmarshal([]byte(valid), &r)
    fmt.Println(r.IsValid())

}

此处为可运行版本https://goplay.space/#Z0eqLpEHO37

在Go中验证HTTP请求的惯用方式

3 个答案: