Question

我读到你不应该关闭* sql.DB变量。

http://go-database-sql.org/accessing.html

它还说我应该：＆＃34;根据需要传递它，或者以某种方式在全球范围内提供它，但保持它开放。＆＃34;

但是这篇文章说我不应该使用全局变量但应该使用闭包： https://medium.com/@benbjohnson/structuring-applications-in-go-3b04be4ff091

我在这里找到了一个关闭的例子： https://gist.github.com/tsenart/5fc18c659814c078378d

我的问题是：我应该如何将此变量传递给不同的包？

例如，如果我有一个名为User的包，就像这样：

package user

import "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类型或仅包含*sql.DB字段的“环境”或“服务”结构定义为字段，并将此处理程序创建为方法。这也可能是全局的（注意任何成员必须是线程安全的）。
定义一个自定义处理程序类型（我的方法），您可以在其中调用router.GET("/user/:id", GetUserByID(env)而不是使用闭包。

一些额外的阅读：

http://www.alexedwards.net/blog/organising-database-access（涵盖多种方法 - 最全面的文章之一）
https://robots.thoughtbot.com/interface-with-your-database-in-go
https://medium.com/@benbjohnson/structuring-applications-in-go-3b04be4ff091
http://elithrar.github.io/article/http-handler-error-handling-revisited/#a-little-different

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

1 个答案: