如何在Go的标准HTTP包中实现自定义404错误页面？

使用标准 HTTP 包显示自定义 404 错误页面

当用户请求不存在的 URL 时，HTTP 服务器通常会显示基本的“404 页面未找到”错误消息。为了提供更多信息或定制化的体验，有必要实现自定义错误页面处理程序。

Go 的标准 HTTP 包中，存在一种机制来处理所有未处理的请求并显示自定义错误页面。让我们深入研究如何实现这一点：

自定义错误处理函数

创建一个函数来处理自定义错误页面。此函数采用三个参数：

• w (http.ResponseWriter)：用于发送自定义错误页面的响应编写器。
• r (*http.Request): 不存在的请求对象URL。
• status (int)：错误的状态代码（通常为 http.StatusNotFound）。

func errorHandler(w http.ResponseWriter, r *http.Request, status int) {
    // Set the HTTP status code.
    w.WriteHeader(status)
    // Customize the error response for specific status codes.
    if status == http.StatusNotFound {
        fmt.Fprint(w, "Custom 404 error message")
    }
}

设置自定义错误Handler

错误处理函数需要链接到HTTP服务器。这是在 http.ListenAndServe 函数中完成的，该函数在指定端口上启动服务器。

http.ListenAndServe(":12345", nil)

将 nil 替换为处理所有请求的自定义 HTTP 请求处理程序。

http.ListenAndServe(":12345", new(http.ServeMux))

在 ServeMux 中，添加特定 URL 的原始路由处理程序。

mux := http.NewServeMux()
mux.HandleFunc("/smth/", smthHandler)
mux.HandleFunc("/", homeHandler)

http.ListenAndServe(":12345", mux)

最后，添加NotFoundHandler 到 ServeMux 来处理特定路由处理程序未处理的所有其他 URL。

mux.NotFoundHandler = http.HandlerFunc(errorHandler)

示例代码

实现上述方法的完整示例代码是如下所示：

package main

import (
    "fmt"
    "net/http"
)

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/smth/", smthHandler)
    mux.HandleFunc("/", homeHandler)
    mux.NotFoundHandler = http.HandlerFunc(errorHandler)

    http.ListenAndServe(":12345", mux)
}

func smthHandler(w http.ResponseWriter, r *http.Request) {
    if r.URL.Path != "/smth/" {
        errorHandler(w, r, http.StatusNotFound)
        return
    }
    fmt.Fprint(w, "Welcome to smth")
}

func homeHandler(w http.ResponseWriter, r *http.Request) {
    if r.URL.Path != "/" {
        errorHandler(w, r, http.StatusNotFound)
        return
    }
    fmt.Fprint(w, "Welcome home")
}

func errorHandler(w http.ResponseWriter, r *http.Request, status int) {
    w.WriteHeader(status)
    if status == http.StatusNotFound {
        fmt.Fprint(w, "Custom 404 error message")
    }
}

此代码定义特定 URL 的路由处理程序 (/smth/和 /) 并分配自定义错误处理函数 (errorHandler) 来处理所有其他未处理的请求。当用户尝试访问不存在的 URL 时，将显示自定义 404 错误页面，而不是默认的“404 页面未找到”消息。

以上是如何在Go的标准HTTP包中实现自定义404错误页面？的详细内容。更多信息请关注PHP中文网其他相关文章！