go - 在 gorilla mux 路由器上设置 404 NotFound 处理程序

Question

这是我用 Go 语言和 gorilla mux 包编写的一个小型演示网络服务器的代码:

package main

import (
    "fmt"
    "github.com/gorilla/mux"
    "net/http"
)

func handler(w http.ResponseWriter, r *http.Request) {
    vars := mux.Vars(r)
    fmt.Fprintf(w, "Hi there, I love %s!", vars["username"])
}

func homeHandler(w http.ResponseWriter, r *http.Request) {
    if r.URL.Path != "/" {
        errorHandler(w, r, http.StatusNotFound)
        return
    }
    vars := mux.Vars(r)
    fmt.Fprintf(w, "Hi there, I love %s!", vars["username"])
}

func main() {
    r := mux.NewRouter()
    r.HandleFunc("/help/{username}/", handler)
    http.Handle("/", r)
    http.ListenAndServe(":8080", nil)
}

但我没有找到实现自定义 404 页面的方法。

但是我不能做一个 r.HandleFunc("/",...) 但是它太贪心了。

Answer 1

Router导出一个 NotFoundHandler 字段，您可以将其设置为自定义处理程序。

r := mux.NewRouter()
r.NotFoundHandler = MyCustom404Handler