小编典典

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

go

我具有以下文件结构:

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

当我尝试访问dbInsertProfile功能,它说NULL ptr exception。如何访问dbdb_utils.go

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

我正在从返回的查询dbinit()正确。


阅读 423

收藏
2020-07-02

共1个答案

小编典典

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

这行:

db, err := NewDB(dbinfo)

创建2个局部变量:dberr,并且该局部变量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