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

来自:网络
时间:2023-01-01
阅读:
目录

1. cast是啥

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

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

go get github.com/spf13/cast

2. 两种API

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

cast.ToxxxE在返回转化后数值的同时,也会返回一个errorcast.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上。

返回顶部
顶部