提问人:lionelmessi 提问时间:12/10/2015 最后编辑:Jonathan Halllionelmessi 更新时间:10/24/2020 访问量:69155
如何在包中的文件之间使用全局变量?
How to use global var across files in a package?
问:
我有以下文件结构:
模型/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?db
InsertProfile
NULL ptr exception
db
db_utils.go
我不想大写(因为它可以访问所有包)。db
我正确地从 in 返回了 QUERY。db
init()
答:
编辑:问题是您使用了 Short 变量声明,并且您只是将创建的值存储在局部变量中,而不是全局变量中。:=
*DB
这一行:
db, err := NewDB(dbinfo)
创建 2 个局部变量:和 ,此局部变量与全局变量无关。您的全局变量将保留 。您必须将创建的变量分配给全局变量。不要使用简短的变量声明,而使用简单的赋值,例如:db
err
db
db
nil
*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)
}
}
评论
:=
*DB
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
因为谁来到这里并想要一个快速的答案。
在文件中: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()
就这样。
评论