Go 实用小技巧

本文持续产出、收录一些笔者觉得好用的 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

原文地址，建议收录