Go 中 init 函数的作用是什么？

在Go语言中，init 函数是一个特殊的函数，它主要用于进行包（package）级别的变量初始化或执行一些特定的设置任务。每个包可以包含多个 init 函数，这些函数按照它们在代码中出现的顺序自动执行，而且在程序的 main 函数执行之前就已经被调用完成了。

init 函数的主要用途：

1. 初始化包内部的变量： 如果包内有一些复杂的变量需要通过函数来初始化，那么 init 函数是完成这种初始化的理想地点。
2. 执行启动前的检查或设置： 比如，检查环境变量是否设置，或者初始化数据库连接等等。
3. 注册： 在某些情况下，包可能需要注册其能力或服务到一个全局查找表中，init 函数可以用来执行这样的注册。

示例

假设我们有一个Web服务的包，需要在启动前确保数据库连接已就绪，并且根据环境变量配置一些内部参数。我们可以使用 init 函数来完成这些任务：

go
package database

import (
    "database/sql"
    "log"
    "os"
)

var db *sql.DB

func init() {
    var err error
    db, err = sql.Open("postgres", os.Getenv("DATABASE_URL"))
    if err != nil {
        log.Fatalf("Error opening database: %v", err)
    }

    // 可以在这里添加更多的初始化设置，比如数据库迁移等
}

// GetDB 是一个返回数据库连接的函数
func GetDB() *sql.DB {
    return db
}

在这个例子中，init 函数确保在包被其他代码导入并使用前，数据库连接已经被正确设置和初始化。这样，其他部分的代码调用 GetDB() 函数时，可以安全地假设数据库连接是有效的，无需担心 nil 或未初始化的情况。

总结

init 函数提供了一种便捷的方式来设置包级别的状态或执行初始化任务，它有助于代码的模块化和清晰的启动逻辑。但是，使用 init 函数也需要小心，因为它的执行顺序和时间可能会影响程序的行为，特别是在涉及多个包有依赖关系时。