go gin 优雅的校验参数-Go服务器开发

在使用 Go 和 Gin 框架进行参数校验时，通常可以通过以下几种方式来优雅地实现参数校验：

1. 使用 Gin 自带的绑定和验证

Gin 提供了 ShouldBindJSON、ShouldBindQuery 和 ShouldBindURI 等方法，这些方法结合结构体标签可以方便地进行参数校验。

首先，你需要安装 go-playground/validator 包，因为 Gin 内部使用这个包来进行验证。

go get github.com/go-playground/validator/v10

然后，你可以定义一个结构体，并为字段添加验证标签。例如：

package main

import (
    "github.com/gin-gonic/gin"
    "net/http"
)

type User struct {
    Name  string `json:"name" binding:"required"`
    Age   int    `json:"age" binding:"required,gte=0,lte=130"` // 年龄必须在0到130之间
}

func main() {
    router := gin.Default()

    router.POST("/user", func(c *gin.Context) {
        var user User
        
        // 绑定并校验请求数据
        if err := c.ShouldBindJSON(&user); err != nil {
            c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
            return
        }

        // 如果没有错误，可以继续处理逻辑
        c.JSON(http.StatusOK, gin.H{"message": "User is valid!", "user": user})
    })

    router.Run()
}

2. 自定义验证器

如果内置的验证规则无法满足需求，你可以自定义验证器。例如：

package main

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"net/http"
	"regexp"
)

type User struct {
	Name string `json:"name" binding:"required"`
	Email string `json:"email" binding:"required,email"`
}

func EmailValidator(fl validator.FieldLevel) bool {
	re := regexp.MustCompile(`^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$`)
	return re.MatchString(fl.Field().String())
}

func main() {
	router := gin.Default()

	if v, ok := binding.Validator.Engine().(*validator.Validate); ok {
		v.RegisterValidation("email", EmailValidator)
	}

	router.POST("/user", func(c *gin.Context) {
		var user User
		
		if err := c.ShouldBindJSON(&user); err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
			return
		}

		c.JSON(http.StatusOK, gin.H{"message": "User is valid!", "user": user})
	})

	router.Run()
}