一个文件遍历小函数的设计

之前写过 关于 win32 下磁盘的遍历方法，URL：http://blog.csdn.net/magictong/archive/2008/08/07/2784420.aspx，下面是将里面实现的磁盘遍历函数重新设计了一下，因为磁盘遍历，文件夹遍历在很多时候还是很常见的，为了不修改遍历的函数体，我将处理文件的部分交由一个回调函数去处理，这样的话，你甚至可以把下面的代码写在dll里面，需要进行文件遍历的地方，只要实现一个回调函数传进去就可以了。

 

函数声明：

1. // ------------------------------------------------------------------------- 
2. // 函数       : ScanDirectory 
3. // 功能       : 遍历一个目录，传入一个函数指针 
4. // 返回值  : BOOL  
5. // 参数       : const TCHAR *pszPath                      传入路径 
6. // 参数       : LPSCANDISK_START_ROUTINE lpFunAddress     函数指针回调指针 
7. // 参数       : PVOID lpParam                             回调函数参数 
8. // 参数       : BOOL bIsRecur/* = TRUE*/                  是否递归 
9. // 附注       : 回调函数原型 DWORD WINAPI ScanProc(TCHAR *pFilePath, LPVOID lpParameter); 
10. //            回调函数的第一个参数是扫描到一个文件的全名 
11. // ------------------------------------------------------------------------- 
12. typedef DWORD (WINAPI *LPSCANDISK_START_ROUTINE)(TCHAR *pFilePath, LPVOID lpParameter);
13. BOOL ScanDirectory
14. (const TCHAR *pszPath, LPSCANDISK_START_ROUTINE lpFunAddress, PVOID lpParam, BOOL bIsRecur = TRUE)

 

函数实现：

1. // ------------------------------------------------------------------------- 
2. // 函数       : ScanDirectory 
3. // 功能       : 遍历一个目录，传入一个函数指针 
4. // 返回值  : BOOL  
5. // 参数       : const TCHAR *pszPath                      传入路径 
6. // 参数       : LPSCANDISK_START_ROUTINE lpFunAddress     函数指针回调指针 
7. // 参数       : PVOID lpParam                             回调函数参数 
8. // 参数       : BOOL bIsRecur/* = TRUE*/                  是否递归 
9. // 附注       : 回调函数原型 DWORD WINAPI ScanProc(TCHAR *pFilePath, LPVOID lpParameter); 
10. //            回调函数的第一个参数是扫描到一个文件的全名 
11. // ------------------------------------------------------------------------- 
12. BOOL ScanDirectory(const TCHAR *pszPath, LPSCANDISK_START_ROUTINE lpFunAddress, PVOID lpParam, BOOL bIsRecur/* = TRUE*/)
13. {
14.     BOOL bRet = FALSE;
16.     TCHAR *s = NULL;
17.     HANDLE hFind = NULL;
18.     WIN32_FIND_DATA fd = {0};
20.     TCHAR szFileName[MAX_PATH] = TEXT("");
21.     ::lstrcpy(szFileName, pszPath);
23.     s = szFileName + ::lstrlen(szFileName);
24.     if (*(s-1) != TEXT('//'))
25.         *s++ = TEXT('//');
26.     ::lstrcpy(s, TEXT("*.*"));
28.     hFind = FindFirstFile(szFileName, &fd);
29.     if (hFind == INVALID_HANDLE_VALUE)
30.         goto Exit0;
31.     do 
32.     {
33.         // 过滤 
34.         if (::lstrcmpi(TEXT("."), fd.cFileName) == 0 || ::lstrcmpi(TEXT(".."), fd.cFileName) == 0)
35.             continue;
37.         ::lstrcpy(s, fd.cFileName);
38.         *(s + ::lstrlen(fd.cFileName)) = L'/0';
40.         // 如果是文件夹则递归 
41.         if ((fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) && bIsRecur)
42.         {
43.             ScanDirectory(szFileName, lpFunAddress, lpParam, bIsRecur);
44.         }
45.         else
46.         {
47.             // 对文件进行处理 
48.             if (lpFunAddress)
49.                 lpFunAddress(szFileName, lpParam);
50.         }
52.     }while(::FindNextFile(hFind, &fd));
54.     bRet = TRUE;
57. Exit0:
58.     if( hFind != INVALID_HANDLE_VALUE )
59.     {
60.         ::FindClose( hFind );
61.         hFind = NULL;
62.     }
64.     return bRet;
65. }

不过这样的实现有个小缺陷，文件特别多的话，效率会比较低，因为每一个文件都调用了一次回调函数，其实接口还可以设计一个过滤器出来，让扫描函数只对我们感兴趣的文件进行回调，再就是文件夹的处理，也可以进行回调，具体怎么做大家也可以练练手，呵呵，欢迎交流。