如何将连接参数 db 传递给 main.go？

一分耕耘，一分收获！既然都打开这篇《如何将连接参数 db 传递给 main.go？》，就坚持看下去，学下去吧！本文主要会给大家讲到等等知识点，如果大家对本文有好的建议或者看到有不足之处，非常欢迎大家积极提出！在后续文章我会继续更新Golang相关的内容，希望对大家都有所帮助！

我目前的作品有问题。我使用 beego 框架构建了一个应用程序，我是 golang 新手。

首先，我构建了另一个名为 utils 的包，并从该包中编写了一些代码来访问我的数据库

func initfirebird() {
    var (
        dbdriver   = beego.appconfig.string("db_connection")
        dbusername = beego.appconfig.string("db_username")
        dbpassword = beego.appconfig.string("db_password")
        dbserver   = beego.appconfig.string("db_host")
        // dbport     = beego.appconfig.string("db_port")
        dbfilename = beego.appconfig.string("db_database")
    )
    conn, _ := sql.open(dbdriver, dbusername+":"+dbpassword+"@"+dbserver+"/"+dbfilename)
    defer conn.close()
}

之后，我转到 main.go 并设置我的 init 函数和 main 函数，如下所示：

func init() {
    utils.InitFirebird()
}

func main() {
    if beego.BConfig.RunMode == "dev" {
        beego.BConfig.WebConfig.DirectoryIndex = true
        beego.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
        }

    var n int

    conn.QueryRow("SELECT Count(*) FROM rdb$relations").Scan(&n)
    fmt.Println("Relations count=", n)

    beego.Run()
}

当我启动我的应用程序时，出现错误并向我提供以下消息：

\main.go:23:2：未定义：conn

我该如何解决这个问题？

任何帮助都会感激

解决方案

首先，如果您想从另一个包访问某些内容，则必须将其导出。在 go 中，如果你想导出一些东西，你可以将其首字母大写命名（在你的例子中，它应该是 conn 而不是 conn）。

其次，当您使用 defer 时，它将在函数返回时执行。在您的情况下，它会立即返回，因此连接会立即关闭。

解决方案：

var conn *sql.db
func initfirebird() {
    var (
        dbdriver   = beego.appconfig.string("db_connection")
        dbusername = beego.appconfig.string("db_username")
        dbpassword = beego.appconfig.string("db_password")
        dbserver   = beego.appconfig.string("db_host")
        // dbport     = beego.appconfig.string("db_port")
        dbfilename = beego.appconfig.string("db_database")
    )
    conn, _ = sql.open(dbdriver, dbusername+":"+dbpassword+"@"+dbserver+"/"+dbfilename)
}

现在在你的主包中：

func init() {
    utils.InitFirebird()
}

func main() {
    if beego.BConfig.RunMode == "dev" {
        beego.BConfig.WebConfig.DirectoryIndex = true
        beego.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
        }

    var n int
    defer utils.Conn.Close() // <-- Close here

    utils.Conn.QueryRow("SELECT Count(*) FROM rdb$relations").Scan(&n)
    fmt.Println("Relations count=", n)

    beego.Run()
}

此处 clos​​e() 不会立即执行，因为 beego.run() 会阻塞。

ps： 不建议使用全局变量传递数据库连接。如果您想了解更多信息，请查看：https://www.alexedwards.net/blog/organising-database-access

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。