Go語言類型轉換工具庫cast的實現(xiàn)
簡介
cast可以在 Go 中輕松安全地從一種類型轉換為另一種類型,cast 提供了簡單的函數(shù)來輕松地將數(shù)字轉換為字符串,將接口轉換為布爾值等。當需要顯示類型轉換時,cast 會智能地執(zhí)行轉換操作。
快速入門
安裝
go get github.com/spf13/cast
使用
package main
import (
"fmt"
"github.com/spf13/cast"
)
func main() {
// ToString
fmt.Println(cast.ToString("leedarjun")) // leedarjun
fmt.Println(cast.ToString(8)) // 8
fmt.Println(cast.ToString(8.31)) // 8.31
fmt.Println(cast.ToString([]byte("one time"))) // one time
fmt.Println(cast.ToString(nil)) // ""
var foo interface{} = "one more time"
fmt.Println(cast.ToString(foo)) // one more time
// ToInt
fmt.Println(cast.ToInt(8)) // 8
fmt.Println(cast.ToInt(8.31)) // 8
fmt.Println(cast.ToInt("8")) // 8
fmt.Println(cast.ToInt(true)) // 1
fmt.Println(cast.ToInt(false)) // 0
var eight interface{} = 8
fmt.Println(cast.ToInt(eight)) // 8
fmt.Println(cast.ToInt(nil)) // 0
}
實際上,cast實現(xiàn)了多種常見類型之間的相互轉換,返回最符合直覺的結果。例如:
nil轉為string的結果為"",而不是"nil";true轉為string的結果為"true",而true轉為int的結果為1;interface{}轉為其他類型,要看它里面存儲的值類型。
這些類型包括所有的基本類型(整形、浮點型、布爾值和字符串)、空接口、nil,時間(time.Time)、時長(time.Duration)以及它們的切片類型,還有map[string]Type(其中Type為前面提到的類型):
byte bool float32 float64 string
int8 int16 int32 int64 int
uint8 uint16 uint32 uint64 uint
interface{} time.Time time.Duration nil
高級轉換
cast提供了兩組函數(shù):
ToType(其中Type可以為任何支持的類型),將參數(shù)轉換為Type類型。如果無法轉換,返回Type類型的零值或nil;ToTypeE以 E 結尾,返回轉換后的值和一個error。這組函數(shù)可以區(qū)分參數(shù)中實際存儲了零值,還是轉換失敗了。
實現(xiàn)上大部分代碼都類似,ToType在內部調用ToTypeE函數(shù),返回結果并忽略錯誤。ToType函數(shù)的實現(xiàn)在文件cast.go中,而ToTypeE函數(shù)的實現(xiàn)在文件caste.go中。
部分源碼如下:
// ToBoolE casts an interface to a bool type.
func ToBool(i interface{}) bool {
v, _ := ToBoolE(i)
return v
}
// ToDuration casts an interface to a time.Duration type.
func ToDuration(i interface{}) time.Duration {
v, _ := ToDurationE(i)
return v
}
ToTypeE函數(shù)都接受任意類型的參數(shù)(interface{}),然后使用類型斷言根據(jù)具體的類型來執(zhí)行不同的轉換。如果無法轉換,返回錯誤。
// ToBoolE casts an interface to a bool type.
func ToBoolE(i interface{}) (bool, error) {
i = indirect(i)
switch b := i.(type) {
case bool:
return b, nil
case nil:
return false, nil
case int:
if i.(int) != 0 {
return true, nil
}
return false, nil
case string:
return strconv.ParseBool(i.(string))
case json.Number:
v, err := ToInt64E(b)
if err == nil {
return v != 0, nil
}
return false, fmt.Errorf("unable to cast %#v of type %T to bool", i, i)
default:
return false, fmt.Errorf("unable to cast %#v of type %T to bool", i, i)
}
}
首先調用indirect函數(shù)將參數(shù)中可能的指針去掉。如果類型本身不是指針,那么直接返回。否則返回指針指向的值。循環(huán)直到返回一個非指針的值:
// From html/template/content.go
// Copyright 2011 The Go Authors. All rights reserved.
// indirect returns the value, after dereferencing as many times
// as necessary to reach the base type (or nil).
func indirect(a interface{}) interface{} {
if a == nil {
return nil
}
if t := reflect.TypeOf(a); t.Kind() != reflect.Ptr {
// Avoid creating a reflect.Value if it's not a pointer.
return a
}
v := reflect.ValueOf(a)
for v.Kind() == reflect.Ptr && !v.IsNil() {
v = v.Elem()
}
return v.Interface()
}
所以,下面代碼輸出都是 8:
package main
import (
"fmt"
"github.com/spf13/cast"
)
func main() {
p := new(int)
*p = 8
fmt.Println(cast.ToInt(p)) // 8
pp := &p
fmt.Println(cast.ToInt(pp)) // 8
}
時間和時長轉換
時間類型的轉換
源碼如下:
// ToTimeE casts an interface to a time.Time type.
func ToTimeE(i interface{}) (tim time.Time, err error) {
return ToTimeInDefaultLocationE(i, time.UTC)
}
// ToTimeInDefaultLocationE casts an empty interface to time.Time,
// interpreting inputs without a timezone to be in the given location,
// or the local timezone if nil.
func ToTimeInDefaultLocationE(i interface{}, location *time.Location) (tim time.Time, err error) {
i = indirect(i)
switch v := i.(type) {
case time.Time:
return v, nil
case string:
return StringToDateInDefaultLocation(v, location)
case json.Number:
s, err1 := ToInt64E(v)
if err1 != nil {
return time.Time{}, fmt.Errorf("unable to cast %#v of type %T to Time", i, i)
}
return time.Unix(s, 0), nil
case int:
return time.Unix(int64(v), 0), nil
case int64:
return time.Unix(v, 0), nil
case int32:
return time.Unix(int64(v), 0), nil
case uint:
return time.Unix(int64(v), 0), nil
case uint64:
return time.Unix(int64(v), 0), nil
case uint32:
return time.Unix(int64(v), 0), nil
default:
return time.Time{}, fmt.Errorf("unable to cast %#v of type %T to Time", i, i)
}
}
根據(jù)傳入的類型執(zhí)行不同的處理:
- 如果是
time.Time,直接返回; - 如果是整型,將參數(shù)作為時間戳(自
UTC時間1970.01.01 00:00:00到現(xiàn)在的秒數(shù))調用time.Unix生成時間。Unix接受兩個參數(shù),第一個參數(shù)指定秒,第二個參數(shù)指定納秒; - 如果是字符串,調用
StringToDate函數(shù)依次嘗試以下面這些時間格式調用time.Parse解析該字符串。如果某個格式解析成功,則返回獲得的time.Time。否則解析失敗,返回錯誤; - 其他任何類型都無法轉換為
time.Time。
字符串轉換為時間
源碼:
// cast/caste.go
func StringToDate(s string) (time.Time, error) {
return parseDateWith(s, []string{
time.RFC3339,
"2006-01-02T15:04:05", // iso8601 without timezone
time.RFC1123Z,
time.RFC1123,
time.RFC822Z,
time.RFC822,
time.RFC850,
time.ANSIC,
time.UnixDate,
time.RubyDate,
"2006-01-02 15:04:05.999999999 -0700 MST", // Time.String()
"2006-01-02",
"02 Jan 2006",
"2006-01-02T15:04:05-0700", // RFC3339 without timezone hh:mm colon
"2006-01-02 15:04:05 -07:00",
"2006-01-02 15:04:05 -0700",
"2006-01-02 15:04:05Z07:00", // RFC3339 without T
"2006-01-02 15:04:05Z0700", // RFC3339 without T or timezone hh:mm colon
"2006-01-02 15:04:05",
time.Kitchen,
time.Stamp,
time.StampMilli,
time.StampMicro,
time.StampNano,
})
}
func parseDateWith(s string, dates []string) (d time.Time, e error) {
for _, dateType := range dates {
if d, e = time.Parse(dateType, s); e == nil {
return
}
}
return d, fmt.Errorf("unable to parse date: %s", s)
}
時長類型的轉換
源碼如下:
// cast/caste.go
func ToDurationE(i interface{}) (d time.Duration, err error) {
i = indirect(i)
switch s := i.(type) {
case time.Duration:
return s, nil
case int, int64, int32, int16, int8, uint, uint64, uint32, uint16, uint8:
d = time.Duration(ToInt64(s))
return
case float32, float64:
d = time.Duration(ToFloat64(s))
return
case string:
if strings.ContainsAny(s, "nsuμmh") {
d, err = time.ParseDuration(s)
} else {
d, err = time.ParseDuration(s "ns")
}
return
default:
err = fmt.Errorf("unable to cast %#v of type %T to Duration", i, i)
return
}
}
根據(jù)傳入的類型進行不同的處理:
- 如果是
time.Duration類型,直接返回; - 如果是整型或浮點型,將其數(shù)值強制轉換為
time.Duration類型,單位默認為ns; - 如果是字符串,分為兩種情況:如果字符串中有時間單位符號
nsuµmh,直接調用time.ParseDuration解析;否則在字符串后拼接ns再調用time.ParseDuration解析; - 其他類型解析失敗。
示例:
package main
import (
"fmt"
"time"
"github.com/spf13/cast"
)
func main() {
now := time.Now()
timestamp := 1579615973
timeStr := "2020-01-21 22:13:48"
fmt.Println(cast.ToTime(now)) // 2020-01-22 06:31:50.5068465 0800 CST m= 0.000997701
fmt.Println(cast.ToTime(timestamp)) // 2020-01-21 22:12:53 0800 CST
fmt.Println(cast.ToTime(timeStr)) // 2020-01-21 22:13:48 0000 UTC
d, _ := time.ParseDuration("1m30s")
ns := 30000
strWithUnit := "130s"
strWithoutUnit := "130"
fmt.Println(cast.ToDuration(d)) // 1m30s
fmt.Println(cast.ToDuration(ns)) // 30μs
fmt.Println(cast.ToDuration(strWithUnit)) // 2m10s
fmt.Println(cast.ToDuration(strWithoutUnit)) // 130ns
}
轉換為切片
實際上,這些函數(shù)的實現(xiàn)基本類似。使用類型斷言判斷類型。如果就是要返回的類型,直接返回。否則根據(jù)類型進行相應的轉換。
我們主要分析兩個實現(xiàn):ToIntSliceE和ToStringSliceE。ToBoolSliceE/ToDurationSliceE與ToIntSliceE基本相同。
ToIntSliceE
源碼如下:
func ToIntSliceE(i interface{}) ([]int, error) {
if i == nil {
return []int{}, fmt.Errorf("unable to cast %#v of type %T to []int", i, i)
}
switch v := i.(type) {
case []int:
return v, nil
}
kind := reflect.TypeOf(i).Kind()
switch kind {
case reflect.Slice, reflect.Array:
s := reflect.ValueOf(i)
a := make([]int, s.Len())
for j := 0; j < s.Len(); j {
val, err := ToIntE(s.Index(j).Interface())
if err != nil {
return []int{}, fmt.Errorf("unable to cast %#v of type %T to []int", i, i)
}
a[j] = val
}
return a, nil
default:
return []int{}, fmt.Errorf("unable to cast %#v of type %T to []int", i, i)
}
}
根據(jù)傳入?yún)?shù)的類型:
- 如果是
nil,直接返回錯誤; - 如果是
[]int,不用轉換,直接返回; - 如果傳入類型為切片或數(shù)組,新建一個
[]int,將切片或數(shù)組中的每個元素轉為int放到該[]int中。最后返回這個[]int; - 其他情況,不能轉換。
ToStringSliceE
源碼如下:
func ToStringSliceE(i interface{}) ([]string, error) {
var a []string
switch v := i.(type) {
case []interface{}:
for _, u := range v {
a = append(a, ToString(u))
}
return a, nil
case []string:
return v, nil
case string:
return strings.Fields(v), nil
case interface{}:
str, err := ToStringE(v)
if err != nil {
return a, fmt.Errorf("unable to cast %#v of type %T to []string", i, i)
}
return []string{str}, nil
default:
return a, fmt.Errorf("unable to cast %#v of type %T to []string", i, i)
}
}
根據(jù)傳入的參數(shù)類型:
- 如果是
[]interface{},將該參數(shù)中每個元素轉為string,返回結果切片; - 如果是
[]string,不需要轉換,直接返回; - 如果是
interface{},將參數(shù)轉為string,返回只包含這個值的切片; - 如果是
string,調用strings.Fields函數(shù)按空白符將參數(shù)拆分,返回拆分后的字符串切片; - 其他情況,不能轉換。
示例:
package main
import (
"fmt"
"github.com/spf13/cast"
)
func main() {
sliceOfInt := []int{1, 3, 7}
arrayOfInt := [3]int{8, 12}
// ToIntSlice
fmt.Println(cast.ToIntSlice(sliceOfInt)) // [1 3 7]
fmt.Println(cast.ToIntSlice(arrayOfInt)) // [8 12 0]
sliceOfInterface := []interface{}{1, 2.0, "darjun"}
sliceOfString := []string{"abc", "dj", "pipi"}
stringFields := " abc def hij "
any := interface{}(37)
// ToStringSliceE
fmt.Println(cast.ToStringSlice(sliceOfInterface)) // [1 2 darjun]
fmt.Println(cast.ToStringSlice(sliceOfString)) // [abc dj pipi]
fmt.Println(cast.ToStringSlice(stringFields)) // [abc def hij]
fmt.Println(cast.ToStringSlice(any)) // [37]
}
轉為map[string]Type類型
cast庫能將傳入的參數(shù)轉為map[string]Type類型,Type為上面支持的類型。
其實只需要分析一個ToStringMapStringE函數(shù)就可以了,其他的實現(xiàn)基本一樣。ToStringMapStringE返回map[string]string類型的值。
源碼如下:
func ToStringMapStringE(i interface{}) (map[string]string, error) {
var m = map[string]string{}
switch v := i.(type) {
case map[string]string:
return v, nil
case map[string]interface{}:
for k, val := range v {
m[ToString(k)] = ToString(val)
}
return m, nil
case map[interface{}]string:
for k, val := range v {
m[ToString(k)] = ToString(val)
}
return m, nil
case map[interface{}]interface{}:
for k, val := range v {
m[ToString(k)] = ToString(val)
}
return m, 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]string", i, i)
}
}
根據(jù)傳入的參數(shù)類型:
- 如果是
map[string]string,不用轉換,直接返回; - 如果是
map[string]interface{},將每個值轉為string存入新的 map,最后返回新的 map; - 如果是
map[interface{}]string,將每個鍵轉為string存入新的 map,最后返回新的 map; - 如果是
map[interface{}]interface{},將每個鍵和值都轉為string存入新的 map,最后返回新的 map; - 如果是
string類型,cast將它看成一個 JSON 串,解析這個 JSON 到map[string]string,然后返回結果; - 其他情況,返回錯誤。
示例:
package main
import (
"fmt"
"github.com/spf13/cast"
)
func main() {
m1 := map[string]string {
"name": "darjun",
"job": "developer",
}
m2 := map[string]interface{} {
"name": "jingwen",
"age": 18,
}
m3 := map[interface{}]string {
"name": "pipi",
"job": "designer",
}
m4 := map[interface{}]interface{} {
"name": "did",
"age": 29,
}
jsonStr := `{"name":"bibi", "job":"manager"}`
fmt.Println(cast.ToStringMapString(m1)) // map[job:developer name:darjun]
fmt.Println(cast.ToStringMapString(m2)) // map[age:18 name:jingwen]
fmt.Println(cast.ToStringMapString(m3)) // map[job:designer name:pipi]
fmt.Println(cast.ToStringMapString(m4)) // map[job:designer name:pipi]
fmt.Println(cast.ToStringMapString(jsonStr)) // map[job:manager name:bibi]
}
到此這篇關于Go語言類型轉換工具庫cast的實現(xiàn)的文章就介紹到這了,更多相關Go語言類型轉換工具庫cast內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
使用golang生成prometheus格式數(shù)據(jù)
Prometheus是一個開源的監(jiān)控系統(tǒng),擁有許多Advanced?Feature,本文將介紹Primetheus?client的使用,并基于golang生成prometheus格式數(shù)據(jù),希望對大家有所幫助2025-02-02
Gin golang web開發(fā)模型綁定實現(xiàn)過程解析
這篇文章主要介紹了Gin golang web開發(fā)模型綁定實現(xiàn)過程解析,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-10-10

