Go filepath.WalkDir 怎样只遍历指定后缀的文件

要在 Go 中递归遍历目录并只收集指定后缀的文件，最直接的做法是使用 filepath.WalkDir，在回调里先判断目录，再用 filepath.Ext 读取扩展名。目录直接返回 nil 继续遍历，文件则统一后缀大小写后加入结果集。

官方文档：https://pkg.go.dev/path/filepath

关键点只有三个：不要把目录误当成文件、不要用完整路径字符串硬截后缀、不要因为单个文件不匹配就终止整棵目录树。

先比较三种目录筛选方式

如果任务只是读取当前目录，os.ReadDir 的边界更清楚；如果是按一个简单通配模式寻找文件，filepath.Glob 更短；需要递归访问子目录并在访问过程中做条件判断时，filepath.WalkDir 更合适。

这里选择 WalkDir，是因为“递归”和“只收集部分文件”同时存在。它的回调接收路径、目录项和错误，既能识别目录，也能在文件进入结果集之前做筛选。

最小可用写法：目录继续走，文件按后缀收集

下面的函数收集目录树中的 Go 和 Markdown 文件。DirEntry.Name() 只取当前项名称，filepath.Ext 返回最后一个扩展名；用 strings.ToLower 统一大小写后，report.GO 和 report.go 会得到相同判断。

package main

import (
    "fmt"
    "path/filepath"
    "strings"
)

func collectFiles(root string, wanted map[string]struct{}) ([]string, error) {
    var files []string

    err := filepath.WalkDir(root, func(path string, d filepath.DirEntry, walkErr error) error {
        // 先处理当前项的访问错误，避免对无效的 DirEntry 继续取名称。
        if walkErr != nil {
            return walkErr
        }
        // 目录返回 nil，表示继续进入它的子项，不把目录加入文件结果。
        if d.IsDir() {
            return nil
        }

        // 只取最后一个扩展名，并统一大小写以兼容 .GO、.Go 等写法。
        ext := strings.ToLower(filepath.Ext(d.Name()))
        if _, ok := wanted[ext]; !ok {
            return nil
        }
        files = append(files, path)
        return nil
    })
    if err != nil {
        // WalkDir 的返回错误代表遍历没有完整结束，交给调用方决定重试或提示。
        return nil, fmt.Errorf("遍历 %q 失败: %w", root, err)
    }
    return files, nil
}

func main() {
    // 用集合表达允许的后缀，查找比在回调中反复比较字符串更直观。
    wanted := map[string]struct{}{`.go`: {}, `.md`: {}}
    files, err := collectFiles("./project", wanted)
    if err != nil {
        fmt.Println(err)
        return
    }
    for _, file := range files {
        fmt.Println(file)
    }
}

注意回调里返回 nil 的含义：对目录表示继续遍历，对不匹配的文件表示忽略当前项并继续访问后续项。只有返回非空错误时，遍历才会提前结束。

扩展名匹配应该怎样选

后缀筛选通常有三种写法，选择时要先明确是否区分大小写，以及是否只关心最后一个扩展名。

例如 archive.tar.gz 的 filepath.Ext 是 .gz。如果业务需要把完整的复合后缀当成一个规则，就不能只比较 Ext，而应先明确规则，或使用完整文件名做匹配。

目录和错误边界不要混在一起

WalkDir 回调中的第三个参数用于报告当前路径的访问错误。出现错误时，不能无条件调用 d.IsDir() 或 d.Name()，应先决定是否返回这个错误。

err := filepath.WalkDir(root, func(path string, d filepath.DirEntry, walkErr error) error {
    // 当前项无法读取时，直接保留错误上下文，不假设 d 一定可用。
    if walkErr != nil {
        return fmt.Errorf("读取 %s: %w", path, walkErr)
    }

    // 目录本身不参与后缀判断；返回 nil 让 WalkDir 继续深入。
    if d.IsDir() {
        return nil
    }

    // 文件不匹配时只跳过这一项，不返回错误也不使用 SkipDir。
    if strings.ToLower(filepath.Ext(d.Name())) != ".go" {
        return nil
    }
    fmt.Println("matched:", path)
    return nil
})
if err != nil {
    // 根目录不存在、权限问题或回调主动返回的错误都会汇总到这里。
    return err
}

filepath.SkipDir 只适合明确“不再进入当前目录”的情况，例如排除名为 vendor 的目录。它不是“不匹配文件”的返回值；对普通文件返回它可能让遍历语义变得难以理解。

按约束做最后选择

因此，标题中的问题可以归纳为：让 WalkDir 负责目录树访问，让 filepath.Ext 负责后缀提取，让回调返回值负责“继续、跳过当前项或停止遍历”。三者职责分开后，代码更容易扩展到大小写归一化、目录排除和多后缀集合。

相关问题

WalkDir 怎样跳过某个目录？在确认当前项是目录后返回 filepath.SkipDir，它只会停止进入当前目录。

为什么不直接用字符串后缀判断？完整路径可能包含同名片段，且大小写和复合扩展名会带来边界问题；使用 filepath.Ext 更容易表达规则。

筛选文件时需要调用 Info 吗？只按目录项类型和后缀筛选时通常不需要；只有大小、权限或修改时间参与规则时，才按需调用 d.Info() 并处理它的错误。