如何组织 gorilla mux 路由?

我正在使用Gorilla Mux编写 REST API,但在组织我的路线时遇到问题,目前我所有的路线都在这样的main.go文件中定义


//main.go

package main


import (

    "NovAPI/routes"

    "fmt"

    "github.com/gorilla/mux"

    "net/http"

)


func main() {


    router := mux.NewRouter().StrictSlash(true)


    router.HandleFunc("/hello", func(res http.ResponseWriter, req *http.Request) {

        fmt.Fprintln(res, "Hello")

    })


    router.HandleFunc("/user", func(res http.ResponseWriter, req *http.Request) {

        fmt.Fprintln(res, "User")

    })


    router.HandleFunc("/route2", func(res http.ResponseWriter, req *http.Request) {

        fmt.Fprintln(res, "Route2")

    })


    router.HandleFunc("/route3", func(res http.ResponseWriter, req *http.Request) {

        fmt.Fprintln(res, "Route3")

    })


    // route declarations continue like this


    http.ListenAndServe(":1128", router)


}

所以我想要做的是取出这个路由声明并将其拆分为多个文件,我该怎么做呢?提前致谢。


芜湖不芜
浏览 186回答 3
3回答

皈依舞

您可以将您的路由器独立模块化成不同的包,并将它们安装在主路由器上详细说明以下问题,您可以提出这种方法,这使其具有相当的可扩展性(并且在某种程度上更易于测试)/api/router.gopackage apiimport (    "net/http"    "github.com/gorilla/mux")func Router() *mux.Router {    router := mux.NewRouter()    router.HandleFunc("/", home)    return router}func home(w http.ResponseWriter, req *http.Request) {    w.Write([]byte("hello from API"))}/main.gopackage mainimport (    "log"    "net/http"    "strings"    "github.com/...yourPath.../api"    "github.com/...yourPath.../user"    "github.com/gorilla/mux")func main() {    router := mux.NewRouter()    router.HandleFunc("/", home)    mount(router, "/api", api.Router())    mount(router, "/user", user.Router())    log.Fatal(http.ListenAndServe(":8080", router))}func mount(r *mux.Router, path string, handler http.Handler) {    r.PathPrefix(path).Handler(        http.StripPrefix(            strings.TrimSuffix(path, "/"),            handler,        ),    )}func home(w http.ResponseWriter, req *http.Request) {    w.Write([]byte("Home"))}

12345678_0001

这样的事情怎么办?//main.gopackage mainimport (    "NovAPI/routes"    "fmt"    "github.com/gorilla/mux"    "net/http")func main() {    router := mux.NewRouter().StrictSlash(true)    router.HandleFunc("/hello", HelloHandler)    router.HandleFunc("/user", UserHandler)    router.HandleFunc("/route2", Route2Handler)    router.HandleFunc("/route3", Route3Handler)    // route declarations continue like this    http.ListenAndServe(":1128", router)}func HelloHandler(res http.ResponseWriter, req *http.Request) {    fmt.Fprintln(res, "Hello")}func UserHandler(res http.ResponseWriter, req *http.Request) {    fmt.Fprintln(res, "User")}func Route2Handler(res http.ResponseWriter, req *http.Request) {    fmt.Fprintln(res, "Route2")}func Route3Handler(res http.ResponseWriter, req *http.Request) {    fmt.Fprintln(res, "Route3")}这样您就可以将处理程序放在其他文件中,甚至其他包中。如果您最终获得了像数据库这样的附加依赖项,您甚至可以使用构造函数技巧来避免需要全局变量://main.gofunc main() {    db := sql.Open(…)    //...    router.HandleFunc("/hello", NewHelloHandler(db))    //...}func NewHelloHandler(db *sql.DB) func(http.ResponseWriter, *http.Request) {    return func(res http.ResponseWriter, req *http.Request) {        // db is in the local scope, and you can even inject it to test your        // handler        fmt.Fprintln(res, "Hello")    }}

凤凰求蛊

我喜欢查看 github 中的其他项目以获取有关如何做事的想法,对于这些情况,我通常首先查看Docker repo。他们的做法是这样的:对于系统的路由,在system_routes.go 中定义所有处理程序,然后在system.go 中的 NewRouter 函数上初始化这些路由。type systemRouter struct {    backend Backend    routes  []router.Route}func NewRouter(b Backend) router.Router {    r := &systemRouter{        backend: b,    }    r.routes = []router.Route{        local.NewOptionsRoute("/", optionsHandler),        local.NewGetRoute("/_ping", pingHandler),        local.NewGetRoute("/events", r.getEvents),        local.NewGetRoute("/info", r.getInfo),        local.NewGetRoute("/version", r.getVersion),        local.NewPostRoute("/auth", r.postAuth),    }    return r}// Routes return all the API routes dedicated to the docker system.func (s *systemRouter) Routes() []router.Route {    return s.routes}请注意,systemRouter 实现了router.Router接口,Routes 函数返回一个 []router.Route,它们的处理程序定义为func(ctx context.Context, w http.ResponseWriter, r *http.Request, vars map[string]string) error而不是 Go 的标准 http 处理程序:func(w http.ResponseWriter, r *http.Request)所以他们有额外的代码可以在makeHttpHandler函数中将Docker API 处理程序转换为 Go HTTP 处理程序。最后,为了将这些路由添加到他们的 mux 路由器,在他们的server.go 上,他们实现了其他几个功能来将中间件添加到他们的处理程序中。如果您认为这是您正在寻找的东西,那么请花时间分析 Docker 代码的路线,如果您需要我详细说明或遗漏任何内容,请发表评论。
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Go