如何將參數傳遞給我的處理程序

Question

我想將我的數據庫對象傳遞給我的處理程序，而不是具有全局對象。但我不知道這是否可行，我使用的是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("https://stackoverflow.com/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"]) 
} 

是否有解決方法？或者我需要一個全局數據庫對象？我是Go的新手，所以請詳細解釋一下可能的答案。

Answer 1

歡迎使用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("https://stackoverflow.com/users/{id}", users.showHandler).Methods("GET") 
router.HandleFunc("https://stackoverflow.com/users/new", users.addHandler) 
//etc 

另一種方法是創建一個包裝函數：

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