Go Gin源码学习(三) 参数解析

2019年9月2日 195次阅读 来源: 大二小的宝

学习目标

在第一篇中看到了Gin提供了很多的获取和解析参数的方法:

// **** 输入数据
//从URL中拿值，比如 /user/:id => /user/john
Param(key string) string    

//从GET参数中拿值，比如 /path?id=john
GetQueryArray(key string) ([]string, bool)  
GetQuery(key string)(string, bool)
Query(key string) string
DefaultQuery(key, defaultValue string) string
GetQueryArray(key string) ([]string, bool)
QueryArray(key string) []string

//从POST中拿数据
GetPostFormArray(key string) ([]string, bool)
PostFormArray(key string) []string 
GetPostForm(key string) (string, bool)
PostForm(key string) string
DefaultPostForm(key, defaultValue string) string

// 文件
FormFile(name string) (*multipart.FileHeader, error)
MultipartForm() (*multipart.Form, error)
SaveUploadedFile(file *multipart.FileHeader, dst string) error

// 数据绑定
Bind(obj interface{}) error //根据Content-Type绑定数据
BindJSON(obj interface{}) error
BindQuery(obj interface{}) error

//--- Should ok, else return error
ShouldBindJSON(obj interface{}) error 
ShouldBind(obj interface{}) error
ShouldBindJSON(obj interface{}) error
ShouldBindQuery(obj interface{}) error

其中从url中获取 从get参数中获取 从post拿数据相信我们都可以想象的到，基本就是从request中的url或者body中获取数据然后返回
但是其中的数据绑定我自己开始是很疑惑的，到底是怎么实现的。疑惑的是如果object中我客户端少输入了参数 或者多输入的参数会是怎么样。举个例子：

type Login struct {
    User     string `json:"user"`
    Password string `json:"password"`
    Age      int
}

//伪代码
l := Login{}
context.Bind(&l)

如果客户端输入的是{“user”:”TAO”,”age”:10} 没有输入password 得到的对象是什么会不会报错

如果客户端输入的是{“user”:”TAO”,”age”:10,”class”:”class”}多输入了一个class的参数 结果是什么

如果客户端输入的是{} 这边会不会报错，会得到什么结果

这些都是需要看源码

源码

先看gin/binding/binding.go

//Binding 接口 定义了一个返回名字的函数
//还有一个Bind 函数接口 这个是关键函数 //从request中获取数据解析
//所有类型的处理对象都继承这个接口来实现多态
type Binding interface {
    Name() string
    Bind(*http.Request, interface{}) error
}

type BindingBody interface {
    Binding
    BindBody([]byte, interface{}) error
}

//在包中定义了8个对象 可以供给context直接使用
var (
    JSON          = jsonBinding{}
    XML           = xmlBinding{}
    Form          = formBinding{}
    Query         = queryBinding{}
    FormPost      = formPostBinding{}
    FormMultipart = formMultipartBinding{}
    ProtoBuf      = protobufBinding{}
    MsgPack       = msgpackBinding{}
)

接下来是继承Binding接口的对象，这样的对象有很多我们就举一个例子JSON，JSON是我们使用的最多的一个方式。
gin/binding/json.go

//type 是Gin处理json对象使用的类 继承Binding
type jsonBinding struct{}

func (jsonBinding) Name() string {
    return "json"
}
//调用decodeJSON方法处理
func (jsonBinding) Bind(req *http.Request, obj interface{}) error {
    return decodeJSON(req.Body, obj)
}

func (jsonBinding) BindBody(body []byte, obj interface{}) error {
    return decodeJSON(bytes.NewReader(body), obj)
}
// 可以看到，这个方法就是从request中获取body然后使用json包的decoder来转换成对象
func decodeJSON(r io.Reader, obj interface{}) error {
    decoder := json.NewDecoder(r)
    if EnableDecoderUseNumber {
        decoder.UseNumber()
    }
    if err := decoder.Decode(obj); err != nil {
        return err
    }
    return validate(obj)
}

最后再看一下context是如何调用的


//这个binding.JSON 就是上面binding中定义的那8个类其中的一个 这里可以直接的使用
func (c *Context) BindJSON(obj interface{}) error {
    return c.MustBindWith(obj, binding.JSON)
}
//这个方法可以接收Binding接口，在Gin中有7 8 个类都是继承了这个接口 都可以调用这个方法来执行解析 
func (c *Context) MustBindWith(obj interface{}, b binding.Binding) (err error) {
    if err = c.ShouldBindWith(obj, b); err != nil {
        c.AbortWithError(http.StatusBadRequest, err).SetType(ErrorTypeBind)
    }

    return
}
//这个方法就是调用上面一段代码定义的json的bind方法
func (c *Context) ShouldBindWith(obj interface{}, b binding.Binding) error {
    return b.Bind(c.Request, obj)
}

看了上面的代码，其实整个binding已经很清晰了。
最终，Gin就是调用了json包中的解析。那么我们上面的一系列问题就很轻松的解决了

  • 如果客户端输入的是{“user”:”TAO”,”age”:10} 没有输入password 结果{“Tao”,10}
  • 如果客户端输入的是{“user”:”TAO”,”age”:10,”class”:”class”}多输入了一个class的参数 结果{“Tao”,10} 如果在login类添加class 这里就会出现class
  • 如果客户端输入的是{} 这边会不会报错 结果{0} 只有age有一个默认值0 其余都是空字符串

这些行为跟json的处理行为是完全一致的。

简单模仿实现

老规矩，在了解了binding的主体流程之后，我们可以自己简单的实现其中几个功能 比如json和xml的解析。我们可以看到以下代码，可以很清晰的看到Gin的实现方式。

package mygin

import (
    "encoding/json"
    "encoding/xml"
    "net/http"
)

var (
    JSON = JsonBinding{}
    XML  = XmlBinding{}
)

//bind接口 方便多态
type Binding interface {
    //返回name
    Name() string
    //具体执行
    Bind(*http.Request, interface{}) error
}

type XmlBinding struct {
}

func (XmlBinding) Name() string {
    return "XmlBinding"
}

func (XmlBinding) Bind(r *http.Request, obj interface{}) error {
    //使用json包中的方法 把string转成对象
    decoder := xml.NewDecoder(r.Body)
    if err := decoder.Decode(obj); err != nil {
        return err
    }
    return nil
}

type JsonBinding struct {
}

func (JsonBinding) Name() string {
    return "jsonBinding"
}

func (JsonBinding) Bind(r *http.Request, obj interface{}) error {
    //使用json包中的方法 把string转成对象
    decoder := json.NewDecoder(r.Body)
    if err := decoder.Decode(obj); err != nil {
        return err
    }

    return nil
}
// context调用
func (c *Context) BindWith(obj interface{}, b Binding) error {
    //调用binding 的Bind方法
    if err := b.Bind(c.Request, obj); err != nil {
        return err
    }

    return nil
}

func (c *Context) BindJson(obj interface{}) error {
    return c.BindWith(obj, JSON)
}

代码比较简单，几乎不需要加注释就能很清楚的看懂。 配合上一篇文章的Gin主流程可以完全的跑起来。结果这里就不写了，上面都描述过了。

    原文作者：大二小的宝
    原文地址: https://segmentfault.com/a/1190000019130634
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