在開發(fā)生產(chǎn)項(xiàng)目的過程中，我注意到經(jīng)常會(huì)發(fā)現(xiàn)自己在重復(fù)編寫代碼，使用某些技巧時(shí)沒有意識(shí)到，直到后來回顧工作時(shí)才意識(shí)到。

為了解決這個(gè)問題，我開發(fā)了一種解決方案，對(duì)我來說非常有幫助，我覺得對(duì)其他人也可能有用。

以下是一些從我的實(shí)用程序庫中隨機(jī)挑選的有用且多功能的代碼片段，沒有特定的分類或特定于系統(tǒng)的技巧。

1. 追蹤執(zhí)行時(shí)間的技巧

如果你想追蹤 Go 中函數(shù)的執(zhí)行時(shí)間，有一個(gè)簡(jiǎn)單高效的技巧可以用一行代碼實(shí)現(xiàn)，使用 defer 關(guān)鍵字即可。你只需要一個(gè) TrackTime 函數(shù)：

// Utility
func TrackTime(pre time.Time) time.Duration {
  elapsed := time.Since(pre)
  fmt.Println("elapsed:", elapsed)

  return elapsed
}

func TestTrackTime(t *testing.T) {
  defer TrackTime(time.Now()) // <--- THIS

  time.Sleep(500 * time.Millisecond)
}

// 輸出：
// elapsed: 501.11125ms

1.5. 兩階段延遲執(zhí)行

Go 的 defer 不僅僅是用于清理任務(wù)，還可以用于準(zhǔn)備任務(wù)，考慮以下示例：

func setupTeardown() func() {
    fmt.Println("Run initialization")
    return func() {
        fmt.Println("Run cleanup")
    }
}

func main() {
    defer setupTeardown()() // <--------
    fmt.Println("Main function called")
}

// 輸出：
// Run initialization
// Main function called
// Run cleanup

這種模式的美妙之處在于，只需一行代碼，你就可以完成諸如以下任務(wù)：

• 打開數(shù)據(jù)庫連接，然后關(guān)閉它。
• 設(shè)置模擬環(huán)境，然后拆除它。
• 獲取分布式鎖，然后釋放它。
• ...

"嗯，這似乎很聰明，但它在現(xiàn)實(shí)中有什么用處呢？"

還記得追蹤執(zhí)行時(shí)間的技巧嗎？我們也可以這樣做：

func TrackTime() func() {
  pre := time.Now()
  return func() {
    elapsed := time.Since(pre)
    fmt.Println("elapsed:", elapsed)
  }
}

func main() {
  defer TrackTime()()

  time.Sleep(500 * time.Millisecond)
}

注意！如果我連接到數(shù)據(jù)庫時(shí)出現(xiàn)錯(cuò)誤怎么辦？

確實(shí)，像 defer TrackTime() 或 defer ConnectDB() 這樣的模式不會(huì)妥善處理錯(cuò)誤。這種技巧最適合用于測(cè)試或者當(dāng)你愿意冒著致命錯(cuò)誤的風(fēng)險(xiǎn)時(shí)使用，參考下面這種面向測(cè)試的方法：

func TestSomething(t *testing.T) {
  defer handleDBConnection(t)()
  // ...
}

func handleDBConnection(t *testing.T) func() {
  conn, err := connectDB()
  if err != nil {
    t.Fatal(err)
  }

  return func() {
    fmt.Println("Closing connection", conn)
  }
}

這樣，在測(cè)試期間可以處理數(shù)據(jù)庫連接的錯(cuò)誤。

2. 預(yù)分配切片

根據(jù)文章《Go 性能提升技巧》中的見解，預(yù)分配切片或映射可以顯著提高 Go 程序的性能。

但是值得注意的是，如果我們不小心使用 append 而不是索引（如 a[i]），這種方法有時(shí)可能導(dǎo)致錯(cuò)誤。你知道嗎，我們可以在不指定數(shù)組長(zhǎng)度（為零）的情況下使用預(yù)分配的切片，就像在上述文章中解釋的那樣？這使我們可以像使用 append 一樣使用預(yù)分配的切片：

// 與其
a := make([]int, 10)
a[0] = 1

// 不如這樣使用
b := make([]int, 0, 10)
b = append(b, 1)

