设为首页 加入收藏

TOP

[系列] Gin框架 - 数据绑定和验证(一)
2019-07-25 14:17:27 】 浏览:111
Tags:系列 Gin 框架 数据 绑定 验证

概述

上篇文章分享了 Gin 框架使用 Logrus 进行日志记录,这篇文章分享 Gin 框架的数据绑定与验证。

有读者咨询我一个问题,如何让框架的运行日志不输出控制台?

解决方案:

engine := gin.Default() //修改成如下
engine := gin.New() 

我是怎么知道的?看框架代码。

Default()

func Default() *Engine {
    debugPrintWARNINGDefault()
    engine := New()
    engine.Use(Logger(), Recovery())
    return engine
}

New() 代码我就不贴了。

我们看到 Default() 使用了两个中间件 Logger(), Recovery(),如果不想使用,那就直接使用 New() 就可以了。

开始今天的文章。

比如,请求 v1/member/add 新增会员方法,nameage 为必填,同时 name 不能等于 admin 字符串,10 <= age <= 120。

直接看代码吧。

首先,先定义一个结构体。

entity/member.go

package entity

// 定义 Member 结构体
type Member struct {
    Name string `form:"name" json:"name" binding:"required,Nameva lid"`
    Age  int    `form:"age"  json:"age"  binding:"required,gt=10,lt=120"`
}

binding 中 required,这个是框架自带的,Nameva lid,这个是自己定义的。

问题一:框架自带的 binding 参数还有哪些?

问题二:自定义验证方法,怎么写?

接下来要说的就是问题二,写一个验证方法。

validator/member/member.go

package member

import (
    "gopkg.in/go-playground/validator.v8"
    "reflect"
)

func Nameva lid(
    v *validator.Validate, topStruct reflect.Value, currentStructOrField reflect.Value,
    field reflect.Value, fieldType reflect.Type, fieldKind reflect.Kind, param string,
) bool {
    if s, ok := field.Interface().(string); ok {
        if s == "admin" {
            return false
        }
    }
    return true
}

接下来,在路由中绑定:

router/router.go

package router

import (
    "ginDemo/middleware/logger"
    "ginDemo/middleware/sign"
    "ginDemo/router/v1"
    "ginDemo/router/v2"
    "ginDemo/validator/member"
    "github.com/gin-gonic/gin"
    "github.com/gin-gonic/gin/binding"
    "gopkg.in/go-playground/validator.v8"
)

func InitRouter(r *gin.Engine)  {

    r.Use(logger.LoggerToFile())

    // v1 版本
    GroupV1 := r.Group("/v1")
    {
        GroupV1.Any("/product/add", v1.AddProduct)
        GroupV1.Any("/member/add", v1.AddMember)
    }

    // v2 版本
    GroupV2 := r.Group("/v2").Use(sign.Sign())
    {
        GroupV2.Any("/product/add", v2.AddProduct)
        GroupV2.Any("/member/add", v2.AddMember)
    }

    // 绑定验证器
    if v, ok := binding.Validator.Engine().(*validator.Validate); ok {
        v.RegisterValidation("Nameva lid", member.Nameva lid)
    }
}

最后,看一下调用的代码。

router/v1/member.go

package v1

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

func AddMember(c *gin.Context) {

    res := entity.Result{}
    mem := entity.Member{}

    if err := c.ShouldBind(&mem); err != nil {
        res.SetCode(entity.CODE_ERROR)
        res.SetMessage(err.Error())
        c.JSON(http.StatusForbidden, res)
        c.Abort()
        return
    }

    // 处理业务(下次再分享)

    data := map[string]interface{}{
        "name" : mem.Name,
        "age"  : mem.Age,
    }
    res.SetCode(entity.CODE_ERROR)
    res.SetData(data)
    c.JSON(http.StatusOK, res)
}

访问看看效果吧。

访问:http://localhost:8080/v1/member/add

{
    "code": -1,
    "msg": "Key: 'Member.Name' Error:Field validation for 'Name' failed on the 'required' tag\nKey: 'Member.Age' Error:Field validation for 'Age' failed on the 'required' tag",
    "data": null
}

访问:http://localhost:8080/v1/member/add?name=1

{
    "code": -1,
    "msg": "Key: 'Member.Age' E  
		
编程开发网
首页 上一页 1 2 下一页 尾页 1/2/2
】【打印繁体】【投稿】【收藏】 【推荐】【举报】【评论】 【关闭】 【返回顶部
上一篇[系列] Gin 框架 - 使用 logrus .. 下一篇[系列] Gin框架 - 自定义错误处理