如何跨包中的文件使用全局变量？

Question

我有以下文件结构：

模型/db.go

type DB struct {

    *sql.DB

}

var db *DB

func init() {

    dbinfo := fmt.Sprintf("user=%s password=%s dbname=%s sslmode=disable",

        DB_USER, DB_PASSWORD, DB_NAME)

    db, err := NewDB(dbinfo)

    checkErr(err)

    rows, err := db.Query("SELECT * FROM profile")

    checkErr(err)

    fmt.Println(rows)

}

func NewDB(dataSourceName string) (*DB, error) {

    db, err := sql.Open("postgres", dataSourceName)

    if err != nil {

        return nil, err

    }

    if err = db.Ping(); err != nil {

        return nil, err

    }

    return &DB{db}, nil

}

模型/db_util.go

func (p *Profile) InsertProfile() {

    if db != nil {

        _, err := db.Exec(...)

        checkErr(err)

    } else {

        fmt.Println("DB object is NULL")

    }

}

当我尝试db在InsertProfile函数中访问时，它显示NULL ptr exception. 我如何访问dbin db_utils.go？

我不想大写db（因为它可以访问所有包）。

我正确地收到了从dbin返回的 QUERY init()。

Answer 1

问题是您使用了Short 变量声明， :=并且您只是将创建的*DB值存储在局部变量中而不是全局变量中。这一行：db, err := NewDB(dbinfo)创建 2 个局部变量：dband err，这个局部变量db与你的全局db变量无关。您的全局变量将保持为nil。您必须将 created 分配给*DB全局变量。不要使用简短的变量声明，而是使用简单的赋值，例如：var err errordb, err = NewDB(dbinfo)if err != nil {    log.Fatal(err)}原答案如下。它是一种指针类型，在使用之前必须对其进行初始化。指针类型的零值是nil。您不必导出它（这就是以大写字母开头的内容）。请注意，您拥有多个文件并不重要，只要它们属于同一个包的一部分，它们就可以访问彼此定义的标识符。一个好的解决方案是在init()自动调用的包函数中执行此操作。请注意，sql.Open()可能只验证其参数而不创建与数据库的连接。要验证数据源名称是否有效，请调用DB.Ping()。例如：var db *sql.DBfunc init() {    var err error    db, err = sql.Open("yourdrivername", "somesource")    if err != nil {        log.Fatal(err)    }    if err = db.Ping(); err != nil {        log.Fatal(err)    }}

Answer 2

icza 已经正确回答了您的具体问题，但值得添加一些关于您做错了什么的额外解释，以便您了解如何在将来不犯错误。在 Go 中，:=赋值语法创建了名称位于 左侧的新变量:=，可能是隐藏包，甚至是父作用域函数/方法变量。举个例子：package mainimport "fmt"var foo string = "global"func main() {    fmt.Println(foo) // prints "global"    // using := creates a new function scope variable     // named foo that shadows the package scope foo    foo := "function scope"     fmt.Println(foo) // prints "function scope"    printGlobalFoo() // prints "global"    if true {        foo := "nested scope"        fmt.Println(foo) // prints "nested scope"        printGlobalFoo() // prints "global"     }     // the foo created inside the if goes out of scope when     // the code block is exited    fmt.Println(foo) // prints "function scope"    printGlobalFoo() // prints "global"    if true {        foo = "nested scope" // note just = not :=    }    fmt.Println(foo) // prints "nested scope"    printGlobalFoo() // prints "global"    setGlobalFoo()    printGlobalFoo() // prints "new value"}func printGlobalFoo() {    fmt.Println(foo)}func setGlobalFoo() {    foo = "new value" // note just = not :=}注意 Go 无法删除或取消设置变量，因此一旦您隐藏了更高范围的变量（例如通过创建与包范围变量同名的函数范围变量），就无法访问更高范围的变量该代码块中的变量。另请注意，这:=是var foo =. 两者的行为方式完全相同，但是:=仅在函数或方法中var才是有效的语法，而该语法在任何地方都有效。

Answer 3

对于谁来到这里并想要快速回答。在db.go文件中：package dbvar db *DBtype DB struct {    *gorm.DB // or what database you want like *mongo.Client}func GetDB() *DB {    if db == nil{        db = ConnectToYourDbFunc("connection_string")    }    return db}然后在您的其他软件包中，您可以通过以下方式获得它：db := db.GetDB()就这样。