多線程編寫檔案搜尋器

      這兩天去上海找工作了，所以部落格一直沒有更新，發下我走之前的一段代碼吧，是關於多線程的檔案搜尋的。

#include <iostream><br />#include <Windows.h><br />#include <string><br />#include <stack><br />using namespace std;<br />stack<string> dir;<br />CRITICAL_SECTION cs;<br />CRITICAL_SECTION cs1;<br />HANDLE quit;<br />struct info{<br />string name;<br />int flag;<br />int *array;<br />int len;<br />};<br />DWORD WINAPI t(LPVOID l){<br />info *info1=(info*)l;<br />string s=info1->name;<br />while(1){<br />Sleep(1);<br />EnterCriticalSection(&cs);<br />int len=dir.size();<br />string d;<br />if(len>0){<br />d=dir.top();<br />dir.pop();<br />}<br />LeaveCriticalSection(&cs);<br />if(len>0){<br />info1->array[info1->flag]=1;</p><p>string dd=d+"//*.*";<br />WIN32_FIND_DATA f;<br />HANDLE h=FindFirstFile(dd.data(),&f);<br />do{<br />if(strcmp(f.cFileName,".")==0 || strcmp(f.cFileName,"..")==0){<br />continue;<br />}<br />if(f.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY){<br />EnterCriticalSection(&cs);<br />dir.push(d+"//"+f.cFileName);<br />LeaveCriticalSection(&cs);<br />}else{<br />if(strstr(strlwr(f.cFileName),strlwr((char*)s.data()))>0){<br />EnterCriticalSection(&cs1);<br />cout<<d+"//"+f.cFileName<<endl;<br />LeaveCriticalSection(&cs1);<br />}<br />}<br />}while(FindNextFile(h,&f));<br />}else{</p><p>if(WaitForSingleObject(quit,0)==WAIT_OBJECT_0){<br />delete l;<br />return 0;<br />}<br />info1->array[info1->flag]=0;<br />int flag=0;<br />for(int i=0;i<info1->len;i++){<br />if(info1->array[i]==1){<br />flag=1;<br />}<br />}<br />if(flag==0){<br />SetEvent(quit);<br />}<br />}<br />}<br />return 0;<br />}</p><p>int main(int argc, char** argv)<br />{<br />InitializeCriticalSection(&cs);<br />InitializeCriticalSection(&cs1);<br />int array[10]={1};<br />HANDLE h[10];<br />quit=CreateEvent(0,true,FALSE,0);<br />dir.push("g:");<br />for(int i=0;i<10;i++){<br />info *info1=new info;<br />info1->array=array;<br />info1->flag=i;<br />info1->len=10;<br />info1->name="xml";<br />h[i]=CreateThread(0,0,t,info1,0,0);<br />}<br />WaitForMultipleObjects(10,h,true,-1);<br />DeleteCriticalSection(&cs);<br />DeleteCriticalSection(&cs1);<br />} </p><p>

      原理是建立一個stack，把搜尋到的目錄放進去，然後每個線程不斷從裡面取目錄，來搜尋，把搜尋到的目錄再放進去。一直到stack為空白，所有線程都沒有目錄可以搜尋的時候，搜尋結束，線程退出。

      本文有不足之處，還望大家多多指正。