本文介绍了如何使用 Go 语言的 `net/http` 包中的 `FileServer` 函数来提供静态文件，并重点讲解了如何通过中间件或自定义 `http.FileSystem` 的方式，限制客户端访问特定的文件类型，例如只允许访问 HTML 文件，保护 JavaScript 等其他资源不被直接访问。
在使用 Go 构建 Web 应用时，经常需要提供静态文件，例如 HTML、CSS、JavaScript 和图片等。net/http 包中的 FileServer 函数提供了一种简单的方式来完成这项任务。然而，在某些情况下，我们可能需要限制客户端对某些文件的访问，例如保护 JavaScript 代码不被直接查看。本文将介绍几种方法来实现这一目标。
使用中间件过滤文件
一种常见的方法是使用中间件来过滤客户端的请求。中间件本质上是一个 HTTP Handler，它在请求到达实际的处理函数之前执行。我们可以创建一个中间件，检查请求的文件名是否符合特定的模式，如果不符合，则返回 404 Not Found 错误。
下面是一个示例代码：
package mainimport ( "log" "net/http" "path/filepath")func GlobFilterHandler(h http.Handler, pattern string) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { path := r.URL.Path fileName := filepath.Base(path) ok, err := filepath.Match(pattern, fileName) if err != nil { log.Println("Error in pattern match:", err) http.Error(w, http.StatusInternalServerError) // 更准确的错误处理 return } if !ok { http.NotFound(w, r) return } h.ServeHTTP(w, r) })}func main() { fileHandler := http.FileServer(http.Dir("static")) // 将目录改为 "static" wrappedHandler := GlobFilterHandler(fileHandler, "*.html") // 只允许访问 HTML 文件 http.Handle("/", wrappedHandler) // 使用 http.Handle 注册处理程序 log.Fatal(http.ListenAndServe(":8080", nil)) // 使用 log.Fatal 处理错误}
在这个例子中，GlobFilterHandler 函数创建了一个中间件，它接受一个 HTTP Handler 和一个模式作为参数。该中间件检查请求的文件名是否符合指定的模式，如果符合，则将请求传递给原始的 HTTP Handler，否则返回 404 Not Found 错误。
在 main 函数中，我们首先创建了一个 FileServer，它将 static 目录作为根目录。然后，我们使用 GlobFilterHandler 函数将 FileServer 包装起来，只允许访问 *.html 文件。最后，我们将包装后的 HTTP Handler 注册到根路径 /。
注意事项：
确保 static 目录存在，并且包含 index.html 文件。filepath.Match 函数的模式匹配语法与 glob 类似。错误处理：在模式匹配出错时，应返回 http.StatusInternalServerError 错误，而不是直接调用 log.Println。注册处理程序：使用 http.Handle 函数将处理程序注册到指定的路径。错误处理：使用 log.Fatal 处理 http.ListenAndServe 返回的错误，确保程序在监听失败时能够正确退出。
自定义 http.FileSystem
另一种方法是自定义 http.FileSystem 接口的实现。http.FileSystem 接口定义了 Open 方法，该方法用于打开指定的文件。我们可以创建一个自定义的 http.FileSystem 实现，在 Open 方法中检查文件名是否符合特定的模式，如果不符合，则返回错误。
下面是一个示例代码：
package mainimport ( "fmt" "log" "net/http" "path/filepath")type GlobDir struct { Dir http.Dir Pattern string}func (d GlobDir) Open(name string) (http.File, error) { baseName := filepath.Base(name) if ok, err := filepath.Match(d.Pattern, baseName); !ok || err != nil { if err != nil { return nil, err } return nil, fmt.Errorf("%s does not match GlobDir pattern.", baseName) } return d.Dir.Open(name)}func main() { fileHandler := http.FileServer(GlobDir{ Dir: http.Dir("static"), // 将目录改为 "static" Pattern: "*.html", // 只允许访问 HTML 文件 }) http.Handle("/", fileHandler) // 使用 http.Handle 注册处理程序 log.Fatal(http.ListenAndServe(":8080", nil)) // 使用 log.Fatal 处理错误}
在这个例子中，GlobDir 结构体实现了 http.FileSystem 接口。Open 方法首先获取文件名，然后检查文件名是否符合指定的模式。如果符合，则调用原始的 http.Dir 的 Open 方法打开文件，否则返回错误。
在 main 函数中，我们创建了一个 GlobDir 实例，并将 static 目录和 *.html 模式传递给它。然后，我们使用 http.FileServer 函数创建一个 HTTP Handler，并将 GlobDir 实例作为参数传递给它。最后，我们将 HTTP Handler 注册到根路径 /。
注意事项：
与第一种方法类似，确保 static 目录存在，并且包含 index.html 文件。GlobDir 结构体需要实现 http.FileSystem 接口的所有方法，即使只重写了 Open 方法。注册处理程序：使用 http.Handle 函数将处理程序注册到指定的路径。错误处理：使用 log.Fatal 处理 http.ListenAndServe 返回的错误，确保程序在监听失败时能够正确退出。
总结
本文介绍了两种方法来限制 FileServer 提供的静态文件的访问。第一种方法是使用中间件来过滤请求，第二种方法是自定义 http.FileSystem 接口的实现。选择哪种方法取决于具体的需求。如果只需要简单的文件过滤，可以使用中间件。如果需要更复杂的逻辑，可以自定义 http.FileSystem 接口的实现。 无论选择哪种方法，都需要仔细考虑安全性，确保只有授权用户才能访问敏感文件。
以上就是使用 Go FileServer 安全地提供静态文件的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1420566.html
微信扫一扫 
支付宝扫一扫 
