如何使用 go 框架进行输入验证?在 go web 开发中,输入验证至关重要。本文教程使用 gorilla/mux 和 validator 框架来实现输入验证。步骤:安装 gorilla/mux 和 validator 框架使用 gorilla/mux 处理用户请求,获取输入数据并进行 json 解析使用 validator 验证 go 结构,包括自定义验证规则使用验证结果处理请求
如何使用 Go 框架进行输入验证
在 Web 开发中,输入验证至关重要,它可以确保用户输入的数据符合预期。在 Go 中,有多种框架可用于实现输入验证,本教程将使用流行的 Gorilla/mux 和 Validator 框架进行演示。
安装
立即学习“go语言免费学习笔记(深入)”;
go get -u github.com/gorilla/mux
go get -u github.com/go-playground/validator/v10
使用 Gorilla/mux
Gorilla/mux 是一个 HTTP 路由器,可用于处理用户请求。
import (
"github.com/gorilla/mux"
)
func InputValidationHandler(w http.ResponseWriter, r *http.Request) {
// 从请求中获取输入数据
body, err := ioutil.ReadAll(r.Body)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
// 解析 JSON 输入
data := struct {
Name string `json:"name"`
Email string `json:"email"`
Address string `json:"address"`
}{}
if err := json.Unmarshal(body, &data); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
// ...输入验证代码...
}
使用 Validator
Validator 是一个用于验证 Go 结构的强大库。
import (
"github.com/go-playground/validator/v10"
)
func InputValidationWithValidator(w http.ResponseWriter, r *http.Request) {
// 创建一个验证器
validate := validator.New()
// 验证输入
body, err := ioutil.ReadAll(r.Body)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
data := struct {
Name string `json:"name" validate:"required,min=3,max=25"`
Email string `json:"email" validate:"required,email"`
Address string `json:"address" validate:"required,min=10"`
}{}
if err := json.Unmarshal(body, &data); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
if err := validate.Struct(data); err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
// ...成功处理...
}
实战案例
我们使用上面的例程来创建一个简单的 HTTP API 端点,它验证用户的注册数据:
router := mux.NewRouter()
router.HandleFunc("/register", InputValidationWithValidator).Methods("POST")
http.ListenAndServe(":8080", router)
这个端点将验证输入数据,包括名称、电子邮件和地址,如果没有通过验证,则返回错误响应。