C++17 中的 directory_iterator 目录遍历详解

在现代编程中，处理文件和目录是常见的任务之一。C++ 提供了多种方式来实现这一功能，其中 std::filesystem::directory_iterator 是 C++17 引入的一个强大工具。本文将详细介绍如何使用 directory_iterator 进行目录遍历，并提供一些实用技巧。

什么是 directory_iterator

std::filesystem::directory_iterator 是 C++17 标准库中的一个类，位于 <filesystem> 头文件中。它用于遍历指定目录下的所有文件和子目录。通过 directory_iterator，你可以轻松地获取目录中的每个条目，并对其进行进一步操作。

基本用法

以下是一个简单的示例，展示如何使用 directory_iterator 遍历目录：

#include <iostream>
#include <filesystem>

namespace fs = std::filesystem;

int main() {
    fs::path directoryPath = "path/to/your/directory";

    try {
        for (const auto& entry : fs::directory_iterator(directoryPath)) {
            if (entry.is_regular_file()) {
                std::cout << "File: " << entry.path().filename() << std::endl;
            } else if (entry.is_directory()) {
                std::cout << "Directory: " << entry.path().filename() << std::endl;
            }
        }
    } catch (const fs::filesystem_error& e) {
        std::cerr << "Error: " << e.what() << std::endl;
    }

    return 0;
}

在这个示例中，我们首先包含了必要的头文件并定义了一个 directory_iterator 对象。然后，我们使用范围基于的 for 循环遍历目录中的每个条目，并根据条目的类型（文件或目录）输出相应的信息。

错误处理

在使用 directory_iterator 时，可能会遇到各种错误，例如目录不存在或权限问题。为了确保程序的健壮性，我们需要捕获这些异常并进行适当的处理。上述示例中已经展示了如何使用 try-catch 块捕获 filesystem_error 异常。

遍历子目录

默认情况下，directory_iterator 只会遍历指定目录下的条目，而不会递归地遍历子目录。如果你需要递归遍历整个目录树，可以使用 recursive_directory_iterator 类。

以下是一个使用 recursive_directory_iterator 的示例：

#include <iostream>
#include <filesystem>

namespace fs = std::filesystem;

int main() {
    fs::path directoryPath = "path/to/your/directory";

    try {
        for (const auto& entry : fs::recursive_directory_iterator(directoryPath)) {
            if (entry.is_regular_file()) {
                std::cout << "File: " << entry.path().filename() << std::endl;
            } else if (entry.is_directory()) {
                std::cout << "Directory: " << entry.path().filename() << std::endl;
            }
        }
    } catch (const fs::filesystem_error& e) {
        std::cerr << "Error: " << e.what() << std::endl;
    }

    return 0;
}

在这个示例中，我们使用了 recursive_directory_iterator 来递归地遍历目录及其所有子目录。

排除特定文件或目录

有时候，你可能希望在遍历时排除某些特定的文件或目录。可以通过检查条目的路径来实现这一点。以下是一个示例，展示如何排除以 .log 结尾的文件：

#include <iostream>
#include <filesystem>

namespace fs = std::filesystem;

int main() {
    fs::path directoryPath = "path/to/your/directory";

    try {
        for (const auto& entry : fs::directory_iterator(directoryPath)) {
            if (entry.is_regular_file() && !entry.path().filename().extension().compare(".log")) {
                continue; // Skip .log files
            }
            std::cout << "File: " << entry.path().filename() << std::endl;
        }
    } catch (const fs::filesystem_error& e) {
        std::cerr << "Error: " << e.what() << std::endl;
    }

    return 0;
}

在这个示例中，我们在遍历过程中检查文件扩展名，如果文件是以 .log 结尾，则跳过该文件。

总结

通过 std::filesystem::directory_iterator 和 std::filesystem::recursive_directory_iterator，你可以方便地在 C++17 中实现目录遍历。结合错误处理和条件过滤，你可以构建出功能强大的文件系统操作程序。希望本文对你有所帮助！