使用 Go 的自定义 404 错误消息

Custom 404 error message with Go

我有这段代码,如果我转到 /time/,它会给出我的自定义 404 错误消息,但如果我转到 /times/ 或只是 / 或 /whatever,那么我将收到默认的 404 错误消息。我想为除 /time/

之外的所有情况显示我的自定义 404
package main

import (
        "fmt"
        "time"
        "flag"
        "os"
        "net/http"
        )

const AppVersion = "timeserver version: 3.0"

func timeserver(w http.ResponseWriter, r *http.Request) {
    if r.URL.Path != "/time/" {
        NotFoundHandler(w, r)
        return
    }

    const layout = "3:04:05 PM"
    t := time.Now().Local()
    fmt.Fprint(w, "<html>\n")
    fmt.Fprint(w, "<head>\n")
    fmt.Fprint(w, "<style>\n")
    fmt.Fprint(w, "p {font-size: xx-large}\n")
    fmt.Fprint(w, "span.time {color: red}\n")
    fmt.Fprint(w, "</style>\n")
    fmt.Fprint(w, "</head>\n")
    fmt.Fprint(w, "<body>\n")
    //fmt.Fprint(w, "The time is now " + t.Format(layout))
    fmt.Fprint(w, "<p>The time is now <span class=\"time\">" + t.Format(layout) + "</span>.</p>\n")
    fmt.Fprint(w, "</body>\n")
    fmt.Fprint(w, "</html>\n")
}

func NotFoundHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprint(w, "custom 404")
}

func main() {
    version := flag.Bool("V", false, "prints current version")
    port := flag.String("port", "8080", "sets service port number")

    flag.Parse()
    if *version {
      fmt.Println(AppVersion)
      os.Exit(0)
    }

    http.HandleFunc("/time/", timeserver)
    http.ListenAndServe("localhost:" + *port, nil)
}

将这一行添加到主线:

http.HandleFunc("/", NotFoundHandler)

“/”的处理程序是 catchall 处理程序。

此外,您应该将处理程序修改为 return 404 状态:

func NotFoundHandler(w http.ResponseWriter, r *http.Request) {
   w.WriteHeader(http.StatusNotFound)
   fmt.Fprint(w, "custom 404")
}