2

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

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

}

谢谢!

4

1 回答 1

11

使用闭包。

在 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 的值。

于 2013-07-22T03:01:00.547 回答