Golang JSON的进阶用法实例讲解

2019-11-10 12:12:06于海丽

痛点

json 是当前最常用的数据传输格式之一,纯文本,容易使用,方便阅读,在通信过程中大量被使用。 

你是否遇到过json中某个字段填入某种类型都适合而陷入两难境地? (例如:定义了一个port字段,你却不知道是填入 8080 ,还是 “8080” 的尴尬局面)

你是否遇到过json反解析报错是因为填入字段的类型不匹配导致的?例如:

json: cannot unmarshal number into Go struct field Host.port of type string

你是否有json某字段兼容2种或者多种的数据结构的需求?

你是否想让程序更优雅,更具有适配性,而不在被这些小细节头痛?

如果你有或者你想,获取你可以看看这篇文章。

重现问题

我们给了用户一个json如下:

{
 "name":"yulibaozi",
 "port":8080
}

但是,业务方却误填了”8080”,结果我们程序反解析报错,导致业务失败。

json: cannot unmarshal number into Go struct field Host.port of type string

或许你认为这是业务方的问题,但我认为我们可以更优雅的解决这个问题。

如何解决问题

我们先定义了一个结构体

type Host struct {
 Name string `json:"name"`
 Port Port `json:"port"`
}

心细的你会发现,Port既不是int也不是string类型,而是Port类型,而Port类型是:

type Type int

const (
 Int Type = iota
 String
)

type Port struct {
 Type Type
 IntVal int
 StrVal string
}

在Port结构体中,我们发现了Type类型, 而Type类型包括了int,string两种类型。接下来就非常重要了,我们需要实现以下这两个接口。

json.Unmarshaller interface
json.Marshaller interface

实现代码如下:

type Port struct {
 Type Type
 IntVal int
 StrVal string
}

// 实现 json.Unmarshaller 接口
func (port *Port) UnmarshalJSON(value []byte) error {
 if value[0] == '"' {
  port.Type = String
  return json.Unmarshal(value, &port.StrVal)
 }
 port.Type = Int
 return json.Unmarshal(value, &port.IntVal)
}

// 实现 json.Marshaller 接口
func (port Port) MarshalJSON() ([]byte, error) {
 switch port.Type {
 case Int:
  return json.Marshal(port.IntVal)
 case String:
  return json.Marshal(port.StrVal)
 default:
  return []byte{}, fmt.Errorf("impossible Port.Type")
 }
}

接下来测试:

测试反解析

测试反解析int

给出json数据:

{"name":"yulibaozi","port":8090}

反解析得到的结构体数据如下:

&{Name:yulibaozi Port:{Type:0 IntVal:8090 StrVal:}}

测试反解析string:

给出json数据:

{"name":"yulibaozi","port":"8090"}

反解析得到的结构体数据如下: