深度搜索遍历文件夹
深度优先搜索遍历文件夹所有文件, 由于使用windows的函数, 必须要使用C语言;
注意字符集的问题,使用"#undef UNICODE", 屏蔽因字符集所产生的问题;
使用vector<string>存储所有文件名, 因为要递归使用, 所以需要设置为静态,返回shared_ptr的指针
代码如下:
/*************************************************
File: main.cpp
Copyright: C.L.Wang
Author: Caroline
Date: 2013-11-27
Description: 深度优先递归遍历目录中所有的文件
Email: morndragon@126.com
**************************************************/
#undef UNICODE
#include <iostream>
#include <string>
#include <vector>
#include <memory>
#include <cstring>
#include <windows.h>
std::shared_ptr<std::vector<std::string> >
fileList(const std::string& folder_path)
{
static std::shared_ptr<std::vector<std::string> >
folder_files(new std::vector<std::string>); //返回指针, 需要迭代使用
WIN32_FIND_DATA FindData;
HANDLE hError;
int file_count(0);
std::string file_path(folder_path); //路径名
std::string full_file_path; //全路径名
file_path.append("/*.*");
hError = FindFirstFile(file_path.c_str(), &FindData);
if (hError == INVALID_HANDLE_VALUE) {
std::cout << "failed to search files." << std::endl;
return nullptr;
}
while(FindNextFile(hError, &FindData))
{
//过虑".", "..", "-q"
if(0 == strcmp(FindData.cFileName, ".") ||
0 == strcmp(FindData.cFileName, "..") ||
0 == strcmp(FindData.cFileName, "-q"))
{
continue;
}
//完整路径
full_file_path.append(folder_path);
full_file_path.append("/");
full_file_path.append(FindData.cFileName);
++file_count;
if (FindData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY){
//std::cout << file_count << " " << full_file_path << "<Dir>" << std::endl;
fileList(full_file_path);
}else{
folder_files->push_back(full_file_path);
//std::cout << file_count << " " << full_file_path << std::endl;
}
full_file_path.clear(); //清空目录
}
return folder_files;
}
int main(void)
{
std::shared_ptr<std::vector<std::string> > folder_files;
folder_files = fileList("E:/Picture/shoes3");
if (folder_files) {
for (size_t i=0; i != folder_files->size(); ++i) {
std::cout << i+1 << " : " << (*folder_files)[i] << std::endl;
}
}
return 0;
}