登录
首页 >  Golang >  Go问答

如何将参数传递给我的处理程序

来源:Golang技术栈

时间:2023-04-28 06:48:53 487浏览 收藏

本篇文章向大家介绍《如何将参数传递给我的处理程序》,主要包括golang,具有一定的参考价值,需要的朋友可以参考一下。

问题内容

我试图将我的数据库对象传递给我的处理程序,而不是拥有一个全局对象。但我不知道这是否可能,我正在使用 Gorilla Mux 包,我可以看到它需要一个闭包作为第二个参数。

// https://github.com/gorilla/mux/blob/master/mux.go#L174
// HandleFunc registers a new route with a matcher for the URL path.
// See Route.Path() and Route.HandlerFunc().
func (r *Router) HandleFunc(path string, f func(http.ResponseWriter,
    *http.Request)) *Route {
    return r.NewRoute().Path(path).HandlerFunc(f)
}

然后定义了我可以使用的参数,理想情况下我希望有第三个这样的参数。

// In my main
router.HandleFunc("/users/{id}", showUserHandler).Methods("GET")

func showUserHandler(w http.ResponseWriter, r *http.Request, db *gorm.DB) {
    fmt.Fprintf(w, "We should fetch the user with id %s", vars["id"])
}

有解决方法吗?还是我需要一个全局 db 对象?我是 Go 新手,所以请详细解释一个可能的答案。

正确答案

欢迎来到 Go。

具有全局变量和特别是数据库对象是可以接受的。

但是,如果您不想这样做,则有几种解决方法,例如,您可以创建一个结构并showHandler在其上定义您的结构。

type Users struct {
    db *gorm.DB
}

func (users *Users) showHandler(w http.ResponseWriter, r *http.Request) {
    //now you can use users.db
}
func (users *Users) addHandler(w http.ResponseWriter, r *http.Request) {
    //now you can use users.db
}

// setup
users := &Users{db: createDB()}
router.HandleFunc("/users/{id}", users.showHandler).Methods("GET")
router.HandleFunc("/users/new", users.addHandler)
//etc

另一种方法是创建一个包装函数:

db := createDB()
router.HandleFunc("/users/{id}", func(w http.ResponseWriter, r *http.Request) {
    showUserHandler(w, r, db)
}).Method("GET")

今天关于《如何将参数传递给我的处理程序》的内容就介绍到这里了,是不是学起来一目了然!想要了解更多关于golang的内容请关注golang学习网公众号!

声明:本文转载于:Golang技术栈 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>