Go中的HTTP请求验证中间件

Question

我试图创建一个通用的HTTP请求验证中间件函数，该函数接受类型（也许是reflect.Type）作为参数，然后使用github.com/go-playground/validator/v10包将JSON解析为所提到类型的结构体并验证其有效性。我尝试通过以下示例代码进行说明...

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

总的来说，我希望能够达到与Node.js中使用Joi包相同的验证器灵活性。

- sajal arora

Answer 1

我不确定是否需要使用中间件，但最近我尝试做某些事情时发现了一篇非常好的教程，您可以在这里看到。

使用Gin，您可以使用绑定：

示例：

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

- Juanes30

---

这对我来说似乎非常完美。非常感谢... - sajal arora

Answer 2

不要使用逗号来分隔结构体标签的键值对，应该使用空格。

您可以使用泛型（类型参数）来替换<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))

- mkopriva