Gin(四):表单提交校验和模型绑定
文章首发于 ISLAND
上一个章节中已经开始逐渐搭建了一个 web 页面,现在我们开始逐步完善页面上的功能,首先要完成的是登录和注册功能。
接受表单数据
注册页面的 HTML 元素不在详细写出,具体页面代码可以直接参考Github 上代码。
页面完成后布局:
注册页面有三个输入框,分别为 email
,password
和 password again
。
完善后端 Gin 代码。我们在 initRouter
中 userGroup
中编写新的接口。
userRouter.POST("/register", handler.UserRegister)
编写完新的接口就要开始编写 Handler
。
func UserRegister(context *gin.Context) {
email := context.PostForm("email")
password := context.DefaultPostForm("password", "Wa123456")
passwordAgain := context.DefaultPostForm("password-again", "Wa123456")
println("email", email, "password", password, "password again", passwordAgain)
}
UserRegister
方法中采用新的方式来接受 Post 请求提交的表单参数,PostForm
和 DefaultPostForm
。PostForm
直接接受参数,而 DefaultPostForm
可以设置一个默认值,如果前端没有进行传值,那么我们可以设置默认值,如上面的代码,如果前端没有将密码传输过来我们可以设置一个默认密码。
当我们运行并且输入的时候,在控制台上可以清楚的看到我们在表单上的输入。
当我们项目功能完善的时候,就可以完善我们的单元测试。
此时的单元测试交之前有点复杂。
首先我们要构造一个结构,该结构是为了帮助我们将我们要提交的信息存放到表单中,同时要指定请求头信息。
func TestUserPostForm(t *testing.T) {
value := url.Values{}
value.Add("email", "youngxhui@gmail.com")
value.Add("password", "1234")
value.Add("password-again", "1234")
w := httptest.NewRecorder()
req, _ := http.NewRequest(http.MethodPost, "/user/register", bytes.NewBufferString(value.Encode()))
req.Header.Add("Content-Type", "application/x-www-form-urlencoded; param=value")
router.ServeHTTP(w, req)
assert.Equal(t, http.StatusOK, w.Code)
}
单元测试编写完成后可以运行单元测试,发现控制台答应了我们在测试中写的数据。
模型绑定
上例中我们的表单仅仅传输了三个参数,如果后期项目出现了十多个参数,每次写一遍都很花费时间,也很消耗经历。下面就对该方法进行改善
Gin 中提供了 模型绑定,将我们的表单数据与我们的模型进行一样绑定。GIn会将数据统一封装到模型中,方便我们日后使用。
首先定义我们的模型,新建 model
文件夹,建立 userModel.go
package model
type UserModel struct {
Email string `form:"email"`
Password string `form:"password"`
PasswordAgain string `form:"password-again"`
}
通过 form:"email"
来对表单中的 email
输入数据进行绑定。然后需要修改一下 Handler
方法。
func UserRegister(context *gin.Context) {
var user model.UserModel
if err := context.ShouldBind(&user); err != nil {
println("err ->", err.Error())
return
}
println("email", user.Email, "password", user.Password, "password again", user.PasswordAgain)
}
此时我们的模型绑定已经写好,运行 TestUserPostForm
测试用例,测试用例可以完美的通过。说明我们的模型绑定方法是正确的。同时模型绑定还是从 json
, xml
,yml
等格式数据的绑定,日后会有介绍和说明。当然也可以通过浏览器中的注册表单进行提交。
数据校验
做后端开发的人都明白一个道理:永远不要相信前端传过来的数据。所有的数据在进过后端时,务必要进行数据的校验。
在模型中可用 binding
来对数据进行校验。Gin 对于数据校验使用的是 validator.v8
库,该库提供多种校验方法。通过 binding:""
方式来进行对数据的校验。
我们将 UserModel
进行修改,添加一些规则,邮箱验证和密码校验,要求第二次重复密码要和第一次密码一致。更多的校验规则可以看官方文档
type UserModel struct {
Email string `form:"email" binding:"email"`
Password string `form:"password"`
PasswordAgain string `form:"password-again" binding:"eqfield=Password"`
}
我们重新写一个测试用例用来测试邮箱和密码校验是否有效。
func TestUserPostFormEmailErrorAndPasswordError(t *testing.T) {
value := url.Values{}
value.Add("email", "youngxhui")
value.Add("password", "1234")
value.Add("password-again", "qwer")
w := httptest.NewRecorder()
req, _ := http.NewRequest(http.MethodPost, "/user/register", bytes.NewBufferString(value.Encode()))
req.Header.Add("Content-Type", "application/x-www-form-urlencoded; param=value")
router.ServeHTTP(w, req)
assert.Equal(t, http.StatusOK, w.Code)
}
运行测试,发现测试虽然通过了,但是会有两行 error
信息
err -> Key: 'UserModel.Email' Error:Field validation for 'Email' failed on the 'email' tag
Key: 'UserModel.PasswordAgain' Error:Field validation for 'PasswordAgain' failed on the 'eqfield' tag
该信息说明了我们的 Email
和 PasswordAgain
信息校验没有通过。
使用Log和重定向
测试通过是因为无论我们代码如何都会返回 200 状态码,这是不符合http 状态码的规范的,所以我们要对http状态码进行规范化。同时我们之前的代码中一直使用 Printf
来打印日志信息,也是不规范的,因为 Printf
打印的日志信息相对局限,所以应该选用 Log
进行日志打印。
func UserRegister(context *gin.Context) {
var user model.UserModel
if err := context.ShouldBind(&user); err != nil {
log.Println("err ->", err.Error())
context.String(http.StatusBadRequest, "输入的数据不合法")
} else {
log.Println("email", user.Email, "password", user.Password, "password again", user.PasswordAgain)
context.Redirect(http.StatusMovedPermanently, "/")
}
}
首先我们将原来只用 Println
打印的数据都改成了 log
去打印数据。
同时将原来的状态码都进行了更改,不同的状态码代表不同的请求响应结果。
最后在请求成功的时候我们对路由进行了重定向,将页面转跳到首页。
同时我们也要将测试用例里的返回状态码进行修改。
总结
本节将表单提交,模型绑定和数据校验有了一个相对细致的介绍,代码中也通过不同的测试用例来检查代码是否正确。
本章节代码
其他章节索引
Gin(一):Hello
Gin(二):路由Router
Gin(三):模板tmpl
Gin(四):表单提交校验和模型绑定
Gin(五):连接MySQL
Gin(六):文件的上传
Gin(七):中间件的使用和定义
Gin(八):Cookie的使用
个人公众号
最新文章都会在公众号上进行分享,欢迎各位大佬关注
本文来自博客园,作者:youngxhui,转载请注明原文链接:https://www.cnblogs.com/youngxhui/p/17730482.html