GoLang Filepath.Walk遍历优化

原生标准库在文件量过大时效率和内存均表现不好

1400万文件遍历Filepath.Walk

1400万文件重写直接调用windows api并处理细节

结论

1400万文件遍历时对比

对比条目 filepath.walk windows api并触发黑科技
运行时间 710秒 22秒
内存占用 480M 38M

关键代码

go 复制代码
//超级快的文件遍历
func FindFileWin(dir string, callbackfunc MyFindFileCallBack) {

	dir = dir + `\`

	finstruct := win.WIN32_FIND_DATAW{}
	handle := win.FindFirstFileW(dir+`*`, &finstruct)
	if win.IsInvalidHandle(handle) {

		for {
			//文件夾
			if (finstruct.DwFileAttributes & win.FILE_ATTRIBUTE_DIRECTORY) != 0 {
				if (finstruct.CfileNameGo != "..") && (finstruct.CfileNameGo != ".") {
					FindFileWin(dir+finstruct.CfileNameGo, callbackfunc)
				}
			} else {
				callbackfunc(dir + finstruct.CfileNameGo)
			}
			if win.FindNextFileW(handle, &finstruct) == 0 {
				break
			}
		}
	}
}
go 复制代码
var (
	kernel32           = syscall.NewLazyDLL("Kernel32.dll")
	procCreateFileW    = kernel32.NewProc("CreateFileW")
	procOpenEventW     = kernel32.NewProc("OpenEventW")
	procSetEvent       = kernel32.NewProc("SetEvent")
	procFindFirstFileW = kernel32.NewProc("FindFirstFileW")
	procFindNextFileW  = kernel32.NewProc("FindNextFileW")

	//procCreateFileA = kernel32.NewProc("CreateFileA")
)

func FindFirstFileW(fileName string, lpFindFileData *WIN32_FIND_DATAW) HANDLE {
	strname := unsafe.Pointer(syscall.StringToUTF16Ptr(fileName))
	handle, _, _ := procFindFirstFileW.Call(
		uintptr(strname),
		uintptr(unsafe.Pointer(lpFindFileData)),
	)
	if handle != 0 {

		lpFindFileData.CfileNameGo = syscall.UTF16ToString(lpFindFileData.cFileName[:])
	}
	return HANDLE(handle)
}
func IsInvalidHandle(handle HANDLE) bool {
	if handle != 0 && int(handle) != -1 {
		return true
	}
	return false
}
func FindNextFileW(hFindFile HANDLE, lpFindFileData *WIN32_FIND_DATAW) BOOL {
	ret, _, _ := procFindNextFileW.Call(
		uintptr(hFindFile),
		uintptr(unsafe.Pointer(lpFindFileData)),
	)
	if ret != 0 {

		lpFindFileData.CfileNameGo = syscall.UTF16ToString(lpFindFileData.cFileName[:])
	}
	return BOOL(ret)
}
相关推荐
编程零零七11 分钟前
Python巩固训练——第一天练习题
开发语言·python·python基础·python学习·python练习题
我爱Jack23 分钟前
时间与空间复杂度详解:算法效率的度量衡
java·开发语言·算法
米饭「」25 分钟前
C++AVL树
java·开发语言·c++
开心就好202532 分钟前
WebView远程调试全景指南:实战对比主流工具优劣与适配场景
后端
用户214118326360238 分钟前
AI 一键搞定!中医药科普短视频制作全流程
后端
SimonKing1 小时前
告别传统读写!RandomAccessFile让你的Java程序快人一步
java·后端·程序员
心愿许得无限大1 小时前
Qt 常用界面组件
开发语言·c++·qt
2401_858286111 小时前
OS15.【Linux】gdb调试器的简单使用
linux·运维·服务器·开发语言·gdb
牛马baby1 小时前
MATLAB下载安装教程(附安装包)2025最新版(MATLAB R2024b)
开发语言·matlab
shenyan~1 小时前
关于 c、c#、c++ 三者区别
开发语言·c++