Go 类型转化工具库cast函数详解

目录

• 1. cast是啥
• 2. 两种API
• 3. 源码分析

1. cast是啥

cast 是在Github上开源的工具库，就像他的名字一样，他为我们提供了非常便捷的类型转化的方法。

我们可以通过下面的地址拉取cast库：

go get github.com/spf13/cast

2. 两种API

cast库中为我们提供了两类常用的API：cast.Toxxx以及cast.ToxxxE（xxx是要转化成的数据类型）。

cast.ToxxxE在返回转化后数值的同时，也会返回一个error，cast.Toxxx底层就是对cast.ToxxxE做了封装，并且抛弃了error的返回值。

cast库为我们实现了丰富且直观的类型转化的封装，以cast.ToString()为例：

   cast.ToString("mayonegg")         // "mayonegg"
   cast.ToString(8)                  // "8"
   cast.ToString(8.31)               // "8.31"
   cast.ToString([]byte("one time")) // "one time"
   cast.ToString(nil)                // ""

   var foo interface{} = "one more time"
   cast.ToString(foo)                // "one more time"

我们可以直观的观察到，cast.ToString()考虑了很多直观的转化情况，以帮助我们由各种数据类型，得到String。

对于cast.ToInt()也是如此：

   cast.ToInt(8)                  // 8
   cast.ToInt(8.31)               // 8
   cast.ToInt("8")                // 8
   cast.ToInt(true)               // 1
   cast.ToInt(false)              // 0

   var eight interface{} = 8
   cast.ToInt(eight)              // 8
   cast.ToInt(nil)                // 0

cast.ToInt()甚至帮我们考虑了eight转为8的情况。

除了基本数据类型，cast库也为我们提供了类似于ToStringMapString的容器数据类型的转化方法。

3. 源码分析

在这一部分，我们以两组源码为例，分析cast底层的实现方式：

• ToStringMapE

func ToStringMapE(i interface{}) (map[string]interface{}, error) {
   var m = map[string]interface{}{}

   switch v := i.(type) {
   case map[interface{}]interface{}:
      for k, val := range v {
         m[ToString(k)] = val
      }
      return m, nil
   case map[string]interface{}:
      return v, nil
   case string:
      err := jsonStringToObject(v, &m)
      return m, err
   default:
      return m, fmt.Errorf("unable to cast %#v of type %T to map[string]interface{}", i, i)
   }
}

我们可以看到，cast的底层还是靠类型断言+switch来实现的，对于需要转化为map[string]interface{}的需求，借助了ToString()方法进行类型转化以及jsonStringToObject()方法进行json解析，ToString()进行类型转化的思想大同小异，知识考虑了更多的情况，这里就不做重复展示了，接下来我们再针对jsonStringToObject()方法做简单分析。

• ToString()

func jsonStringToObject(s string, v interface{}) error {
   data := []byte(s)
   return json.Unmarshal(data, v)
}

ToString()方法底层调用了Go自带的json.Unmarshal()方法，进行json字符串的解析，将解析出的map绑定到m上。