Go filepath.WalkDir 如何按扩展名统计文件而不跟随链接

如果要统计一个目录树里有多少个 .go、.md 或 .json 文件，Go 里直接用 filepath.WalkDir 就够了。它默认按字典序遍历，并且不会跟随目录中的符号链接；回调里只要先处理错误、跳过目录，再判断普通文件和扩展名即可。需要注意的是：如果传入的 root 本身就是符号链接，WalkDir 会遍历它指向的目标。

核心写法是“err → 目录 → 符号链接 → 普通文件 → 扩展名”。不要用 Info() 的目标属性替代 DirEntry 本身的类型判断，否则很容易把统计范围写宽。

先把 WalkDir 的边界看清楚

WalkDir 会回调根节点本身，之后按操作系统的路径分隔符传入每个子项。目录内出现名为 assets-link 的符号链接时，它会被访问为一个目录项，但不会继续进入链接目标。因此，统计函数不需要自己维护“已访问路径”来防止链接目录循环。

不过“遇到链接不跟随”和“完全忽略链接项”是两件事。若链接名以 .go 结尾，想统计真实普通文件数量时也应排除它；若业务需要把链接作为条目计数，则应另设指标，而不是把它混入普通文件统计。

按扩展名统计普通文件的最小实现

下面的函数把目标扩展名统一转成小写，返回值用 map[string]int 保存计数。它不读取文件内容，也不调用 Info()，所以扫描大量文件时只做目录项级别的判断。

package main

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

func countExtensions(root string) (map[string]int, error) {
	counts := make(map[string]int)
	err := filepath.WalkDir(root, func(path string, d fs.DirEntry, err error) error {
		// 访问目录或文件失败时保留原始错误，调用方才能决定是否降级。
		if err != nil {
			return err
		}
		// 目录只负责继续遍历，不参与扩展名统计。
		if d.IsDir() {
			return nil
		}
		// 明确排除目录中的符号链接，避免把链接名当成普通文件。
		if d.Type()&fs.ModeSymlink != 0 || !d.Type().IsRegular() {
			return nil
		}
		ext := strings.ToLower(filepath.Ext(d.Name()))
		if ext != "" {
			counts[ext]++
		}
		return nil
	})
	if err != nil {
		return nil, fmt.Errorf("scan %q: %w", root, err)
	}
	return counts, nil
}

这里用 d.Name() 取最后一级文件名，再交给 filepath.Ext。例如 main.test.go 的扩展名是 .go，没有点的文件返回空字符串。strings.ToLower 让 README.GO 与 readme.go 归到同一组；如果业务要求大小写敏感，可以去掉这一步。

为什么“不跟随链接”仍要检查类型

WalkDir 的“不跟随”是遍历行为，不代表回调不会看到链接项。回调仍可能收到一个 DirEntry，因此统计逻辑应该主动用 d.Type()&fs.ModeSymlink 排除它。DirEntry.Info() 对符号链接描述的是链接本身，不是目标；但这一步会产生额外的属性读取，没有必要为了扩展名统计而调用。

如果只想排除某个目录，例如缓存目录，可以在目录分支里返回 filepath.SkipDir。它只跳过当前目录；对普通文件返回这个值时，语义是跳过该文件的父目录，通常不是你想要的行为。

迁移和回归时检查这几个细节

从旧的 filepath.Walk 迁移到 WalkDir 时，回调参数从 fs.FileInfo 变成了 fs.DirEntry。只统计文件名和类型时，新的写法更直接；需要大小、修改时间等属性时，再按需调用 d.Info()，并处理它可能返回的错误。

• 准备一个普通文件、一个无后缀文件、一个大写后缀文件和一个目录内符号链接。
• 确认链接目标中的文件没有被计入，根路径本身是链接时单独记录预期。
• 在无权限目录或扫描期间文件被删除时，确认错误策略符合业务：严格返回，或只对明确允许的错误记录后继续。
• 在 Windows 上不要手工拼接斜杠；传给 WalkDir 的路径由 filepath 负责平台适配。

常见问题

filepath.Ext 能识别多个后缀吗？

它只返回最后一个点开始的后缀，所以 archive.tar.gz 返回 .gz。需要识别复合后缀时，应对文件名做单独规则判断。

可以用 d.Info().IsDir() 代替 d.IsDir() 吗？

可以，但会增加一次信息读取，而且处理符号链接时更容易把目标语义和链接语义混在一起。只判断目录优先使用 d.IsDir()。

统计空扩展名文件应该怎么做？

把 ext != "" 分支改成单独的 counts[""]++，并为这个键定义清楚含义，例如“无后缀普通文件”，不要与未知类型混用。