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()})
|
|||
|
|
}
|
|||
|
|
}
|