Go using mux Router - 如何将我的数据库传递给我的处理程序

欢迎各位小伙伴来到golang学习网，相聚于此都是缘哈哈哈！今天我给大家带来《Go using mux Router - 如何将我的数据库传递给我的处理程序》，这篇文章主要讲到golang等等知识，如果你对Golang相关的知识非常感兴趣或者正在自学，都可以关注我，我会持续更新相关文章！当然，有什么建议也欢迎在评论留言提出！一起学习！

问题内容

目前，我尝试使用 Go 创建一个小型 Web 项目，用于在服务器上处理数据。

我尝试将我的数据库连接传递给我的 HandlerFunc(tions) 但它没有按预期工作。我对golang很陌生，所以也许我不了解这个lang的一些基本原理。

我的主要功能如下所示：

func main() {

    db, err := config.NewDB("username:password@/databasename?charset=utf8&parseTime=True")
    if err != nil {
        log.Panic(err)
    }   
    env := &config.Env{DB: db} 

    router := NewRouter(env)
    log.Fatal(http.ListenAndServe(":8080", router))
}

我的路由器：

func NewRouter(env *config.Env) *mux.Router {
    router := mux.NewRouter().StrictSlash(true)
    for _, route := range routes {
        var handler http.Handler

        handler = route.HandlerFunc
        handler = Logger(handler, route.Name)

        router.
            Methods(route.Method).
            Path(route.Pattern).
            Name(route.Name).
            Handler(handler)
    }   
    return router
}

和我的路线：

type Route struct {
    Name        string
    Method      string
    Pattern     string
    HandlerFunc http.HandlerFunc
}

type Routes []Route

var routes = Routes{
    Route{
        "Index",
        "GET",
        "/",
        controller.Index,
    },  
    Route{
        "Show",
        "GET",
        "/todos/{todoId}",
        controller.TodoShow,
    },  
    Route{
        "Create",
        "POST",
        "/todos",
        controller.TodoCreate,
    },  
}

那么 - 我怎样才能将我的“env”（或 env.DB）传递给我的 FuncHandlers？我尝试了很多东西，但都没有奏效。

正确答案

你有三个选择：

1. 使您的数据库连接池成为全局的，这样您就不必通过它。sql.DB对于并发访问是安全的，这是最简单的方法。缺点是它使测试变得更加困难并且混淆了池的“来源” - 例如

   var db *sql.DB

func main() {
    var err error
    db, err = sql.Open(...)
    // Now accessible globally, no need to pass it around
    // ...
 }

2. 将您的处理程序包装在一个闭包中，这使得内部处理程序可以访问它。您需要将其适应您的路线范围方法——这在 IMO 中有点迟钝，并且更难看到存在哪些路线，但我离题了——或者例如：

   func SomeHandler(db *sql.DB) http.HandlerFunc { fn := func(w http.ResponseWriter, r *http.Request) { res, err := db.GetThings() // etc. }

   return http.HandlerFunc(fn) }

func main() {
    db, err := sql.Open(...)
    http.HandleFunc("/some-route", SomeHandler(db))
    // etc.
}

3. 创建一个接受处理程序的自定义处理程序类型 - 例如

   type AppHandler struct { Handler func(env *config.Env, w http.ResponseWriter, r *http.Request) Env *config.Env }

// ServeHTTP allows your type to satisfy the http.Handler interface.
func (ah *AppHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    ah.Handler(ah.Env, w, r)
}

func SomeHandler(env *config.Env, w http.ResponseWriter, r *http.Request) {
    res, err := env.DB.GetThings()
    // etc.
}

请注意（无耻的插件！）我已经[详细介绍了最后一种方法](http://elithrar.github.io/article/http-handler- error-handling-revisited/)，Alex Edwards 也有一篇关于在 Go 程序中访问数据库池的方法的[优秀博客文章。](http://www.alexedwards.net/blog/organising-database- access)

我能给出的唯一严格建议是，您应该避免在请求上下文中传递数据库池，这是低效且不好的做法（请求上下文是针对临时的、按请求的对象）。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。