“佳人犹唱醉翁词，四十三年如电抹。”——（北宋） 苏轼 《木兰花·次欧公西湖韵》

    昨天又在网络上摘抄了一段工具类的代码，因为以后可能还会用到，所以简单记录一下。使用C++获取指定目录中的所有文件名，这里不包括该目录下的子目录的文件，如果需要包括的话可以在网上再搜一搜，有很多可以实现这个功能的代码。话不多说，直接上代码：

#include <stdio.h>
#include <string.h>
#include <dirent.h>

void getFiles(std::string path, std::vector<std::string> &files) {
    DIR *dir;
    struct dirent *ptr;
    if ((dir = opendir(path.c_str())) == NULL) {
        perror("Open dir error...");
        return;
    }

    while ((ptr = readdir(dir)) != NULL) {
        if (strcmp(ptr->d_name, ".") == 0 || strcmp(ptr->d_name, "..") == 0)    ///current dir OR parrent dir
            continue;
        else if (ptr->d_type == 8)    ///file
        {
            std::string strFile;
            strFile = path;
            strFile += "/";
            strFile += ptr->d_name;
            files.push_back(strFile);
        } else {
            continue;
        }
    }
    closedir(dir);
}

     这个函数调用也是非常简单，直接给出路径，调用函数方法获取文件名就可以：

#include <iostream>

int main() {
    std::string path = "/home/dong/data";
    std::vector<std::string> files;
    getFiles(path, files);
    for (int i = 0; i < files.size(); ++i) {
        std::cout << files[i] << std::endl;
    }
    std::cout << "Hello, World!" << std::endl;
    return 0;
}

    这个功能比较简单，所以也就简单记录一下！

参考：

1、C++获取指定路径下的所有文件（windows和Linux版）