【问题标题】:Go - Variable initialized and non nil, but nil for other functionsGo - 变量已初始化且非 nil,但其他函数为 nil
【发布时间】:2016-07-20 17:03:04
【问题描述】:

写一个符合我当前问题的标题有点困难.. 我有一个 main() 函数,它使用另一个包(database_sql)中的函数。此函数初始化一个全局变量 sql.DB*。初始化后,变量不是nil,但是对于其他函数,这个变量仍然是nil..让我们看下面的代码!

main.go

package main

import (
    "net/http"
    db "./database_sql"
    router "./router"
)

func main() {

    db.Init_SQL();

    router.Init_routes()

    http.ListenAndServe(":8080", router.GetRouter())

}

db.go

package database_sql

import (
    "fmt"
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
)

var DB *sql.DB // global variable to share it between main and the HTTP handler                                                                                                                              
//var dbi *databaseinfos                                                                                                                                                                                     

func Init_SQL() {
    dbi := databaseinfos{
            user: "something",
            password: "something",
            name: "something",
            address: "something",
            port: "2323",
            url: ""}
    dbi.url = (dbi.user + ":" + dbi.password + "@tcp(" + dbi.address + ":" + dbi.port + ")/" + dbi.name)

    db_init_var, err := sql.Open("mysql", dbi.url)
    if err != nil {
            fmt.Println("Error on initializing database connection: %s", err.Error())
    }

    err = db_init_var.Ping()
    if err != nil {
            fmt.Println("Error on opening database connection: %s", err.Error())
    }

    // Here, as you can test, my variable is initialized
    if err == nil {
            DB = db_init_var
            if DB == nil {
                    fmt.Println("NIL DB !!")
            }
            if db_init_var == nil {
                    fmt.Println("NIL db_init_var !!")
            }
            fmt.Println(dbi.url)
    }
}

现在,一切正常!我现在将使用用户名/密码测试http://xxxxxxx/login。一切正常,现在是时候使用上面声明的 DB 变量在数据库中发出请求了。

request_user.go

package database_sql

import (
    "encoding/json"
    "fmt"
    "net/http"
    m "models"
)

func CanLogin(user m.User) (bool, m.User, m.StatusBack) {

    // Prepare statement for reading data                                                                                                                                                                
    stmtOut, err := DB.Prepare("SELECT username, password, token FROM users WHERE username = ? AND password = ?")
    if err != nil {
            return false, user, m.StatusBack{ToString: err.Error(), IdStatus: http.StatusInternalServerError}
    }
defer stmtOut.Close()

    // Query the user                                                                                                                                                                                    
err = stmtOut.QueryRow(user.Username, user.Password).Scan(&user.Username, &user.Password, &user.UUID) // WHERE user is registered                                                                    
    if err != nil {
            return false, user, m.StatusBack{ToString: "User not found.", IdStatus: http.StatusNotFound}
    } else {

            j, err := json.Marshal(user)
            if err == nil {
                    return true, user, m.StatusBack{ToString: string(j), IdStatus: http.StatusOK}
            } else {
                    return false, user, m.StatusBack{ToString: err.Error(), IdStatus: http.StatusInternalServerError}
    }
    }

}

但是,当我发出我的 sql 请求时,以下行表示 DB 为 nil。 stmtOut, err := DB.Prepare("SELECT username, password, token FROM users WHERE username = ? AND password = ?")

我做了很多测试,比如尝试用'='初始化它而不用':='。我尝试了“if DB == nil”语句,但 DB 始终为零。为什么?我在初始化路由器之前初始化了我的数据库变量,所以从技术上讲,当我的路由器重定向到一个函数时,这个函数不会使用数据库变量(DB)nil,不是吗?

感谢您的回答!

编辑 2 我把你的答案涂红了,然后,我像你问的那样编辑了我的代码,并添加了一些内容。在 init_SQL 之后,DB 不是 nil !

main.go

package main

import (
    "fmt"
    "net/http"
    db "./database_sql"
    router "./router"
)

func main() {

    if db.Init_SQL() == true {

            if db.DB == nil {
                    fmt.Println("db.DB is nil !")
            }

            if router.Init_routes() == true {

                    http.ListenAndServe(":8080", router.GetRouter())

            }
    }
}

所以,现在我的 2 个初始化函数在成功时返回 true,在失败时返回 false。我认为这可能是一个异步问题,我这样做是为了等待每个函数的结束以继续我的“逐步”

【问题讨论】:

  • 我假设“错误打开数据库”不会打印,对吗?除了 Init_SQL 之外,您没有理由在 DB.Prepare 行上失败 Ping() db
  • 能否检查一下db.Init_SQL();之后是否设置了全局DB变量?
  • 我建议您在db.Init_SQL()router.Init_routes() 之后打印出DB 值,也可以在您在调用堆栈中进行的每个 函数调用之前/之后,在此当您的DB 被重置或根本没有设置时,您可能会有一个想法。
  • 感谢您的回答,我添加了“EDIT 2

标签: go initialization global-variables


【解决方案1】:

首先,像这样导出全局变量并不符合习惯。复杂类型应由运行时初始化(您正在使用main()),但从运行时全局保留以控制处置。

您还缺少db.Close()

我相信几年前我在第一次使用 MySQL 时遇到了与您的模式相同的问题。创建本地范围指针,然后将其分配给全局变量的方式有点奇怪。通常最好将它直接分配给全局变量。但我认为核心问题是您需要将指针分配给指针。

我使用的模式是将可测试的数据库/sql *DB 保持在我初始化它的全局状态。为什么要在轮子起作用时重新发明轮子:

package main

import ( 
  ...
  "database/sql"

  "mypackage-that-inits-db/database"
)

var db *sql.DB

find main() {

  var err error
  db, err = database.Init_SQL(...params)
  if err != nil {
    ...
  }
  defer db.Close()

  InitUsers(db)
  InitStats(db)
  ...

  http.ListenAndServe(...)

  // now inject db global dependency in other code
  //

  globalStats := NewStatsEngine(db)
  globalStats.RecordUpdateToSomething(...)

  users := getUsers(db)

   ... etc

}

这通常是您在其他代码中看到的模式。

注意对 defer 和 Close 的控制,它属于 caller 的位置。

不能轻易测试

另外请注意,您在其他包中的代码现在可以通过创建模拟 sql.DB 对象并将其注入 NewStatsEngine、getUsers 等变得容易测试,而不必模拟全局变量、测试、拆除和重新设置以进行测试测试、测试、拆解等。同样从 Go 1.5 开始,测试可以同时运行,所以如果你像这样离开你的包,你甚至需要在全局数据库变量周围放置一个互斥锁。切换到 IoC 模式,例如我在上面的代码中演示的依赖注入,可以更轻松地测试(和调试)你的代码。

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2017-04-13
    • 1970-01-01
    • 2019-06-25
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2014-01-29
    相关资源
    最近更新 更多