Go 实用小技巧
2019-12-13 本文已影响0人
昵称不用太拉风
本文持续产出、收录一些笔者觉得好用的 Go 语法小技巧,还有一些初学 Go 遇到的“坑”,帮助节约日常开发时间,更多的时间可以做自己喜欢做的事情。
记录函数耗时
由于 Go 函数支持 defer
和闭包两大强大的特性,所以我们可以使用一行语句即可实现统计函数耗时的功能
// 函数主体
func TimeConsuming(tag string) func() {
now := time.Now().UnixNano() / 1000000
return func() {
after := time.Now().UnixNano() / 1000000
fmt.Printf("%q time cost %d ms\n", tag, after-now)
}
}
// 你需要统计耗时的函数
func main() {
// 函数首部调用即可,不影响函数整体美观
defer TimeConsuming("main")()
// ...
}
Switch 的基本使用
// 在 Go 中输出 one,如果是 Java 则输出 one two three
// 在 Go 中匹配到匹配项后会自动跳出 switch,不需要在每一条 case 下写 break
func main() {
i := 1
fmt.Println("Write ", i, " as ")
switch i {
case 1:
fmt.Println("one")
case 2:
fmt.Println("two")
case 3:
fmt.Println("three")
}
}
定义二维 Slice
twoD := make([][]int, 3)
for i := 0; i < 3; i++ {
innerLen := i + 1
twoD[i] = make([]int, innerLen)
for j := 0; j < innerLen; j++ {
twoD[i][j] = i + j
}
}
获取时间戳
now := time.Now()
// 时间戳:秒
secs := now.Unix()
// 时间戳:纳秒
nanos := now.UnixNano()
// 时间戳:毫秒
millis := nanos / 1000000
控制并发数量(防止某些情况下宕机)
// 20 就是允许同时运行的最大 goroutine 数量
var sema = make(chan struct{}, 20)
func dirents(dir string) []os.FileInfo {
// 控制语句写在函数的开头位置即可
sema <- struct{}{}
defer func() { <-sema }()
entries, err := ioutil.ReadDir(dir)
if err != nil {
fmt.Fprintf(os.Stderr, "du1: %v\n", err)
return nil
}
return entries
}
定义包级别的枚举字段
type Operation int
const (
Add Operation = iota + 1
Subtract
Multiply
)
大家平时有遇到什么比较多的问题或者比较实用的技巧可以贴出来,本文持续解答和收录。
QQ 学习交流群:646026429