feat(Go-StudyExample): 2020/12/9 ：新增Gin框架使用示例

example/gin/30-DiyStructValid_test.go

@@ -0,0 +1,64 @@
+/*
+ *  @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()})
+	}
+}