go - 使用 Golang Mux Router 和 http.FileServer 预期的根文件和自定义 404

Question

我正在用 Go 构建一个小型网站，但遇到了自定义 404 页面的问题。这是我当前的路由器代码:

r := mux.NewRouter()

// Page routing
r.HandleFunc("/", homeHandler).Methods("GET")
r.HandleFunc("/example1", exampleOneHandler).Methods("GET")
r.HandleFunc("/example2", exampleTwoHandler).Methods("GET", "POST")

// Other paths for static assets omitted

// Psuedo-root directory for icon files, etc.
r.PathPrefix("/").Handler(http.FileServer(http.Dir("public")))

// 404 Page
r.NotFoundHandler = http.HandlerFunc(notFoundHandler)

log.Fatal(http.ListenAndServe(":1234", r))

因此，如果第 11 行没有 PathPrefix 定义，notFoundHandler 将按预期命中并返回自定义 404 HTML。

但是，现在我添加了该路径定义来满足根目录标准，例如 favicon.ico、robots.txt、应用程序图标等，并且可以按预期工作。然而，作为副作用，任何与/example1 或/example2 不匹配的内容(比如/example3)都会通过 http.FileServer 在 psuedo-root 目录中查找名为“example3”的文件。找不到之后，FileServer 直接向 http.ResponseWriter 写入“404 not found”，完全绕过 mux 的 NotFoundHandler。

我能看到的唯一可行的解​​决方案是为每个文件添加一个显式路由，但这似乎是一个相当残酷的解决方案。有没有更优雅的方法来解决我所缺少的这个问题？

Answer 1

对于没有注册处理程序的路径，将激活 notFoundHandler。由于您向路径“/”注册了一个处理程序(文件服务器)，因此该处理程序将始终匹配所有请求，因此不会再调用未找到的处理程序。

要避免这种情况，最简单的方法是手动注册必须位于根目录的所有文件(这些文件并不多，或者至少不应该)。如果您有一堆其他非标准静态文件，请将它们发布到 /static/ 等路径下，因此未注册的路径(例如 /example3)将触发 not-found处理程序。

另一个解决方案可能是将文件服务器保留在路径 "/" 处，但将其包装起来，如果它返回 404 状态代码，请调用您自己的 notFoundHandler 来提供自定义错误页面。与我之前的建议相比，这更复杂，所以我仍然会选择那个。