3. 鏈?zhǔn)秸{(diào)用

鏈?zhǔn)秸{(diào)用技術(shù)可以應(yīng)用于函數(shù)（指針）接收器。為了說明這一點(diǎn)，讓我們考慮一個(gè) Person 結(jié)構(gòu)，它有兩個(gè)函數(shù) AddAge 和 Rename，用于對(duì)其進(jìn)行修改。

type Person struct {
  Name string
  Age  int
}

func (p *Person) AddAge() {
  p.Age++
}

func (p *Person) Rename(name string) {
  p.Name = name
}

如果你想給一個(gè)人增加年齡然后給他們改名字，常規(guī)的方法是：

func main() {
  p := Person{Name: "Aiden", Age: 30}

  p.AddAge()
  p.Rename("Aiden 2")
}

或者，我們可以修改 AddAge 和 Rename 函數(shù)接收器，使其返回修改后的對(duì)象本身，即使它們通常不返回任何內(nèi)容。

func (p *Person) AddAge() *Person {
  p.Age++
  return p
}

func (p *Person) Rename(name string) *Person {
  p.Name = name
  return p
}

通過返回修改后的對(duì)象本身，我們可以輕松地將多個(gè)函數(shù)接收器鏈在一起，而無需添加不必要的代碼行：

p = p.AddAge().Rename("Aiden 2")

4. Go 1.20 允許將切片解析為數(shù)組或數(shù)組指針

當(dāng)我們需要將切片轉(zhuǎn)換為固定大小的數(shù)組時(shí)，不能直接賦值，例如：

a := []int{0, 1, 2, 3, 4, 5}
var b [3]int = a[0:3]

// 在變量聲明中不能將 a[0:3]（類型為 []int 的值）賦值給 [3]int 類型的變量
// （不兼容的賦值）

為了將切片轉(zhuǎn)換為數(shù)組，Go 團(tuán)隊(duì)在 Go 1.17 中更新了這個(gè)特性。隨著 Go 1.20 的發(fā)布，借助更方便的字面量，轉(zhuǎn)換過程變得更加簡(jiǎn)單：

// Go 1.20
func Test(t *testing.T) {
   a := []int{0, 1, 2, 3, 4, 5}
   b := [3]int(a[0:3])

  fmt.Println(b) // [0 1 2]
}

// Go 1.17
func TestM2e(t *testing.T) {
  a := []int{0, 1, 2, 3, 4, 5}
  b := *(*[3]int)(a[0:3])

  fmt.Println(b) // [0 1 2]
}

只是一個(gè)快速提醒：你可以使用 a[:3] 替代 a[0:3]。我提到這一點(diǎn)是為了更清晰地說明。

5. 使用 "import _" 進(jìn)行包初始化

有時(shí)，在庫中，你可能會(huì)遇到結(jié)合下劃線 (_) 的導(dǎo)入語句，如下所示：

import (
  _ "google.golang.org/genproto/googleapis/api/annotations"
)

這將執(zhí)行包的初始化代碼（init 函數(shù)），而無需為其創(chuàng)建名稱引用。這允許你在運(yùn)行代碼之前初始化包、注冊(cè)連接和執(zhí)行其他任務(wù)。

讓我們通過一個(gè)示例來更好地理解它的工作原理：

// 下劃線
package underscore

func init() {
  fmt.Println("init called from underscore package")
}
// main
package main

import (
  _ "lab/underscore"
)

func main() {}
// 輸出：init called from underscore package

6. 使用 "import ." 進(jìn)行導(dǎo)入

在了解了如何使用下劃線進(jìn)行導(dǎo)入后，讓我們看看如何更常見地使用點(diǎn) (.) 運(yùn)算符。

作為開發(fā)者，點(diǎn) (.) 運(yùn)算符可用于在不必指定包名的情況下使用導(dǎo)入包的導(dǎo)出標(biāo)識(shí)符，這對(duì)于懶惰的開發(fā)者來說是一個(gè)有用的快捷方式。

很酷，對(duì)吧？這在處理項(xiàng)目中的長(zhǎng)包名時(shí)特別有用，比如 externalmodel 或 doingsomethinglonglib。

為了演示，這里有一個(gè)簡(jiǎn)單的例子：

package main

import (
  "fmt"
  . "math"
)

func main() {
  fmt.Println(Pi) // 3.141592653589793
  fmt.Println(Sin(Pi / 2)) // 1
}

7. Go 1.20 允許將多個(gè)錯(cuò)誤合并為單個(gè)錯(cuò)誤

Go 1.20 引入了對(duì)錯(cuò)誤包的新功能，包括對(duì)多個(gè)錯(cuò)誤的支持以及對(duì) errors.Is 和 errors.As 的更改。

在 errors 中添加的一個(gè)新函數(shù)是 Join，我們將在下面詳細(xì)討論它：

var (
  err1 = errors.New("Error 1st")
  err2 = errors.New("Error 2nd")
)

func main() {
  err := err1
  err = errors.Join(err, err2)

  fmt.Println(errors.Is(err, err1)) // true
  fmt.Println(errors.Is(err, err2)) // true
}

如果有多個(gè)任務(wù)導(dǎo)致錯(cuò)誤，你可以使用 Join 函數(shù)而不是手動(dòng)管理數(shù)組。這簡(jiǎn)化了錯(cuò)誤處理過程。

8. 檢查接口是否為真正的 nil

即使接口持有的值為 nil，也不意味著接口本身為 nil。這可能導(dǎo)致 Go 程序中的意外錯(cuò)誤。因此，重要的是要知道如何檢查接口是否為真正的 nil。

func main() {
  var x interface{}
  var y *int = nil
  x = y

  if x != nil {
    fmt.Println("x != nil") // <-- 實(shí)際輸出
  } else {
    fmt.Println("x == nil")
  }

  fmt.Println(x)
}

// 輸出：
// x != nil
// <nil>

我們?nèi)绾未_定 interface{} 值是否為 nil 呢？幸運(yùn)的是，有一個(gè)簡(jiǎn)單的工具可以幫助我們實(shí)現(xiàn)這一點(diǎn)：

