10. Go使用validator库进行参数校验

package main
 
import (
    "fmt"
    "net/http"
 
    "github.com/go-playground/validator/v10"
)
 
// 定义一个结构体用于表单验证
type RegisterRequest struct {
    Username string `json:"username" validate:"required"`
    Password string `json:"password" validate:"required,min=6"`
    Email    string `json:"email" validate:"required,email"`
}
 
// 使用validator.New()创建一个验证器实例
var validate *validator.Validate = validator.New()
 
func Register(w http.ResponseWriter, r *http.Request) {
    if r.Method != http.MethodPost {
        http.Error(w, http.StatusText(http.StatusMethodNotAllowed), http.StatusMethodNotAllowed)
        return
    }
 
    var registerRequest RegisterRequest
    if err := json.NewDecoder(r.Body).Decode(&registerRequest); err != nil {
        http.Error(w, err.Error(), http.StatusBadRequest)
        return
    }
 
    // 使用validate.Struct()对结构体进行验证
    if err := validate.Struct(registerRequest); err != nil {
        if _, ok := err.(*validator.InvalidValidationError); ok {
            http.Error(w, err.Error(), http.StatusInternalServerError)
            return
        }
 
        // 验证失败，输出错误信息
        fmt.Fprintln(w, err)
        return
    }
 
    // 验证成功，继续处理注册逻辑
    fmt.Fprintln(w, "注册成功")
}
 
func main() {
    http.HandleFunc("/register", Register)
    http.ListenAndServe(":8080", nil)
}

这段代码定义了一个用于注册的结构体RegisterRequest，并使用validator.New()创建了一个验证器实例。在Register函数中，它首先尝试解析请求体为RegisterRequest类型的实例，然后使用validate.Struct()方法对解析后的实例进行验证。如果验证失败，它会输出错误信息；如果验证成功，则继续处理注册逻辑。这个例子展示了如何在Go中使用validator库进行参数验证，并且是一个简单的注册接口示例。