【C++】多线程全盘文件搜索

Fluray

新建一个控制台，要支持MFC。
只用了MFC一个函数AfxBeginThread，其余的全是win API。线程函数修改自《windows程序设计 第2版》 王艳平 张铮 编著

1. // 多线程文件搜索.cpp : 定义控制台应用程序的入口点。
   
2. //
   
3. 
   
4. #include "stdafx.h"
   
5. 
   
6. #include <afxwin.h>
   
7. #include <process.h>
   
8. #include <iostream>
   
9. #include <stddef.h>
   
10. 
    
11. #define BUFSIZE 1024
    
12. using namespace std;
    
13. //int g_FileName=0;
    
14. struct CDirectoryNode : public CNoTrackObject
    
15. {
    
16.     CDirectoryNode* pNext;  // CTypedSimpleList类模板要使用此成员
    
17.     char szDir[MAX_PATH];   // 要查找的目录
    
18. };
    
19. 
    
20. class CRapidFinder  
    
21. {
    
22. public:
    
23.     CRapidFinder(int nMaxThread);
    
24.     virtual ~CRapidFinder();
    
25.     BOOL CheckFile(LPCTSTR lpszFileName);
    
26. 
    
27.     int m_nResultCount;                             // 结果数目
    
28.     int m_nThreadCount;                             // 活动线程数目
    
29.     CTypedSimpleList<CDirectoryNode*> m_listDir;    // 目录列表
    
30.     CRITICAL_SECTION m_cs;                          // 关键代码段
    
31. 
    
32.     const int m_nMaxThread;                 // 最大线程数目
    
33.     char m_szMatchName[MAX_PATH];           // 要搜索的文件
    
34.     HANDLE m_hDirEvent;                     // 向m_listDir中添加新的目录后置位（受信）
    
35.     HANDLE m_hExitEvent;                    // 各搜索线程将要退出时置位（受信）
    
36. };
    
37. // m_nMaxThread成员是一个const类型的变量，必须使用成员初始化列表来初始化它的值
    
38. CRapidFinder::CRapidFinder(int nMaxThread) : m_nMaxThread(nMaxThread)
    
39. {
    
40.     m_nResultCount = 0;
    
41.     m_nThreadCount = 0;
    
42.     m_szMatchName[0] = '\0';
    
43. 
    
44.     m_listDir.Construct(offsetof(CDirectoryNode, pNext));
    
45.     m_hDirEvent = ::CreateEvent(NULL, FALSE, FALSE, NULL);
    
46.     m_hExitEvent = ::CreateEvent(NULL, FALSE, FALSE, NULL);
    
47.     ::InitializeCriticalSection(&m_cs);
    
48. }
    
49. 
    
50. CRapidFinder::~CRapidFinder()
    
51. {
    
52.     ::CloseHandle(m_hDirEvent);
    
53.     ::CloseHandle(m_hExitEvent);
    
54.     ::DeleteCriticalSection(&m_cs);
    
55. }
    
56. //包含此文件名的文件都会被查找到
    
57. //BOOL CRapidFinder::CheckFile(LPCTSTR lpszFileName)
    
58. //{
    
59. //  char string[MAX_PATH];
    
60. //  char strSearch[MAX_PATH];
    
61. //  strcpy(string, lpszFileName);
    
62. //  strcpy(strSearch, m_szMatchName);
    
63. //
    
64. //  // 将字符串string和strSearch中的字符全部转化成大写
    
65. //  _strupr(string);
    
66. //  _strupr(strSearch);
    
67. //
    
68. //  // 找出字符串strSearch在字符串string中第一次出现的位置
    
69. //  // 如果string中不包含strSearch，strstr函数返回NULL
    
70. //  if(strstr(string, strSearch) != NULL)
    
71. //      return TRUE;
    
72. //
    
73. //  return FALSE;
    
74. //}
    
75. //必须是相同问文件名(大小写可以不同)
    
76. BOOL CRapidFinder::CheckFile(LPCTSTR lpszFileName)
    
77. {
    
78.     char string[MAX_PATH];
    
79.     char strSearch[MAX_PATH];
    
80.     strcpy(string, lpszFileName);
    
81.     strcpy(strSearch, m_szMatchName);
    
82. 
    
83.     // 将字符串string和strSearch中的字符全部转化成大写
    
84.     _strupr(string);
    
85.     _strupr(strSearch);
    
86. 
    
87.     //如果两个文件名称相同
    
88.     if(strcmp(string, strSearch) == 0)
    
89.         return TRUE;
    
90. 
    
91.     return FALSE;
    
92. }
    