func IsNil(x interface{}) bool {
  if x == nil {
    return true
  }

  return reflect.ValueOf(x).IsNil()
}

9. 在 JSON 中解析 time.Duration

當(dāng)解析 JSON 時(shí)，使用 time.Duration 可能是一個(gè)繁瑣的過程，因?yàn)樗枰谝幻氲暮竺嫣砑?9 個(gè)零（即 1000000000）。為了簡(jiǎn)化這個(gè)過程，我創(chuàng)建了一個(gè)名為 Duration 的新類型：

type Duration time.Duration

為了將字符串（如 "1s" 或 "20h5m"）解析為 int64 類型的持續(xù)時(shí)間，我還為這個(gè)新類型實(shí)現(xiàn)了自定義的解析邏輯：

func (d *Duration) UnmarshalJSON(b []byte) error {
  var s string
  if err := json.Unmarshal(b, &s); err != nil {
    return err
  }
  dur, err := time.ParseDuration(s)
  if err != nil {
    return err
  }
  *d = Duration(dur)
  return nil
}

但是，需要注意的是，變量 'd' 不應(yīng)為 nil，否則可能會(huì)導(dǎo)致編組錯(cuò)誤?；蛘?，你還可以在函數(shù)開頭對(duì) 'd' 進(jìn)行檢查。

10. 避免裸參數(shù)

當(dāng)處理具有多個(gè)參數(shù)的函數(shù)時(shí)，僅通過閱讀其用法來理解每個(gè)參數(shù)的含義可能會(huì)令人困惑。考慮以下示例：

printInfo("foo", true, true)

如果不檢查 printInfo 函數(shù)，那么第一個(gè) 'true' 和第二個(gè) 'true' 的含義是什么呢？當(dāng)你有一個(gè)具有多個(gè)參數(shù)的函數(shù)時(shí)，僅通過閱讀其用法來理解參數(shù)的含義可能會(huì)令人困惑。

但是，我們可以使用注釋使代碼更易讀。例如：

// func printInfo(name string, isLocal, done bool)

printInfo("foo", true /* isLocal */, true /* done */)

有些 IDE 也支持這個(gè)功能，可以在函數(shù)調(diào)用建議中顯示注釋，但可能需要在設(shè)置中啟用。

以上是我分享的一些實(shí)用技巧，但我不想讓文章過長(zhǎng)，難以跟進(jìn)，因?yàn)檫@些技巧與特定主題無關(guān)，涵蓋了各種類別。

如果你覺得這些技巧有用，或有自己的見解要分享，請(qǐng)隨時(shí)留言。我重視你的反饋，并樂于在回應(yīng)此文章時(shí)點(diǎn)贊或推薦你的想法。