基于委托机制的（目录）文件遍历

这个示例程序基本实现了Dir命令的行为，但是是使用了ObjectiveC的委托机制，思想是不错的，共享之

需要的用到可以直接拿来用，因为委托机制真的很不错

1. //  
   
2. // 文件名：EnumFile.h  
   
3. // 工程名：EnumFile  
   
4. // 作  者：ccnyou   
   
5. // 完成时间：2012年6月9日  
   
6. //  
   
7.   
   
8. #ifndef EnumFile_h__  
   
9. #define EnumFile_h__  
   
10.   
    
11. #include <stdio.h>  
    
12. #include <windows.h>  
    
13. #include <DIRECT.H>  
    
14. #include <STRING.H>  
    
15.   
    
16. class EnumDirectoryDelegate  
    
17. {  
    
18. public:  
    
19.     //发现文件夹  
    
20.     virtual void DidFoundDirectory(const char *pszDirectoryName){}  
    
21.     //发现文件  
    
22.     virtual void DidFoundFile(const char *pszFileName){}  
    
23.       
    
24.     //即将进入目录  
    
25.     virtual void WillEntryDirectory(const char *pszDirectoryName){}  
    
26.     //已经进入目录  
    
27.     virtual void DidEntryDirectory(const char *pszDirectoryName){}  
    
28.     //询问是否应该进入到目录 TRUE，FALSE 忽略之  
    
29.     virtual BOOL ShouldEntryDirectory(const char *pszDirectoryName) { return TRUE; }  
    
30. };  
    
31.   
    
32.   
    
33. void EnumDirectory(const char *pszPath,EnumDirectoryDelegate *pDelegate = NULL);  
    
34.   
    
35. #endif // EnumFile_h__  
    
36. 
    
37. [cpp] view plaincopyprint?
    
38. //  
    
39. // 文件名：EnumFile.cpp  
    
40. // 工程名：EnumFile  
    
41. // 作  者：ccnyou   
    
42. // 完成时间：2012年6月9日  
    
43. //  
    
44.   
    
45. #include "stdafx.h"  
    
46. #include "EnumFile.h"  
    
47.   
    
48. static BOOL IsFileNameValid(const char *pszFileName)  
    
49. {  
    
50.     if (strcmp(pszFileName,".") == 0)  
    
51.     {  
    
52.         return FALSE;  
    
53.     }  
    
54.   
    
55.     if (strcmp(pszFileName,"..") == 0)  
    
56.     {  
    
57.         return FALSE;  
    
58.     }  
    
59.   
    
60.     return TRUE;  
    
61. }  
    
62.   
    
63. void EnumDirectory(const char *pszPath,EnumDirectoryDelegate *pDelegate/* = NULL*/)  
    
64. {   
    
65.     WIN32_FIND_DATA FileData;   
    
66.     HANDLE          hFindFile;  
    
67.     BOOL            bShouldEntryDir = TRUE;  
    
68.     EnumDirectoryDelegate DefaultDelegate;  
    
69.   
    
70.     char szCurDir[MAX_PATH];   
    
71.     char szSubDir[MAX_PATH];   
    
72.   
    
73.   
    
74.     if (pDelegate == NULL)  
    
75.     {  
    
76.         pDelegate = &DefaultDelegate;  
    
77.     }  
    
78.   
    
79.     pDelegate->DidEntryDirectory(pszPath);  
    
80.   
    
81.     GetCurrentDirectory(MAX_PATH,szCurDir);   
    
82.     SetCurrentDirectory(pszPath);   
    
83.   
    
84.     hFindFile = FindFirstFile("*.*",&FileData);   
    
85.     if(hFindFile!=INVALID_HANDLE_VALUE)   
    
86.     {   
    
87.         do  
    
88.         {   
    
89.             if(FileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)    
    
90.             {  
    
91.                 pDelegate->DidFoundDirectory(FileData.cFileName);  
    
92.   
    
93.                 if(IsFileNameValid(FileData.cFileName))   
    
94.                 {   
    
95.                     // 合成完整路径名   
    
96.                     sprintf(szSubDir,"%s\\%s",pszPath,FileData.cFileName);  
    
97.                       
    
98.                     bShouldEntryDir = pDelegate->ShouldEntryDirectory(szSubDir);  
    
99.                     // 递归枚举子目录   
    
100.                     if (bShouldEntryDir == TRUE)  
     
101.                     {  
     
102.                         pDelegate->WillEntryDirectory(FileData.cFileName);  
     
103.                         EnumDirectory(szSubDir,pDelegate);   
     
104.                     }  
     
105.                 }   
     
106.             }  
     
107.             else  
     
108.             {  
     
109.                 pDelegate->DidFoundFile(FileData.cFileName);  
     
110.             }  
     
111.         }while(FindNextFile(hFindFile,&FileData) != NULL);   
     
112.     }   
     
113.     SetCurrentDirectory(szCurDir);   
     
114. }   
     
