多線程編寫文件搜索器

      這兩天去上海找工作了，所以博客一直沒有更新，發下我走之前的一段代碼吧，是關於多線程的文件搜索的。

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

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

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