[C++知识库]【学习体会】C++遍历文件夹（win10或者ubuntu）

win10 C++遍历文件夹

#include<io.h>//_finddata_t需要io.h头文件
#include<string>
#include<vector>
void getFiles(const std::string & path, std::vector<std::string> & files)
{
	//文件句柄
	long long hFile = 0;
	//文件信息，_finddata_t需要io.h头文件
	struct _finddata_t fileinfo;
	std::string p;
	if ((hFile = _findfirst(p.assign(path).append("\\*").c_str(), &fileinfo)) != -1)
	{
		do
		{
			//如果是目录,迭代之
			//如果不是,加入列表
			if ((fileinfo.attrib & _A_SUBDIR))
			{
				if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)//不考虑隐藏文件夹“.”和“..”,不进行迭代文件夹“.”和“..”
					getFiles(p.assign(path).append("\\").append(fileinfo.name), files);//迭代得到同个目录下的所有文件
			}
			else
			{
				files.push_back(p.assign(path).append("\\").append(fileinfo.name));
			}
		} while (_findnext(hFile, &fileinfo) == 0);//查找hFile文件句柄下的同一级的下一个目录
		_findclose(hFile);
	}
}

ubuntu C++ 遍历文件夹

#include<string>
#include<vector>
#include <sys/stat.h>
#include <dirent.h>
void getFiles(const char* path, std::vector<std::string>& files){

    const std::string path0 = path;
    DIR* pDir;
    struct dirent* ptr;

    struct stat s;
    lstat(path, &s);

    if(!S_ISDIR(s.st_mode)){
        std::cout << "not a valid directory: " << path << std::endl;
        return;
    }

    if(!(pDir = opendir(path))){
        std::cout << "opendir error: " << path << std::endl;
        return;
    }
    int i = 0;
    std::string subFile;
    while((ptr = readdir(pDir)) != 0){
        subFile = ptr -> d_name;
        if(subFile == "." || subFile == "..")
            continue;
        subFile = path0 + subFile;
        std::cout << ++i << ": " << subFile << std::endl;
        files.push_back(subFile);
    }
    closedir(pDir);

}