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!")
    }
}

然后在你的main函数中使用它：r.HandleFunc("/", c.Index(db))

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

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

1 个答案: