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

小编典典

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

我具有以下文件结构：

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

当我尝试访问db的InsertProfile功能，它说NULL ptr exception。如何访问db的db_utils.go？

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

我正在从返回的查询db中init()正确。

阅读 458

2020-07-02

共1个答案

小编典典

编辑：
问题是您使用了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()自动调用的package 函数中执行此操作。

请注意，这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)
    }
}

2020-07-02