Golang使用gorilla/mux实现路由管理

使用gorilla/mux可构建灵活路由，支持参数解析、HTTP方法限制、子路由及自定义匹配。通过r.NotFoundHandler可处理404错误，结合httptest可进行路由测试。

使用

gorilla/mux

gorilla/mux

解决方案

1. 安装

   gorilla/mux

   登录后复制
   :

   立即学习“go语言免费学习笔记（深入）”；

   首先，你需要获取

   gorilla/mux

   登录后复制
   包。在你的项目目录下，运行以下命令：

   go get github.com/gorilla/mux

   登录后复制

2. 基本路由:

   package main
   
   import (
       "fmt"
       "log"
       "net/http"
   
       "github.com/gorilla/mux"
   )
   
   func homeHandler(w http.ResponseWriter, r *http.Request) {
       fmt.Fprintln(w, "Welcome Home!")
   }
   
   func main() {
       r := mux.NewRouter()
       r.HandleFunc("/", homeHandler)
   
       log.Fatal(http.ListenAndServe(":8080", r))
   }

   登录后复制

   这段代码创建了一个新的

   mux.Router

   登录后复制
   实例，并将根路径

   /

   登录后复制
   映射到

   homeHandler

   登录后复制
   函数。

3. 带参数的路由:

   gorilla/mux

   登录后复制
   允许你在 URL 中定义参数，并在处理函数中访问它们。

   func articleHandler(w http.ResponseWriter, r *http.Request) {
       vars := mux.Vars(r)
       articleID := vars["id"]
   
       fmt.Fprintf(w, "Article ID: %s\n", articleID)
   }
   
   func main() {
       r := mux.NewRouter()
       r.HandleFunc("/articles/{id}", articleHandler)
   
       log.Fatal(http.ListenAndServe(":8080", r))
   }

   登录后复制

   在这里，

   {id}

   登录后复制
   是一个路由变量。你可以通过

   mux.Vars(r)

   登录后复制
   获取所有变量，然后使用键名访问特定的变量。

4. HTTP 方法限制:

   你可以限制路由只响应特定的 HTTP 方法。

   

   工资查查移动工资条

   大部分的工资还是以打印工资条的形式进行，偶有公司使用邮件发放工资条，而工资条的现代形式应该是移动工资条，以实现信息的备忘、到达、管理、对帐、环保、高效等需求……，用户已经习惯使用手机（或以其它移动方式）实现一切需求，应用的移动化是大势所趋。工资查查就在这样的背景下诞生，北京亦卓科技于2017的开发并推出了微信小程序工资查查。由于对有用户对数据隐私与安全性的考虑，北京亦卓科技在推出了云端应用--工资

   0

   查看详情

   func createHandler(w http.ResponseWriter, r *http.Request) {
       fmt.Fprintln(w, "Creating a new resource...")
   }
   
   func main() {
       r := mux.NewRouter()
       r.HandleFunc("/resources", createHandler).Methods("POST")
   
       log.Fatal(http.ListenAndServe(":8080", r))
   }

   登录后复制

   这个路由只会在收到

   POST

   登录后复制
   请求时被触发。

5. 使用子路由:

   子路由允许你创建更复杂的路由结构。

   func productsHandler(w http.ResponseWriter, r *http.Request) {
       fmt.Fprintln(w, "All products...")
   }
   
   func productHandler(w http.ResponseWriter, r *http.Request) {
       vars := mux.Vars(r)
       productID := vars["id"]
       fmt.Fprintf(w, "Product ID: %s\n", productID)
   }
   
   func main() {
       r := mux.NewRouter()
       productsRouter := r.PathPrefix("/products").Subrouter()
       productsRouter.HandleFunc("/", productsHandler)
       productsRouter.HandleFunc("/{id}", productHandler)
   
       log.Fatal(http.ListenAndServe(":8080", r))
   }

   登录后复制

   所有以

   /products

   登录后复制
   开头的 URL 都会被路由到

   productsRouter

   登录后复制
   ，然后根据剩余的路径进行进一步的路由。

如何处理

gorilla/mux

当

gorilla/mux

NotFoundHandler

func notFoundHandler(w http.ResponseWriter, r *http.Request) {
    w.WriteHeader(http.StatusNotFound)
    fmt.Fprintln(w, "Custom 404 - Page not found!")
}

func main() {
    r := mux.NewRouter()
    r.HandleFunc("/", homeHandler)
    r.NotFoundHandler = http.HandlerFunc(notFoundHandler)

    log.Fatal(http.ListenAndServe(":8080", r))
}

通过设置

r.NotFoundHandler

gorilla/mux

除了基本的路由功能，

gorilla/mux

• Host 路由: 根据请求的 Hostname 匹配路由。
• Scheme 路由: 根据请求的 Scheme (http 或 https) 匹配路由。
• 自定义匹配器: 你可以编写自己的匹配器函数，实现更复杂的路由逻辑。
• 中间件:

  gorilla/mux

  登录后复制
  可以与中间件集成，实现请求预处理和后处理。

如何测试

gorilla/mux

测试路由是确保你的应用正常工作的重要部分。你可以使用

net/http/httptest

import (
    "net/http"
    "net/http/httptest"
    "testing"

    "github.com/gorilla/mux"
)

func TestHomeHandler(t *testing.T) {
    r := mux.NewRouter()
    r.HandleFunc("/", homeHandler)

    req, err := http.NewRequest("GET", "/", nil)
    if err != nil {
        t.Fatal(err)
    }

    rr := httptest.NewRecorder()
    r.ServeHTTP(rr, req)

    if status := rr.Code; status != http.StatusOK {
        t.Errorf("handler returned wrong status code: got %v want %v",
            status, http.StatusOK)
    }

    expected := "Welcome Home!\n"
    if rr.Body.String() != expected {
        t.Errorf("handler returned unexpected body: got %v want %v",
            rr.Body.String(), expected)
    }
}

这个例子展示了如何创建一个模拟的 HTTP 请求，然后使用

r.ServeHTTP

以上就是Golang使用gorilla/mux实现路由管理的详细内容，更多请关注php中文网其它相关文章！