百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术教程 > 正文

go语言中的init初始化函数

mhr18 2024-11-25 10:39 23 浏览 0 评论

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")
}

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

  1. 初始化常量/变量(虽然b在init函数后面,但是会首先初始化)
  2. 初始化init函数
  3. 执行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函数。

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

下图说明了这个问题:

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 通,然后将其分配给全局变量。

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

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

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

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

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{})
}

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

相关推荐

【推荐】一个开源免费、AI 驱动的智能数据管理系统,支持多数据库

如果您对源码&技术感兴趣,请点赞+收藏+转发+关注,大家的支持是我分享最大的动力!!!.前言在当今数据驱动的时代,高效、智能地管理数据已成为企业和个人不可或缺的能力。为了满足这一需求,我们推出了这款开...

Pure Storage推出统一数据管理云平台及新闪存阵列

PureStorage公司今日推出企业数据云(EnterpriseDataCloud),称其为组织在混合环境中存储、管理和使用数据方式的全面架构升级。该公司表示,EDC使组织能够在本地、云端和混...

对Java学习的10条建议(对java课程的建议)

不少Java的初学者一开始都是信心满满准备迎接挑战,但是经过一段时间的学习之后,多少都会碰到各种挫败,以下北风网就总结一些对于初学者非常有用的建议,希望能够给他们解决现实中的问题。Java编程的准备:...

SQLShift 重大更新:Oracle→PostgreSQL 存储过程转换功能上线!

官网:https://sqlshift.cn/6月,SQLShift迎来重大版本更新!作为国内首个支持Oracle->OceanBase存储过程智能转换的工具,SQLShift在过去一...

JDK21有没有什么稳定、简单又强势的特性?

佳未阿里云开发者2025年03月05日08:30浙江阿里妹导读这篇文章主要介绍了Java虚拟线程的发展及其在AJDK中的实现和优化。阅前声明:本文介绍的内容基于AJDK21.0.5[1]以及以上...

「松勤软件测试」网站总出现404 bug?总结8个原因,不信解决不了

在进行网站测试的时候,有没有碰到过网站崩溃,打不开,出现404错误等各种现象,如果你碰到了,那么恭喜你,你的网站出问题了,是什么原因导致网站出问题呢,根据松勤软件测试的总结如下:01数据库中的表空间不...

Java面试题及答案最全总结(2025版)

大家好,我是Java面试陪考员最近很多小伙伴在忙着找工作,给大家整理了一份非常全面的Java面试题及答案。涉及的内容非常全面,包含:Spring、MySQL、JVM、Redis、Linux、Sprin...

数据库日常运维工作内容(数据库日常运维 工作内容)

#数据库日常运维工作包括哪些内容?#数据库日常运维工作是一个涵盖多个层面的综合性任务,以下是详细的分类和内容说明:一、数据库运维核心工作监控与告警性能监控:实时监控CPU、内存、I/O、连接数、锁等待...

分布式之系统底层原理(上)(底层分布式技术)

作者:allanpan,腾讯IEG高级后台工程师导言分布式事务是分布式系统必不可少的组成部分,基本上只要实现一个分布式系统就逃不开对分布式事务的支持。本文从分布式事务这个概念切入,尝试对分布式事务...

oracle 死锁了怎么办?kill 进程 直接上干货

1、查看死锁是否存在selectusername,lockwait,status,machine,programfromv$sessionwheresidin(selectsession...

SpringBoot 各种分页查询方式详解(全网最全)

一、分页查询基础概念与原理1.1什么是分页查询分页查询是指将大量数据分割成多个小块(页)进行展示的技术,它是现代Web应用中必不可少的功能。想象一下你去图书馆找书,如果所有书都堆在一张桌子上,你很难...

《战场兄弟》全事件攻略 一般事件合同事件红装及隐藏职业攻略

《战场兄弟》全事件攻略,一般事件合同事件红装及隐藏职业攻略。《战场兄弟》事件奖励,事件条件。《战场兄弟》是OverhypeStudios制作发行的一款由xcom和桌游为灵感来源,以中世纪、低魔奇幻为...

LoadRunner(loadrunner录制不到脚本)

一、核心组件与工作流程LoadRunner性能测试工具-并发测试-正版软件下载-使用教程-价格-官方代理商的架构围绕三大核心组件构建,形成完整测试闭环:VirtualUserGenerator(...

Redis数据类型介绍(redis 数据类型)

介绍Redis支持五种数据类型:String(字符串),Hash(哈希),List(列表),Set(集合)及Zset(sortedset:有序集合)。1、字符串类型概述1.1、数据类型Redis支持...

RMAN备份监控及优化总结(rman备份原理)

今天主要介绍一下如何对RMAN备份监控及优化,这里就不讲rman备份的一些原理了,仅供参考。一、监控RMAN备份1、确定备份源与备份设备的最大速度从磁盘读的速度和磁带写的带度、备份的速度不可能超出这两...

取消回复欢迎 发表评论: