如何将类型传递给http处理程序

Question

我试图通过为它们创建一个新包来将我的http go代码分成"控制器",但是无法弄清楚如何将db类型传递给处理程序.我希望能够将我在main.go中创建的Db类型传递到index.go中的Index处理程序.如果这是解决这个问题的错误方法,请让我知道一个更好的方法(我正在学习,并希望现在保持简单).我的代码到目前为止:

main.go:

package main

import (

    "database/sql"
    "fmt"
    _ "github.com/go-sql-driver/mysql"
    "github.com/gorilla/mux"
    "log"
    "mvc3/app/c"
    "net/http"
)

var Db *sql.DB

func main() {

    fmt.Println("Starting up!")

    var err error
    Db, err = sql.Open("mysql", "root@/dev?charset=utf8")
    if err != nil {
        log.Fatalf("Error on initializing database connection: %s", err.Error())
    }

    Db.SetMaxIdleConns(100)

    err = Db.Ping()
    if err != nil {
        log.Fatalf("Error on opening database connection: %s", err.Error())
     }

     r := mux.NewRouter()
     r.HandleFunc("/", c.Index)

    http.Handle("/", r)
    http.ListenAndServe(":8080", nil)
}

/app/c/index.go:

package c

import (
    "fmt"
    "net/http"
)

func Index(w http.ResponseWriter, r *http.Request) {

    fmt.Fprintf(w, "Hello world!")

}

谢谢!

Answer 1

使用一个闭包.

在app/c中将索引更改为:

func Index(db *sql.DB) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        // do stuff with db here
        fmt.Fprintf(w, "Hello world!")
    }
}

然后在你的主函数中使用它如下: r.HandleFunc("/", c.Index(db))

Index函数返回一个符合HandleFunc类型的匿名函数,并且还会关闭传递给db的值,使您的处理程序可以访问该变量.