Go 第三方库推荐:类型转换如此简单
大家好,我是 polarisxu。
强类型语言有它的优势,但也有不便利的地方,最典型的就是类型转换。Golang 作为一门强类型语言,而且不支持隐式类型转换,因此这个问题更突出。虽然 Go 提供了不少方式进行类型转换,包括相关的标准库,比如 strconv 包。
然而,strconv 包使用没那么方便,比如 "8"
转为 int 类型:
s := "8"
i, err := strconv.Atoi(s)
你必须对 err 进行处理,因为返回两个值,没法直接将结果传递给接收 int 参数的函数,使用不方便。
今天给大家介绍一个第三方库,专门处理类型转换的问题。
01 为什么需要类型转换
有一些场景会需要使用类型转换:
- 从 yaml、toml、json 等配置文件中读取数据;
- 从网络接收请求数据;
- 其他通过 interface{} 处理数据的情况;
- 。。。
转换为正确的类型,能充分利用强类型的好处,让程序更健壮、更安全。
02 spf13/cast
第三方包 github.com/spf13/cast 专门解决类型转换的问题,这个包产生于 hugo。当时主要用于处理 yaml 等配置文件数据的转换。该包不会 panic。
该包目前有 1.6k+ 的 Star,有超过 4000 多个开源项目使用了该包。
这个包使用很简单,主要有两套函数:
1)To_ 形式函数
这些函数始终返回所需的类型。如果无法正确转换为对应的类型,则返回目标类型的零值。
支持的类型包括所有的基本数据类型,还支持 time.Time、time.Duration、slice、map 等常用类型。
比如:
cast.ToString("mayonegg") // "mayonegg"
cast.ToString(8) // "8"
cast.ToString(8.31) // "8.31"
cast.ToString([]byte("one time")) // "one time"
cast.ToString(nil) // ""
cast.ToTime("2021-08-10 22:00:00") // 2021-08-10 22:00:00 +0000 UTC
注意,转换为 time.Time 时,需要注意时区问题。ToTime 默认使用 UTC,如果想用其他时区,得类似这么做:
secondsEastOfUTC := int((8 * time.Hour).Seconds())
beijing := time.FixedZone("Beijing Time", secondsEastOfUTC)
fmt.Println(cast.ToTimeInDefaultLocation("2021-08-10 22:00:00", beijing))
当然,你也可以这样:
fmt.Println(cast.ToTimeInDefaultLocation("2021-08-10 22:00:00", time.Local))
不过,Local 表示本地时区,要明确这个本地是不是你想要的。
2)To_E 形式函数
E 表示 error,也就是说,这一系列函数会返回 error。在无法进行类型转换时,会将错误原因返回。To_ 形式内部调用的是 To_E 形似,只是它忽略了错误。
这种形式就不举例了。一般地,除非你需要区分零值是因为出错导致的还是本身就是零值,否则应该使用 To_ 系列函数,毕竟更省事。
03 总结
大概率,不少公司都有自己类似的库。如果没有,可以考虑使用该库,这样的轮子,没太多必要造。不过这个库有一点我不太喜欢,就是没法指定默认值。
比如,我想在转换失败时,返回我的默认值,而不是默认零值,这个包做不到。常见的场景就是,处理用户可选输入,如果用户没输入,给一个默认值。
配置文件也有这样的场景,比如某个配置项如果没有配置,我希望硬编码一个默认值。因为 cast 不支持,依赖 cast 的 github.com/spf13/viper 库也不支持默认值,导致我会写出这样的繁琐代码:
viper.SetDefault("listen.port", "2021")
port := viper.GetString("listen.port")
我更希望的是这样的代码:
port := viper.GetString("listen.port", "2021") // listen.port 没设置时,返回 2021