golang中的init初始化函数-百木园

0.1、索引

https://waterflow.link/articles/1666090530880

1、概念

1.1、源文件里的代码执行顺序

init 函数是用于初始化应用程序状态的函数。它不接受任何参数并且不返回任何结果（一个 func() 函数）。初始化包时，将初始化包中的所有常量和变量声明。然后，执行初始化函数。下面是一个初始化主包的例子：

package main

import \"fmt\"

// 1
var a = func() int {
	fmt.Println(\"var a\")
	return 0
}()

// 2
func init()  {
	fmt.Println(\"init\")
}

// 1
var b = func() int {
	fmt.Println(\"var b\")
	return 1
}()

// 3
func main() {
	fmt.Println(\"main\")
}

上面代码的初始化顺序是：

初始化常量/变量（虽然b在init函数后面，但是会首先初始化）
初始化init函数
执行main函数

我们看下打印的结果：

go run 2.go
var a
var b
init
main

1.2、不同包的init函数执行顺序

初始化包时会执行一个 init 函数。在下面的例子中，我们定义了两个包，main 和 redis，其中 main 依赖于 redis。首先， 2 .go 是主包：

package main

import (
	\"fmt\"
	\"go-demo/100gomistakes/2/redis\"
)

// 2
func init()  {
	fmt.Println(\"main init\")
}

// 3
func main() {
	err := redis.Store(\"ni\", \"hao\")
	fmt.Println(err)
}

我们可以看到main包中调用了redis包的方法。

我们再看下redis包中的内容：

package redis

import \"fmt\"

// 1
func init()  {
	fmt.Println(\"redis init\")
}

func Store(key, value string) error {
	return nil
}

因为main依赖redis，所以先执行redis包的init函数，然后是main包的init，然后是main函数本身。上面的代码中标明了执行顺序。

1.3、同一个包不同文件init执行顺序

我们可以为每个包定义多个初始化函数。当我们这样做时，包内的 init 函数的执行顺序是基于源文件的字母顺序。例如，如果一个包包含一个 a.go 文件和一个 b.go 文件，并且都有一个 init 函数，则首先执行 a.go init 函数。

但是如果我们把文件a.go改为ca.go，则会先执行b.go的init函数。

所以我们不应该依赖包中初始化函数的顺序。实际上，这可能很危险，因为可以重命名源文件，从而可能影响执行顺序。

下图说明了这个问题：
wp-content/uploads/2022/10/https://www.173top.cn/wp-content/uploads/2022/10/1666690396-20221025093316-6357ad5c04d1a.png

1.4、同一个文件中的init函数

当然，我们还可以在同一个源文件中定义多个初始化函数：

package main

import (
	\"fmt\"
	\"go-demo/100gomistakes/2/redis\"
)


func init()  {
	fmt.Println(\"main init1\")
}

func init()  {
	fmt.Println(\"main init2\")
}


func main() {
	err := redis.Store(\"ni\", \"hao\")
	fmt.Println(err)
}

执行顺序是按照源文件顺序执行的，我们看下打印的结果：

go run 2.go
redis2 init
redis init
main init1
main init2
<nil>

1.5、以副作用方式执行的init函数

我们开发的时候经常会使用gorm查询数据库，所以我们经常会看到在初始化db连接的时候会有下面的代码：

package models

import (
	_ \"github.com/go-sql-driver/mysql\"
	\"github.com/jinzhu/gorm\"
)

这种情况就是说，我们没有强依赖mysql包（没有直接使用mysql的公共函数）。但是我们需要初始化mysql包里的一些数据，也就是执行mysql包里的init函数。这个时候我们就可以在这个包的前面加上_。

需要注意的是init函数不能被当作普通函数调用，会编译报错。

2、使用场景

首先，让我们看一个使用 init 函数可能被认为不合适的示例：持有数据库连接池。在示例的 init 函数中，我们使用 sql.Open 打开一个数据库。这个全局变量db会被其他函数调用：

package main

import (
	\"database/sql\"
	\"fmt\"
	_ \"github.com/go-sql-driver/mysql\"
	\"log\"
)

var db *sql.DB

func init() {
	d, err := sql.Open(\"mysql\",
		\"root:liufutian@tcp(127.0.0.1:3306)/test\")
	if err != nil {
		log.Panic(err)
	}

  err := db.Ping()
	if err != nil {
		log.Panic(err)
	}
  
	db = d
}

func main() {


}

在本例中，我们打开数据库，检查是否可以 ping 通，然后将其分配给全局变量。

但是这种实现会带来一些问题：

init 函数中的错误管理是有限的。实际上，由于 init 函数不返回错误，因此发出错误信号的唯一方法之一就是panic，导致应用程序停止。在我们的示例中，如果打开数据库失败，无论如何都会停止应用程序。但是，不一定要由包本身来决定是否停止应用程序。有可能调用者想实现重试或使用回退机制。在这种情况下，在 init 函数中打开数据库会阻止客户端包实现其错误处理逻辑。
如果我们向这个文件添加测试，init 函数将在运行测试用例之前执行，但是我们想要的是创建db连接的时候需要测试。因此，此示例中的 init 函数使编写单元测试变得复杂。
该示例需要将数据库连接池分配给全局变量。全局变量有一些严重的缺点；例如：
- 任何函数都可以更改包中的全局变量
- 单元测试可能会更复杂，因为依赖于全局变量的函数将不再被隔离

在大多数情况下，我们应该倾向于封装一个变量而不是让它保持全局。

由于这些原因，之前的初始化可能应该封装到一个函数中处理，如下所示：

func createDB() (*sql.DB, error) {
	d, err := sql.Open(\"mysql\",
		\"root:liufutian@tcp(127.0.0.1:3306)/test\")
	if err != nil {
		return nil, err
	}
	err = db.Ping()
	if err != nil {
		return nil, err
	}
	return d, nil
}

这样写的话，我们解决了之前讨论的主要缺点：

是否处理错误留给调用者
可以创建一个集成测试来检查此功能是否有效
连接池封装在函数中

但是这样就是不能使用init函数了么？在我们上面的引入mysql驱动的例子中，说明使用init还是有帮助的：

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

上面的例子，通过注册提供的驱动名称使数据库驱动程序可用。

来源：https://www.cnblogs.com/liuyuede123/p/16803696.html
本站部分图文来源于网络，如有侵权请联系删除。

golang中的init初始化函数