腳本之家服務(wù)器常用軟件

快捷導(dǎo)航

詳解Go語言中init的使用與常見應(yīng)用場景

更新時間：2024年02月18日 15:44:51 作者：波羅學(xué)

Go?中有一個特別的?init()?函數(shù),它主要用于包的初始化,這篇文章將以此為主題介紹?Go?中?init()?函數(shù)的使用和常見使用場景,希望對大家有所幫助

init() 函數(shù)的執(zhí)行時機

首先，init() 的執(zhí)行時機處于包級別變量聲明和 main() 函數(shù)執(zhí)行之間。

這意味著在包中聲明的全局變量，如果附帶初始化表達式，這些表達式將在任何 init() 函數(shù)執(zhí)行之前進行初始化。

我們通過一個示例演示，代碼如下：

var Age = GetAge()

func GetAge() int {
    return 18
}

func init() {
    fmt.Printf("You're %d years old.\n", Age)
    Age = 3
}

func main() {
    fmt.Printf("You're %d years old.\n", Age)
}

輸出：

You're 18 years old
You're 3 years old

從輸出可知，GetAge() 函數(shù)作為 Age 的初始化函數(shù)，于 init() 函數(shù)前執(zhí)行，賦值 Age 為 3。而 init() 函數(shù)于其后執(zhí)行，賦值 Age 為 3。main() 函數(shù)則在最后執(zhí)行，輸出最終的 Age 值。

這個順序是符合我們預(yù)期的。

與被引入包的 init() 函數(shù)

如果一個包導(dǎo)入了其他包，被導(dǎo)入包的初始化 init() 則會先于導(dǎo)入它的包的變量初始化和 init 函數(shù)前執(zhí)行。

舉例來說明吧！

假設(shè)，我們有一個 main 包，它導(dǎo)入了 sub 包，并且同樣有一個 init() 函數(shù)：

// main.go

package main

import (
    "fmt"
    _ "demo/sub"
)

var age = GetAge()

func GetAge() int {
  fmt.Println("main initialize variables.")
  return 18
}

func init() {
    fmt.Println("main package init")
}

func main() {
    fmt.Println("main function")
}

而 sub 包中包含定義的 init() 函數(shù)

// sub/sub.go

package sub

import "fmt"

var age = GetAge()

func GetAge() int {
  fmt.Println("sub initialize variables.")
  return 18
}

func init() {
    fmt.Println("sub package init")
}

// 其他可能的函數(shù)和聲明

當你運行 main.go 時，輸出將會按照以下順序出現(xiàn)：

sub initialize variables.
sub package init
main initialize variables.
main package init
main function

這個示例清晰地展示了包的初始化順序：首先是被導(dǎo)入包（sub）的 init() 函數(shù)，然后是導(dǎo)入它的包（main）的 init() 函數(shù)，最后是 main 函數(shù)。

這也確保了依賴包在使用前已經(jīng)被正確初始化。

特別說明：

init() 區(qū)別于其他函數(shù)，不需要我們顯式調(diào)用，它會自動被 Go runtime 調(diào)用。而且，每個包中的 init() 只會被執(zhí)行一次。

一個包其實可有多個 init()，無論是在分部在包中的同一個文件中還是多個文件中。如果分布在多個文件中，執(zhí)行順序通常是按照文件名的字典順序。

為說明這個問題，我們首先修改 sub.go 文件，內(nèi)容如下：

// sub/sub.go

package sub

import "fmt"

var age = GetAge()

func GetAge() int {
  fmt.Println("sub initialize variables.")
  return 18
}

func init() {
    fmt.Println("sub init 1")
}

func init() {
    fmt.Println("sub init 2")
}

新增一個 sub1.go 文件，如下所示：

// sub/sub1.go

package sub

import "fmt"

var age = GetAge1()

func GetAge1() int {
  fmt.Println("sub1 initialize variables.")
  return 18
}

func init() {
    fmt.Println("sub1 init")
}

輸出：

sub initialize variables.
sub init 1
sub init 2
sub1 initialize variables.
sub1 init
main initialize variables.
main package init
main function

結(jié)果符合預(yù)期。

init() 的使用場景

init() 函數(shù)通常用于進行一些必要的設(shè)置或初始化操作，例如初始化包級別的變量與命令行參數(shù)、配置加載、環(huán)境檢查、甚至注冊插件等。

