如何在包中的文件之间使用全局变量?

How to use global var across files in a package?

提问人:lionelmessi 提问时间:12/10/2015 最后编辑:Jonathan Halllionelmessi 更新时间:10/24/2020 访问量:69155

问:

我有以下文件结构:

模型/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")
    }
}

当我尝试在函数中访问时,它说.如何访问 in?dbInsertProfileNULL ptr exceptiondbdb_utils.go

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

我正确地从 in 返回了 QUERY。dbinit()

指针 go database-connection nullreferenceexception

评论


答:

60赞 icza 12/10/2015 #1

编辑:问题是您使用了 Short 变量声明,并且您只是将创建的值存储在局部变量中,而不是全局变量中。:=*DB

这一行:

db, err := NewDB(dbinfo)

创建 2 个局部变量:和 ,此局部变量与全局变量无关。您的全局变量将保留 。您必须将创建的变量分配给全局变量。不要使用简短的变量声明,而使用简单的赋值,例如:dberrdbdbnil*DB

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

原答案如下。


它是一种指针类型,您必须在使用它之前对其进行初始化。指针类型的零值为 。nil

您不必导出它(这就是以大写字母开头的作用)。请注意,只要有多个文件是同一包的一部分,它们就可以访问彼此定义的标识符。

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

请注意,sql.Open() 可能只是验证其参数,而不创建与数据库的连接。若要验证数据源名称是否有效,请调用 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)
    }
}

评论

0赞 lionelmessi 12/10/2015
完全!这就是我正在做的.
0赞 lionelmessi 12/10/2015
然而,它在另一个文件中为零。
2赞 icza 12/10/2015
@lionelmessi 我的猜测是,您使用了简短的变量声明,并且您只是将创建的值存储在局部变量中,而不是全局变量中。如果不看到您的来源,就无法分辨,所以请善意并发布。:=*DB
0赞 castelinos 6/25/2023
非常感谢!我被困在这个上面很长一段时间。
17赞 Endophage 12/10/2015 #2

ICZA已经正确地回答了你的具体问题,但值得对你做错了什么添加一些额外的解释,这样你就可以了解如何在未来不犯错误。在 Go 中,赋值的语法会创建新变量,其名称位于 的左侧,可能是影子包,甚至是父作用域函数/方法变量。举个例子::=:=

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 =:=var

11赞 ttrasn 10/24/2020 #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()

就这样。