如何在Go中使用http.FileServer禁用目录列表功能？

好问题！默认情况下，Go的http.FileServer在访问没有index文件的目录时，会返回该目录的文件列表——这在很多生产环境的Web服务中是需要禁用的。下面给你两种常用且恰当的实现方式：

方法一：包装自定义FileSystem（推荐） #

这个方法最符合Go的接口设计思路，我们通过包装http.FileSystem，拦截目录请求：当访问的是目录且不存在指定的index文件时，返回os.ErrNotExist，让http.FileServer自动返回404响应。

具体代码实现如下：

首先定义一个包装结构体，嵌入原http.FileSystem：

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

type noDirListFS struct {
    http.FileSystem
}

// 重写Open方法
func (fs noDirListFS) Open(name string) (http.File, error) {
    // 先调用原FileSystem的Open方法
    f, err := fs.FileSystem.Open(name)
    if err != nil {
        return nil, err
    }

    // 获取文件/目录的状态信息
    stat, err := f.Stat()
    if err != nil {
        f.Close()
        return nil, err
    }

    // 如果是目录，检查是否存在index.html（你可以换成自己的index文件名，比如index.htm）
    if stat.IsDir() {
        _, err := fs.FileSystem.Open(path.Join(name, "index.html"))
        if err != nil {
            // 没有index文件，关闭原文件并返回不存在的错误
            f.Close()
            return nil, os.ErrNotExist
        }
    }

    // 一切正常，返回原文件
    return f, nil
}

然后修改你的main函数中的http.Handle部分：

http.Handle("/", http.FileServer(noDirListFS{http.Dir(*directory)}))

这样一来，当用户访问没有index文件的目录时，就会得到404响应，而不是文件列表。

方法二：用自定义Handler包装 #

如果你需要更灵活的控制（比如返回403禁止访问而不是404），可以写一个自定义的Handler来拦截目录请求：

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

func noDirListWrapper(next http.Handler, dir string) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 尝试打开请求路径对应的文件/目录
        file, err := http.Dir(dir).Open(r.URL.Path)
        if err == nil {
            stat, _ := file.Stat()
            if stat.IsDir() {
                // 检查目录下是否有index文件
                _, indexErr := http.Dir(dir).Open(path.Join(r.URL.Path, "index.html"))
                if indexErr != nil {
                    // 没有index文件，返回403禁止访问
                    w.WriteHeader(http.StatusForbidden)
                    fmt.Fprintf(w, "Directory listing is not allowed")
                    file.Close()
                    return
                }
            }
            file.Close()
        }
        // 正常请求，交给原FileServer处理
        next.ServeHTTP(w, r)
    })
}

然后在main函数中使用：

fileServer := http.FileServer(http.Dir(*directory))
http.Handle("/", noDirListWrapper(fileServer, *directory))

总结 #

• 方法一更简洁优雅，符合Go的接口扩展方式，推荐用于大多数场景；
• 方法二更灵活，适合需要自定义响应内容或状态码的场景。

内容的提问来源于stack exchange，提问作者Tobias