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

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")
    }
}

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

我不想大写 db(因为它将允许访问所有软件包)。

我正确地从 db中的 init()返回了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() 可以仅验证其参数而无需创建与数据库的连接。要验证数据源名称是否有效，请调用 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)
    }
}