下面简单滴配置一下:
var DB *sql.DB
/**
*@desc 初始化数据库
*@author Carver
*/
func init() {
pwd, pwdError := os.Getwd()
if pwdError != nil {
os.Exit(1)
panic(pwdError)
}
envUrl := pwd + "/config/local.config.toml"
var config tomlConfig
_, configError := toml.DecodeFile(envUrl, &config)
if configError != nil {
panic(configError)
}
//"用户名:密码@[连接方式](主机名:端口号)/数据库名"。 下面的信息需要读取配置文件
connStr := Sprintf("%s:%s@(%s)/%s", config.Db.UserName, config.Db.Password, config.Db.Addr, config.Db.Database)
//设置连接数据库的参数
db, _ := sql.Open("mysql", connStr)
//连接数据库
err := db.Ping()
if err != nil {
Println("数据库连接失败❌")
return
} else {
Println("数据库连接成功✅")
}
DB = db
}