迹忆客 专注技术分享

在本文中，我们将介绍在 C++ 中获取指定目录中的文件列表的方法。

使用 std::filesystem::directory_iterator 来获取目录中的文件列表

该方法是 C++17 中添加的 <filesystem> 库的一部分。请注意，某些较早的编译器可能不支持这个方法，但这是获得目录中文件列表的原生 C++ 解决方案。directory_iterator 在一个基于范围的 for 循环中使用，它把目录的路径作为参数。在这个例子中，我们指定了根目录的路径，但是你应该根据你的系统修改此变量，以传递一个有效的路径。

#include <filesystem>
#include <iostream>
#include <string>
#include <vector>

using std::cin;
using std::cout;
using std::endl;
using std::string;
using std::filesystem::directory_iterator;

int main() {
  string path = "/";

  for (const auto& file : directory_iterator(path)) cout << file.path() << endl;

  return EXIT_SUCCESS;
}

输出：

Archive2| lib| .| ..| lib64| Archive| dev| var| bin| usr| etc| tmp| proc|

使用 opendir/readdir 函数获取目录中的文件列表

该方法更为冗长，但是它是文件系统操作的可靠替代方法。

要使用函数 opendir/readdir，你应该包含 dirent.h 头文件。

opendir 打开一个由名字指定的目录，并返回它的指针，它是 DIR*。

readdir 对 DIR*参数进行操作，并返回一个名为 dirent 的特殊结构。

一旦我们把 readdir 调用放在 while 循环中，与 nullptr 比较，它就像一个目录迭代器一样，我们可以通过每个返回的 dirent 结构获得对文件名的访问。

#include <dirent.h>

#include <iostream>
#include <vector>

using std::cin;
using std::cout;
using std::endl;
using std::vector;

int main() {
  DIR *dir;
  struct dirent *diread;
  vector<char *> files;

  if ((dir = opendir("/")) != nullptr) {
    while ((diread = readdir(dir)) != nullptr) {
      files.push_back(diread->d_name);
    }
    closedir(dir);
  } else {
    perror("opendir");
    return EXIT_FAILURE;
  }

  for (auto file : files) cout << file << "| ";
  cout << endl;

  return EXIT_SUCCESS;
}

请注意，由于 opendir 和 readdir 函数在目录错误/结束时返回 NULL 指针，所以很多 if 语句充当了异常处理机制。

使用 std::filesystem::recursive_directory_iterator 来获取所有子目录中的文件列表

当需要搜索多个子目录的特定文件名时，这个方法很有用。该方法与 directory_iterator 保持一致。你只需要用基于范围的循环来遍历它，并根据需要对文件进行操作。

#include <filesystem>
#include <iostream>
#include <string>
#include <vector>

using std::cin;
using std::cout;
using std::endl;
using std::string;
using std::filesystem::recursive_directory_iterator;

int main() {
  string path = "./";

  for (const auto& file : recursive_directory_iterator(path))
    cout << file.path() << endl;

  return EXIT_SUCCESS;
}