Подготовить HY2XS к production-сборке

apps/controller/validator.go

@@ -0,0 +1,41 @@
+package controller
+
+import (
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"github.com/go-playground/validator/v10"
+	"hy2xs-admin/model/constant"
+	"hy2xs-admin/model/vo"
+	"net/http"
+	"regexp"
+)
+
+var validate *validator.Validate
+
+func init() {
+	validate = validator.New()
+	_ = validate.RegisterValidation("validateStr", validateStr)
+}
+
+func validateStr(f validator.FieldLevel) bool {
+	field := f.Field().String()
+	// 字符串必须6-32位是字母或者数字或部分特殊字符的组合
+	reg := "^[a-zA-Z0-9!@#$%^&*()_+-=]{6,32}$"
+	compile := regexp.MustCompile(reg)
+	return field == "" || compile.MatchString(field)
+}
+
+func validateField[T interface{}](c *gin.Context, field T) (T, error) {
+	if c.Request.Method == http.MethodGet {
+		_ = c.ShouldBindQuery(&field)
+	} else if c.Request.Method == http.MethodPost ||
+		c.Request.Method == http.MethodPut ||
+		c.Request.Method == http.MethodDelete {
+		_ = c.ShouldBindJSON(&field)
+	}
+	if err := validate.Struct(&field); err != nil {
+		vo.Fail(constant.InvalidError, c)
+		return field, fmt.Errorf(constant.InvalidError)
+	}
+	return field, nil
+}