使用 http.FileServer 处理自定义 404 页面

在使用 http.FileServer 提供静态网站时，为处理 404 错误自定义页面至关重要。本文将提供一种使用自定义中间件的解决方案，该中间件将首先尝试打开文件，如果文件不存在，则调用自定义 404 处理程序。否则，将请求分派给标准库中的静态文件服务器。这种方法允许您轻松创建自己的 404 页面，并为您的用户提供更好的用户体验。

我目前正在使用基本的 http.FileServer 设置来提供简单的静态站点。我需要使用自定义未找到页面处理 404 错误。我已经对这个问题进行了很多研究，但无法确定最好的解决方案是什么。

我看到了一些关于 GitHub 问题的回复，大致如下：

您可以实现自己的 ResponseWriter，它在 WriteHeader 之后写入自定义消息。

这似乎是最好的方法，但我有点不确定这实际上是如何实现的。如果有任何简单的实现示例，我们将不胜感激！

解决方案

我认为这个可以用你自己的中间件来解决。您可以尝试先打开该文件，如果它不存在，则调用您自己的 404 处理程序。否则只需将调用分派到标准库中的静态文件服务器。

看起来是这样的：

package main

import (
    "fmt"
    "net/http"
    "os"
    "path"
)

func notFound(w http.ResponseWriter, r *http.Request) {
    // Here you can send your custom 404 back.
    fmt.Fprintf(w, "404")
}

func customNotFound(fs http.FileSystem) http.Handler {
    fileServer := http.FileServer(fs)
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        _, err := fs.Open(path.Clean(r.URL.Path)) // Do not allow path traversals.
        if os.IsNotExist(err) {
            notFound(w, r)
            return
        }
        fileServer.ServeHTTP(w, r)
    })
}

func main() {
    http.ListenAndServe(":8080", customNotFound(http.Dir("/path/to/files")))
}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。