HTTP请求验证中间件In Go

slwdgvem  于 2022-12-07  发布在  Go
关注(0)|答案(2)|浏览(200)

我尝试创建一个通用的HTTP请求验证器中间件函数,它接受type(可能是reflect.type)作为参数,然后使用包github.com/go-playground/validator/v10将JSON解组到上述类型的struct中,并验证该结构。
实施例

type LoginRequestBody struct {
   Username string `json:"username",validate:"required"`
   Password string `json:"password",validate:"required"`
}

type SignupReqBody struct {
   Username string `json:"username",validate:"required"`
   Password string `json:"password",validate:"required"`
   Age      int    `json:"age",validate:"required"`
}

// sample routers with a common middleware validator function
router.POST("/login", ReqValidate("LoginRequestBody"), LoginController)
router.POST("/signup", ReqValidate("SignupReqBody"), SignupController)

func ReqValidate(<something>) gin.HandlerFunc {
   return func (c *gin.Context) {
      // unmarshalling JSON into a struct
      // common validation logic...
      c.Next()
   }
}

总的来说,我希望使用Joi包实现与Node.js中相同的验证器灵活性。

fnx2tebb

fnx2tebb1#

我不知道是否有必要使用中间件,但我最近尝试做一些事情,我发现了一个很好的教程,你可以看到here
您可以使用绑定:

范例:

package main
import (
  "github.com/gin-gonic/gin"
  "net/http"
)
type AnyStruct struct {
   Price uint `json:"price" binding:"required,gte=10,lte=1000"`
}
func main() {
  engine:=gin.New()
  engine.POST("/test", func(context *gin.Context) {
     body:=AnyStruct{}
     if err:=context.ShouldBindJSON(&body);err!=nil{
        context.AbortWithStatusJSON(http.StatusBadRequest,
        gin.H{
            "error": "VALIDATEERR-1",
            "message": "Invalid inputs. Please check your inputs"})
        return
     }
     context.JSON(http.StatusAccepted,&body)
  })
  engine.Run(":3000")
}
xt0899hw

xt0899hw2#

不要使用逗号来分隔结构标记键值对,使用空格。
您可以使用泛型(类型参数)来替换<something>,但您的控制器需要将具体类型作为其参数。
例如:

func ReqValidate[T any](next func(*gin.Context, *T)) gin.HandlerFunc {
    return func(c *gin.Context) {
        params := new(T)
        if err := c.ShouldBindJSON(params); err != nil {
            c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
            return
        }
        next(c, params)
    }
}

然后是控制器:

type LoginRequestBody struct {
    Username string `json:"username" validate:"required"`
    Password string `json:"password" validate:"required"`
}

func LoginController(c *gin.Context, params *LoginRequestBody) {
    // ...
}

type SignupReqBody struct {
    Username string `json:"username" validate:"required"`
    Password string `json:"password" validate:"required"`
    Age      int    `json:"age" validate:"required"`
}

func SignupController(c *gin.Context, params *SignupReqBody) {
    // ...
}

然后是路由:

router := gin.Default()
router.POST("/login", ReqValidate(LoginController))
router.POST("/signup", ReqValidate(SignupController))

相关问题