获取当前目录下的文件，目录 （封装）

本人正在开发一个简单的游戏服务器引擎，期间遇到载入文件夹下的所有lua文件，需要遍历目录和子目录获取所有的lua文件，运用到的是windows函数获取目录下的文件和目录，由于windows提供的函数为获取目录下的制定类型的文件，但不能获取子目录中的文件。本人进行一次函数封装，共提供3个基本函数：

// 查找当前目录下的所有目录(不包括当前目录)
void FindAllDir(const char* pCurDir, vector<string>& vtDirs);
// 查找当前目录下的所有文件(不包括子目录)，制定查找文件类型，如：*.txt,*.lua,*.*
void FindAllFile(const char* pCurDir, const char* pFileType, vector<string>& vtFiles);
// 查找当前目录下的所有文件(包括子目录)，制定查找文件类型，如：*.txt,*.lua,*.*
void FindAllFileE(const char* pCurDir, const char* pFileType, vector<string>& vtFiles);

通过FindAllFileE这个函数可以获取当前目录下和子目录下的所有lua文件，函数调用方法为：

// lua目录char szLuaPath[MAX_PATH] = {0};_snprintf(szLuaPath, MAX_PATH, "%s\\lua\\", GetRootPath());vector<string> luaFiles;CFileAndDirFinder cFileAndDirFinder;cFileAndDirFinder.FindAllFileE(szLuaPath, "*.lua", luaFiles);// 遍历所有lua文件并通过lua_tinker进行加载。vector<string>::iterator itLuaFile = luaFiles.begin();for (; itLuaFile!=luaFiles.end(); ++itLuaFile){lua_tinker::dofile(gL, itLuaFile->c_str());}return true;

/* 函数使用案例：

vector<string> luaFiles;//GetLuaFileList(GetRootPath(), luaFiles);

CFileAndDirFinder cFileAndDirFinder;

//cFileAndDirFinder.FindAllDir(GetRootPath(), luaFiles);

//cFileAndDirFinder.FindAllFile(GetRootPath(), "*.txt", luaFiles);

cFileAndDirFinder.FindAllFileE(GetRootPath(), "*.txt", luaFiles);

vector<string>::iterator it = luaFiles.begin();

for (; it!=luaFiles.end(); ++it)

{

cout << it->c_str() << endl;

}

*/

#pragma onceclass CFileAndDirFinder{public:CFileAndDirFinder(void);~CFileAndDirFinder(void);// 查找当前目录下的所有目录(不包括当前目录)void FindAllDir(const char* pCurDir, vector<string>& vtDirs);// 查找当前目录下的所有文件(不包括子目录)，制定查找文件类型，如：*.txt,*.lua,*.*void FindAllFile(const char* pCurDir, const char* pFileType, vector<string>& vtFiles);// 查找当前目录下的所有文件(包括子目录)，制定查找文件类型，如：*.txt,*.lua,*.*void FindAllFileE(const char* pCurDir, const char* pFileType, vector<string>& vtFiles);};/* 函数使用案例：vector<string> luaFiles;//GetLuaFileList(GetRootPath(), luaFiles);CFileAndDirFinder cFileAndDirFinder;//cFileAndDirFinder.FindAllDir(GetRootPath(), luaFiles);//cFileAndDirFinder.FindAllFile(GetRootPath(), "*.txt", luaFiles);cFileAndDirFinder.FindAllFileE(GetRootPath(), "*.txt", luaFiles);vector<string>::iterator it = luaFiles.begin();for (; it!=luaFiles.end(); ++it){cout << it->c_str() << endl;}*/

#include "StdAfx.h"#include "FileAndDirFinder.h"CFileAndDirFinder::CFileAndDirFinder(void){}CFileAndDirFinder::~CFileAndDirFinder(void){}void CFileAndDirFinder::FindAllDir( const char* pCurDir, vector<string>& vtDirs ){// 当前目录char szDir[MAX_PATH] = {0};_snprintf(szDir, MAX_PATH, "%s\\*.*", pCurDir);WIN32_FIND_DATA findFileData = {0};HANDLE hFind = FindFirstFile((LPCSTR)szDir, &findFileData);if (INVALID_HANDLE_VALUE == hFind){return ;}do{/* 返回的文件名中会包含"."和".."。“.'代表本目录，".."代表上一层目录。一般情况下需要把这两个名称过滤掉。比如要进行文件删除操作*/if (findFileData.cFileName[0] != '.')//不是当前路径或者父目录的快捷方式{if(findFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY){// 这是一个普通目录char tmpDir[MAX_PATH] = {0};_snprintf(tmpDir, MAX_PATH, "%s\\%s", pCurDir, findFileData.cFileName);vtDirs.push_back(tmpDir);// 递归调用查找子目录FindAllDir(tmpDir, vtDirs);}}}while (FindNextFile(hFind, &findFileData));FindClose(hFind);hFind = INVALID_HANDLE_VALUE;}void CFileAndDirFinder::FindAllFile( const char* pCurDir, const char* pFileType, vector<string>& vtFiles ){// 当前目录char szDir[MAX_PATH] = {0};_snprintf(szDir, MAX_PATH, "%s\\%s", pCurDir, pFileType);WIN32_FIND_DATA findFileData = {0};HANDLE hFind = FindFirstFile((LPCSTR)szDir, &findFileData);if (INVALID_HANDLE_VALUE == hFind){return ;}do{/* 返回的文件名中会包含"."和".."。“.'代表本目录，".."代表上一层目录。一般情况下需要把这两个名称过滤掉。比如要进行文件删除操作*/if (findFileData.cFileName[0] != '.')//不是当前路径或者父目录的快捷方式{// 这是一个文件char tmpFile[MAX_PATH] = {0};_snprintf(tmpFile, sizeof(tmpFile)-1, "%s\\%s", pCurDir, findFileData.cFileName);vtFiles.push_back(tmpFile);}}while (FindNextFile(hFind, &findFileData));FindClose(hFind);hFind = INVALID_HANDLE_VALUE;}void CFileAndDirFinder::FindAllFileE( const char* pCurDir, const char* pFileType, vector<string>& vtFiles ){// 获取子目录vector<string> vtDirs;FindAllDir(pCurDir, vtDirs);// 加入当前目录vtDirs.push_back(pCurDir);vector<string>::iterator itDir = vtDirs.begin();for (; itDir!=vtDirs.end(); ++itDir){FindAllFile(itDir->c_str(), pFileType, vtFiles);}}

源代码下载地址：FileAndDirFinder.rar