Подтвердить что ты не робот

Как передать аргументы моему обработчику

Я пытаюсь передать свой объект базы данных вместе с моими обработчиками вместо глобального объекта. Но я не знаю, возможно ли это, я использую пакет 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, поэтому, пожалуйста, объясните потенциальный ответ в деталях.

4b9b3361

Ответ 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("/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")