93. UINT FinderEntry(LPVOID lpParam)
    
94. {
    
95.     CRapidFinder* pFinder = (CRapidFinder*)lpParam;
    
96.     CDirectoryNode* pNode = NULL;   // 从m_listDir中取出的节点
    
97.     BOOL bActive = TRUE;        // 指示当前线程的状态
    
98. 
    
99.     // 循环处理m_listDir列表中的目录
    
100.     while(1)
     
101.     {
     
102.         // 从m_listDir列表中取出一个新的目录
     
103.          ::EnterCriticalSection(&pFinder->m_cs);
     
104.         if(pFinder->m_listDir.IsEmpty())
     
105.         {
     
106.             bActive = FALSE;
     
107.         }
     
108.         else
     
109.         {
     
110.             pNode = pFinder->m_listDir.GetHead();
     
111.             pFinder->m_listDir.Remove(pNode);
     
112.         }
     
113.         ::LeaveCriticalSection(&pFinder->m_cs);
     
114. 
     
115.         // m_listDir为空的话就试图在m_hDirEvent事件上等待
     
116.         if(!bActive)
     
117.         {
     
118.             // 准备进入等待状态
     
119.             ::EnterCriticalSection(&pFinder->m_cs);
     
120.             pFinder->m_nThreadCount--;
     
121.             if(pFinder->m_nThreadCount == 0)  // 查看是否已经查找完毕
     
122.             {
     
123.                 ::LeaveCriticalSection(&pFinder->m_cs);
     
124.                 break;
     
125.             }
     
126.             ::LeaveCriticalSection(&pFinder->m_cs);
     
127. 
     
128.             // 进入等待状态
     
129.             ResetEvent(pFinder->m_hDirEvent);
     
130.             ::WaitForSingleObject(pFinder->m_hDirEvent, INFINITE);
     
131. 
     
132.             // 变成活动线程后进入下一次循环
     
133.             ::EnterCriticalSection(&pFinder->m_cs);
     
134.             pFinder->m_nThreadCount++;
     
135.             ::LeaveCriticalSection(&pFinder->m_cs);
     
136.             bActive = TRUE;
     
137.             continue;
     
138.         }
     
139.             
     
140.         
     
141.         // 在pNode指向的目录中查找文件
     
142. 
     
143.         WIN32_FIND_DATA fileData;
     
144.         HANDLE hFindFile;
     
145.         // 设置成X:\XXXX\*.*的格式
     
146.         if(pNode->szDir[strlen(pNode->szDir)-1] != '\\')
     
147.             strcat(pNode->szDir, "\");
     
148.         strcat(pNode->szDir, "*.*");
     
149.         hFindFile = ::FindFirstFile(pNode->szDir, &fileData);
     
150.         if(hFindFile != INVALID_HANDLE_VALUE)
     
151.         {
     
152.             do
     
153.             {
     
154.                 if(fileData.cFileName[0] == '.')
     
155.                     continue;
     
156.                 if(fileData.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY)
     
157.                 {
     
158.                     // 将搜索到的目录添加到目录列表中
     
159. 
     
160.                     // 为新的节点申请内存空间，设置完整的目录名称
     
161.                     CDirectoryNode* p = new CDirectoryNode;
     
162.                     strncpy(p->szDir, pNode->szDir, strlen(pNode->szDir)-3);
     
163.                     strcat(p->szDir, fileData.cFileName);
     
164.                     // 添加到列表中
     
165.                     ::EnterCriticalSection(&pFinder->m_cs);
     
166.                     pFinder->m_listDir.AddHead(p);
     
167.                     ::LeaveCriticalSection(&pFinder->m_cs);
     
168.                     // 促使一个“非活动线程”变成“活动线程”
     
169.                     ::SetEvent(pFinder->m_hDirEvent);   
     
170.                 }
     
171.                 else
     
172.                 {
     
173.                     // 检查搜索到的文件
     
174.                     if(pFinder->CheckFile(fileData.cFileName))
     
175.                     {
     
176.                         ::EnterCriticalSection(&pFinder->m_cs);
     
177.                         ::InterlockedIncrement((long*)&pFinder->m_nResultCount);
     
178.                         ::LeaveCriticalSection(&pFinder->m_cs);
     
179.                           char FileName[256];
     
180.                   int  i;
     
181.                    for (i=strlen(pNode->szDir)-4;i>=0;i--)
     
182.                         FileName[i]=pNode->szDir[i];
     
183.                 FileName[strlen(pNode->szDir)-3]='\0';
     
184.                 strcat(FileName, fileData.cFileName);
     
185.                 printf("%s  \n",FileName);
     
186.                     
     
187.                     }      
     
188.                 }
     
189.             }while(::FindNextFile(hFindFile, &fileData));
     
190.         }
     
191. 
     
192.         // 此节点保存的目录已经搜索完毕，释放内存空间，进入下次循环
     
193.         delete pNode;
     
194.         pNode = NULL;
     
195.     }
     
196. 
     
197.     // 促使一个搜索线程从WaitForSingleObject函数返回，并退出循环
     
198.     ::SetEvent(pFinder->m_hDirEvent);
     
199. 
     
200.     // 判断此线程是否是最后一个结束循环的线程，如果是就通知主线程
     
201.     if(::WaitForSingleObject(pFinder->m_hDirEvent, 0) != WAIT_TIMEOUT)
     
202.     // 如果此时pFinder->m_hDirEvent所对应的事件对象为受信状态，
     
203.     // WaitForSingleObject函数的返回值将是WAIT_OBJECT_0
     
204.     {
     
205.         // 通知主线程最后一个搜索线程即将退出，文件搜索完毕
     
206.         ::SetEvent(pFinder->m_hExitEvent);
     
207.    
     
208.     }
     
209.     return 0;
     
210. }
     
211. 
     
212. 
     
213. void main()
     
214. {
     
215. 
     
216.     char szFile[256];
     
217.     char szPath[256] ;
     
218.     char     szLogicalDriveStrings[BUFSIZE];
     
219.     DWORD   iLength;
     
220.     int iSub;
     
221.     int count=0;
     
222.     clock_t start,end;//程序运行的起始和结束时间
     
223.     float costtime;//程序耗时
     
224.     printf("请输入想查找的文件名：");
     
225.     scanf("%s",szFile);
     
226.     ZeroMemory(szLogicalDriveStrings, BUFSIZE);
     
227.     iLength = GetLogicalDriveStringsA(BUFSIZE-1, szLogicalDriveStrings);
     
228.     start=clock();//开始计时
     
229.     for(iSub=0; iSub<iLength; iSub+=4)
     
230.     {
     
231.     CRapidFinder* pFinder = new CRapidFinder(64);
     
232.     CDirectoryNode* pNode = new CDirectoryNode;
     
233.     //如果不是固定磁盘驱动器:本地硬盘或移动硬盘，忽略
     
234.         if(GetDriveType(szLogicalDriveStrings+iSub)!=3)
     
235.             continue;
     
236.         /*printf("%s  ", szLogicalDriveStrings+iSub);*/
     
237.     // 设置参数信息
     
238.     strcpy(pNode->szDir, szLogicalDriveStrings+iSub);
     
239.     pFinder->m_listDir.AddHead(pNode);
     
240.     strcpy(pFinder->m_szMatchName, szFile);
     
241. 
     
242.     // 创建辅助线程并等待查找结束
     
243.     pFinder->m_nThreadCount = pFinder->m_nMaxThread;
     
244.     for(int i=0; i<pFinder->m_nMaxThread; i++)
     
245.     {
     
246.         AfxBeginThread(FinderEntry, pFinder);
     
247.     }
     
248.     ::WaitForSingleObject(pFinder->m_hExitEvent, INFINITE);
     
249. 
     
250.     // 打印出结果
     
251.     /*printf("%s 中最终查找到的文件的个数为：%d \n",szLogicalDriveStrings+iSub, pFinder->m_nResultCount);*/
     
252.     count=count+pFinder->m_nResultCount;
     
253.     delete pFinder;
     
254.     }
     
255.     end=clock();//计时结束
     
256.     costtime= (float)(end - start) / CLOCKS_PER_SEC;  //转换时间格式
     
257.     /*ShellExecute(NULL,"open",g_FileName,NULL,NULL,SW_SHOWNORMAL);*/
     
258.     printf("使用64个线程全盘搜索文件%s，一共搜索到%d个，用时%.5f s",szFile,count,costtime);
     
259.     system("pause");
     
260. }

复制代码

邪丶帝

楼主 这个只支持 W7？还是通用？？

fsjaky

谢谢楼主分享 正好在写这方面的程序

samson987

感謝大大的分享

曦。风起

学习了 谢谢分享

tx7790

感谢楼主分享