可能各位看官們有更好的方法,請不吝賜教。
#ifdef _WIN32 #include <io.h> #else #include <unistd.h> #include <stdio.h> #include <dirent.h> #include <sys/stat.h> #endif
win32平台引用io.h里面的信息。
主要定義了一個void dfsFolder(string folderPath, int depth = 0);方法對其進行遞歸遍歷。
具體的實現為
1 void dfsFolder( string folderPath, int depth) 2 { 3 #ifdef WIN32 4 _finddata_t FileInfo; 5 string strfind = folderPath + "\\*"; 6 long Handle = _findfirst(strfind.c_str(), &FileInfo); 7 8 if (Handle == -1L) 9 { 10 cerr << "can not match the folder path" << endl; 11 exit(-1); 12 } 13 do{ 14 //判斷是否有子目錄 15 if (FileInfo.attrib & _A_SUBDIR) 16 { 17 //這個語句很重要 18 if( (strcmp(FileInfo.name,".") != 0 ) &&(strcmp(FileInfo.name,"..") != 0)) 19 { 20 string newPath = folderPath + "\\" + FileInfo.name; 21 dfsFolder(newPath); 22 } 23 } 24 else 25 { 26 string filename = (folderPath + "\\" + FileInfo.name); 27 28 cout << folderPath << "\\" << FileInfo.name << " " << endl; 29 } 30 }while (_findnext(Handle, &FileInfo) == 0); 31 32 _findclose(Handle); 33 #else 34 DIR *dp; 35 struct dirent *entry; 36 struct stat statbuf; 37 if((dp = opendir(folderPath.c_str())) == NULL) { 38 fprintf(stderr,"cannot open directory: %s\n", folderPath.c_str()); 39 return; 40 } 41 chdir(folderPath.c_str()); 42 while((entry = readdir(dp)) != NULL) { 43 lstat(entry->d_name,&statbuf); 44 if(S_ISDIR(statbuf.st_mode)) { 45 46 if(strcmp(".",entry->d_name) == 0 || 47 strcmp("..",entry->d_name) == 0) 48 continue; 49 printf("%*s%s/\n",depth,"",entry->d_name); 50 dfsFolder(entry->d_name,depth+4); 51 } else { 52 string filename = entry->d_name; 53 54 printf("%*s%s\n",depth,"",entry->d_name); 55 } 56 } 57 chdir(".."); 58 closedir(dp); 59 #endif 61 }
chdir 是C語言中的一個系統調用函數(同cd),其中對目錄"."及".."進行特殊的判斷,因為"."是表示當前目錄,".."表示父目錄。
如果不進行特殊判定的話則進入死循環。
一個很簡單的功能,希望可以幫助得到需要的同學們~