您可以轻松地限制FileServer
, 这是 aHttpHandler
通过将另一个包裹HttpHandler
在它周围。例如,使用这个只允许提供*.js
文件的包装器:
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)
if ok, err := filepath.Match(pattern, fileName); !ok || err != nil {
if err != nil {
log.Println("Error in pattern match:", err)
}
http.NotFound(w, r)
return
}
h.ServeHTTP(w, r)
})
}
func main() {
fileHandler := http.FileServer(http.Dir("/tmp/dtest"))
wrappedHandler := GlobFilterHandler(fileHandler, "*.js")
}
你可以在这里找到一篇博客文章,它很好地描述了基本思想。
您拥有的另一个选择是扩展http.Dir
并制作您自己的http.FileSystem
实现,这正是您想要的:
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 not match GlobDir pattern.", baseName)
}
return d.Dir.Open(name)
}
func main() {
fileHandler := http.FileServer(GlobDir{
Dir: http.Dir("/tmp/dtest"),
Pattern: "*.js",
})
http.ListenAndServe(":8080", fileHandler)
}
第二种解决方案实现了http.FileSystem
被http.FileServer
. 它检查输入文件名是否与提供的模式匹配,然后将控制权交给原始http.Dir
. 这可能是你想去这里的方式。