C++ - 深度搜索遍历文件夹

深度搜索遍历文件夹

深度优先搜索遍历文件夹所有文件, 由于使用windows的函数, 必须要使用C语言;

注意字符集的问题,使用"#undef UNICODE", 屏蔽因字符集所产生的问题;

使用vector<string>存储所有文件名, 因为要递归使用, 所以需要设置为静态,返回shared_ptr的指针

代码如下:

/*************************************************
File: main.cpp
Copyright: C.L.Wang
Author: Caroline
Date: 2013-11-27
Description: 深度优先递归遍历目录中所有的文件
Email: [e-mail]
**************************************************/

#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;
}