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

Question

我有以下文件结构：

models/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
}

models/db_util.go

func (p *Profile) InsertProfile() {
    if db != nil {
        _, err := db.Exec(...)
        checkErr(err)
    } else {
        fmt.Println("DB object is NULL")
    }
}

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

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

我正在从 init() 中的 db 正确获取返回的 QUERY。

Answer 1

编辑： 问题是您使用了 Short variable declaration := 并且您只是将创建的 *DB 值存储在局部变量中并且不在全球范围内。

这一行：

db, err := NewDB(dbinfo)

创建2个局部变量：db和err，这个局部db和你的全局db变量没有任何关系。您的全局变量将保持 nil。您必须将创建的 *DB 分配给全局变量。不要使用简短的变量声明，而是简单的 assignment，例如：

var err error
db, err = NewDB(dbinfo)
if err != nil {
    log.Fatal(err)
}

原回答如下。

是指针类型，使用前必须先初始化。指针类型的零值为 nil。

您不必导出它（这就是以大写字母开头的方式）。请注意，您拥有多个文件并不重要，只要它们是同一包的一部分，它们就可以访问彼此定义的标识符。

一个好的解决方案是在自动调用的包 init() 函数中执行此操作。

注意 sql.Open() may just validate its arguments without creating a connection to the database. To verify that the data source name is valid, call DB.Ping().

例如：

var db *sql.DB

func 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 中，用于赋值的语法 := 创建新变量，其名称位于 := 左侧，可能是阴影包，甚至是父作用域 function/method 变量。例如：

package main

import "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 = 的 shorthand。两者的行为方式完全相同，但是 := 只是函数或方法中的有效语法，而 var 语法在任何地方都有效。

Answer 3

对于那些来到这里并想要快速回答的人。

在 db.go 文件中：

package db

var db *DB

type 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()

就这些了。

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

How to use global var across files in a package?

pointers

database-connection

go

nullreferenceexception