写一个与我当前问题相匹配的标题有点困难。。我有一个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=?")
我做了很多测试,比如试图用'='和no':='初始化它。我尝试了"if DB==nil"语句,但DB始终为nil。。为什么?我在初始化路由器之前初始化了数据库var,所以从技术上讲,当我的路由器重定向到一个函数时,这个函数不会使用数据库var(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个init函数在成功时返回true,在失败时返回false。我认为这可能是一个异步问题,我这样做是为了等待每个函数的结束来继续我的"逐步"
首先,导出这样的全局var不是idomatic。复杂类型应该由运行时初始化(您正在使用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的控制,它属于调用者中的此处。
无法轻松测试
还要注意的是,通过创建一个mock sql.DB
对象并将其注入NewStatsEngine、getUsers等,您在其他包中的代码现在变得可以很容易地进行测试,而不必为了测试、测试、拆卸等而模拟全局var、测试、拆下和重新设置。此外,自Go 1.5以来,测试可以并发运行,因此您甚至需要放置一个互斥锁。如果你这样留下你的包,锁定你的全局数据库变量。切换到IoC模式,比如我在上面的代码中演示的依赖注入,可以让测试(和调试)代码变得更容易。