115. 
     
116. 
     
117. //  
     
118. // 文件名：main.cpp  
     
119. // 工程名：EnumFile  
     
120. // 作  者：ccnyou   
     
121. // 完成时间：2012年6月9日  
     
122. //  
     
123.   
     
124. #include "StdAfx.h"  
     
125. #include <stdio.h>  
     
126. #include <STDLIB.H>  
     
127. #include "EnumFile.h"  
     
128.   
     
129. class MyDelegate : public EnumDirectoryDelegate  
     
130. {  
     
131. public:  
     
132.       
     
133.     void WillEntryDirectory(const char *pszDirectoryName)  
     
134.     {  
     
135.         printf("\n");  
     
136.     }  
     
137.       
     
138.     void DidFoundDirectory(const char *pszDirectoryName)  
     
139.     {  
     
140.         printf("%-40s[目录]\n",pszDirectoryName);  
     
141.     }  
     
142.       
     
143.     void DidEntryDirectory(const char *pszDirectoryName)  
     
144.     {  
     
145.         printf("当前目录:\n%s\n文件列表:\n",pszDirectoryName);  
     
146.     }  
     
147.       
     
148.     void DidFoundFile(const char *pszFileName)  
     
149.     {  
     
150.         printf("%-40s[文件]\n",pszFileName);  
     
151.     }  
     
152.   
     
153. //去掉注释，可以只遍历当前目录，而不进入子目录  
     
154. //  BOOL ShouldEntryDirectory(const char *pszDirectoryName)  
     
155. //  {  
     
156. //   
     
157. //      return FALSE;  
     
158. //  }  
     
159.   
     
160. };  
     
161.   
     
162. int main(int argc, char* argv[])  
     
163. {  
     
164.     char szTestPath[MAX_PATH];  
     
165.     MyDelegate Delegate;  
     
166.       
     
167.     GetCurrentDirectoryA(MAX_PATH,szTestPath);  
     
168.     strcat(szTestPath,"\\Test");  
     
169.       
     
170.     printf("生成目录\n");  
     
171.     CreateDirectory("Test",NULL);  
     
172.     CreateDirectory("Test\\A",NULL);  
     
173.     CreateDirectory("Test\\B",NULL);  
     
174.       
     
175.     printf("生成目录测试文件\n");  
     
176.     system("echo FileA > Test\\A\\FileA1");  
     
177.     system("echo FileA > Test\\A\\FileA2");  
     
178.     system("echo FileA > Test\\A\\FileA3");  
     
179.       
     
180.     system("echo FileB > Test\\B\\FileB1");  
     
181.     system("echo FileB > Test\\B\\FileB2");  
     
182.     system("echo FileB > Test\\B\\FileB3");  
     
183.     system("echo FileB > Test\\B\\FileB4");  
     
184.       
     
185.     printf("开始遍历目录...\n");  
     
186.     EnumDirectory(szTestPath,&Delegate);  
     
187.     printf("\n遍历目录完成\n");  
     
188.       
     
189.     return 0;  
     
190. }  

复制代码

其中，stdafx.h 是工程自动生成的，如果需要没有生成，可以自己手工创建一个stdafx.h