Huangzj
65 lines
2.0 KiB
Go
65 lines
2.0 KiB
Go
/*
|
||
* @Author : huangzj
|
||
* @Time : 2020/12/8 15:12
|
||
* @Description：
|
||
*/
|
||
|
||
package gin
|
||
|
||
import (
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/gin-gonic/gin/binding"
|
||
"github.com/go-playground/validator/v10"
|
||
"net/http"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
// Booking 包含绑定和验证的数据。
|
||
type Booking struct {
|
||
CheckIn time.Time `form:"check_in" binding:"required,diyValid" time_format:"2006-01-02"`
|
||
CheckOut time.Time `form:"check_out" binding:"required,gtfield=CheckIn" time_format:"2006-01-02"`
|
||
}
|
||
|
||
//read note 自定义的校验类，需要实现 validator.Func方法(参数是validator.FieldLevel,这里把所有的参数都集合进去了，之前的版本是多个参数)
|
||
func bookDataValid(level validator.FieldLevel) bool {
|
||
if date, ok := level.Field().Interface().(time.Time); ok {
|
||
today := time.Now()
|
||
if today.Year() > date.Year() || today.YearDay() > date.YearDay() {
|
||
return false
|
||
}
|
||
}
|
||
|
||
return true
|
||
}
|
||
|
||
func TestDiyStructValid(t *testing.T) {
|
||
route := gin.Default()
|
||
|
||
//read note 注册自定义的校验类,这边设置的key就是需要在结构体上绑定数据时候书写的tag标签，注意看上面Booking 的CheckIn标签(只有在结构体上书写了该标签，这个校验器才会去校验对应的字段)
|
||
if v, ok := binding.Validator.Engine().(*validator.Validate); ok {
|
||
_ = v.RegisterValidation("diyValid", bookDataValid)
|
||
}
|
||
|
||
route.GET("/bookable", getBookable)
|
||
_ = route.Run(":8085")
|
||
|
||
//请求示例
|
||
_ = `
|
||
$ curl "localhost:8085/bookable?check_in=2018-04-16&check_out=2018-04-17"
|
||
{"message":"Booking dates are valid!"}
|
||
|
||
$ curl "localhost:8085/bookable?check_in=2018-03-08&check_out=2018-03-09"
|
||
{"error":"Key: 'Booking.CheckIn' Error:Field validation for 'CheckIn' failed on the 'diyValid' tag"}
|
||
`
|
||
}
|
||
|
||
func getBookable(c *gin.Context) {
|
||
var b Booking
|
||
if err := c.ShouldBindWith(&b, binding.Query); err == nil {
|
||
c.JSON(http.StatusOK, gin.H{"message": "Booking dates are valid!"})
|
||
} else {
|
||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||
}
|
||
}
|