項目開發(fā)中，組件依賴管理通常比較令人頭疼。但一些簡單的依賴關(guān)系，即使沒有如 wire 這樣依賴注入工具的加持，通過 init 也可管理。

命令行參數(shù)

對于開發(fā)一個簡單的命令行應(yīng)用，init() 和標準庫 flag 包結(jié)合，可快速完成命令命令行參數(shù)的初始化。

package main

import (
    "flag"
    "fmt"
)

var name string
var help bool

func init() {
    flag.StringVar(&name, "name", "World", "a name to say hello to")
    flag.StringVar(&help, "name", "World", "display help information")
    flag.Parse()
}

func main() {
    if help {
        fmt.Fprintf(os.Stderr, "Usage of %s:\n", os.Args[0])
        flag.PrintDefaults()
        os.Exit(1)
    } 
    fmt.Printf("Hello, %s!\n", name)
}

以上示例中，init() 函數(shù)解析了命令行參數(shù)并初始化變量 name 和 help 變量。

配置加載

init 函數(shù)的領(lǐng)哇一個常見場景是配置加載。配置通常是程序啟動時要盡早執(zhí)行的操作。

例如，你有一個 web 服務(wù)，要在啟動服務(wù)器前加載數(shù)據(jù)庫配置、API 密鑰或其他服務(wù)配置。

var config AppConfig

func init() {
    configFile, err := os.Open("config.json")
    if err != nil {
        log.Fatal(err)
    }
    defer configFile.Close()
    jsonParser := json.NewDecoder(configFile)
    jsonParser.Decode(&config)
}

如果配置加載都出現(xiàn)問題，很大程度說明服務(wù)配置不正常，要立刻退出服務(wù)。我們可使用 log.Fatal(err) （更優(yōu)雅）或 panic(err) 退出服務(wù)。

環(huán)境檢查

init() 還可以用于檢查和驗證程序運行所需的環(huán)境。如，我們要確保必要的環(huán)境變量已設(shè)置，或者必要的外部服務(wù)可用。

如我們的必須依賴一個需要認證的外部服務(wù)，示例代碼：

func init() {
    if os.Getenv("XXX_API_KEY") == "" {
        log.Fatal("XXX_API_KEY environment variable not set")
    }

    apiKey := os.Getenv("XXX_API_KEY")
    // instantiating Component
    // ...
}

通過，如果要實例化的組件不需要賴加載，創(chuàng)建和配置驗證同時 init() 中完成即可。

注冊插件或服務(wù)

如果你的程序用的是插件架構(gòu)，我們可以在程序啟動時注冊這些插件。init() 正可以用來自動注冊這些插件。

示例代碼：

func init() {
    plugin.Register("myPlugin", NewMyPlugin)
}

Go 的數(shù)據(jù)庫驅(qū)動管理可作為這種場景的典型案例。

Go 的 database 操作通常依賴 database/sql 包，它提供了一種通用接口與 SQL 或類 SQL 數(shù)據(jù)庫交互。而具體的驅(qū)動實現(xiàn)（如 MySQL、PostgreSQL、SQLite 等）通常是通過實現(xiàn) database/sql 包定義接口來提供支持。

這種架構(gòu)下，init() 被用于驅(qū)動的自動注冊。

例如，如下這個 MySQL 驅(qū)動的實現(xiàn)：

package mysql

import (
    "database/sql"
)

func init() {
    sql.Register("mysql", &MySQLDriver{})
}

type MySQLDriver struct {
    // 驅(qū)動的實現(xiàn)
}

我們只要導(dǎo)入這個 database driver 包，它的 init() 就會被調(diào)用，將驅(qū)動注冊到 database/sql 包中。

我們使用的時候，通過 database/sql 接口即可使用該 MySQL 驅(qū)動，而不需關(guān)心它的實現(xiàn)細節(jié)。

import (
    "database/sql"
    _ "github.com/go-sql-driver/mysql" // 導(dǎo)入 MySQL 驅(qū)動
)

func main() {
    db, err := sql.Open("mysql", "user:password@/dbname")
    // ...
}

通過這種方式，Go 的數(shù)據(jù)庫驅(qū)動代碼更加模塊化和靈活性。使用方只需關(guān)心與 database/sql 交互即可，而不必關(guān)心驅(qū)動的實現(xiàn)細節(jié)。

實際的場景案例，我覺得肯定不止這么多。對于任何需要提前初始化和驗證的場景，可適當考慮是否可通過使用 init() 來簡化代碼。