我应该如何在代码的几个地方传递 *sql.DB 变量？

Question

我读到你不应该关闭 *sql.DB 变量。http://go-database-sql.org/accessing.html它还说我应该：“根据需要传递它，或者以某种方式在全球范围内提供它，但保持开放。”但是这篇文章说我不应该使用全局变量而应该使用闭包：https : //medium.com/@benbjohnson/structuring-applications-in-go-3b04be4ff091我在这里找到了一个关闭的例子：https : //gist.github.com/tsenart/5fc18c659814c078378d我的问题是：我应该如何将此变量传递给不同的包？例如，如果我有一个名为 User 的包，如下所示：package userimport "errors"var userNotFound = errors.New("User was not found.")// Should I have a pointer to sql.DB as a property of the User type?type User struct {    // db *sql.DB    Id       int    Email    string    Username string}// Should I pass in *sql.DB as parameter in the function?func FindById(id int) (*User, error) {    // Access *sql.DB somehow    // Do query and look if user with id is found    // Should I return an error if the user is not found?    return &User{}, nil}然后我应该有一个指向 sql.DB 的指针作为 User 类型的属性吗？或者我应该在 findById 方法中传递一个指向它的指针吗？如果我想通过 id 查找用户，我应该如何从如下所示的主函数中完成所有这些操作？func getUserById(db *sql.DB) httprouter.Handle {    return func(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {        fmt.Fprint(w, ps.ByName("id"))        // I will create a new(User) here, how should I use the *sql.DB in the user package?    }}func main() {    dsn := fmt.Sprintf("%s:%s@%s(%s:%s)/%s?charset=utf8",        cfg.DbUser, cfg.DbPass, cfg.DbProtocol, cfg.DbAddress, cfg.DbPort, cfg.DbName)    db, err := sql.Open("mysql", dsn)    err = db.Ping()    if err != nil {        log.Fatal(err)    }    router := httprouter.New()    router.GET("/api/user/:id", getUserById(db))    router.NotFound = &DefaultHandler{}    log.Fatal(http.ListenAndServe(":8080", router))}我该怎么办？什么是好的首选方式或最佳实践？

Answer 1

超级简单？一个全局var db *sql.DB对象。*sql.DB是线程安全的，因此可以并发访问。

或者，您可以在包装*sql.DB. 我的观点是func (u *User) FindByID(id string) (*User, error)没有多大意义——你接受一个用户指针，但返回一个指向用户的新指针？

作为一个简化的示例，您可以将代码更改为类似于以下内容：

type DB struct {

    *sql.DB

}

func NewDB(host, port string) (*DB, error) {

    db, err := sql.Open(...)

    if err != nil {

        return nil, err

    }

    return &DB{db}, nil 

}

func (db *DB) UserByID(id string) (*User, error) {

    res, err := db.Query(...)

    // etc.

}

func (db *DB) UsersList(limit int) ([]*User, error) {

    res, err := db.Query(...)

    // etc.

}

为了从您的处理程序调用这些函数，您可以：

• 使用闭包（就像你现在所做的那样）

• 定义一个“环境”或“服务”结构，其中包含您的DB类型或只是 vanilla*sql.DB作为字段，并在此基础上创建您的处理程序作为方法。这也可以是全局的（注意任何成员都必须是线程安全的）。

• 定义一个自定义处理程序类型（我的方法），您可以在其中调用router.GET("/user/:id", GetUserByID(env)而不是使用闭包。