C++目录项迭代无一例外

问题描述

在 C++17 中，遍历某个目录 ICar 中的项目变得很容易：

dir

不幸的是，这种方式可能会抛出异常，我想在我的程序中避免这种情况。

不抛出异常的迭代也是可以的：

for ( auto& dirEntry: std::filesystem::directory_iterator(dir) )
{
  if ( !dirEntry.is_regular_file() ) continue;
...

但它在 C++ 中更加冗长。例如，我不能使用基于范围的 for。这个更大的代码大小对我来说真的很重要，因为我有很多地方需要迭代。 有没有办法简化迭代目录项的代码，同时还能避免异常？

解决方法

我认为可以创建一个安全的包装迭代器，它的操作符++不会抛出异常，如下

// object of this struct can be passed to range based for
struct safe_directory
{
    std::filesystem::path dir;
    std::error_code & ec;
};

//iterator of directory items that will save any errors in (ec) instead of throwing exceptions
struct safe_directory_iterator
{
    std::filesystem::directory_iterator it;
    std::error_code & ec;
    safe_directory_iterator & operator ++() { it.increment( ec ); return * this; }
    auto operator *() const { return *it; }
};

safe_directory_iterator begin( const safe_directory & sd )
{
    return safe_directory_iterator{ std::filesystem::directory_iterator( sd.dir,sd.ec ),sd.ec };
}
 
std::filesystem::directory_iterator end( const safe_directory & )
{
    return {};
}

bool operator !=( const safe_directory_iterator & a,const std::filesystem::directory_iterator & b )
{
    return !a.ec && a.it != b;
}

然后就可以在这样的程序中使用

int main()
{
    std::error_code ec;
    safe_directory sdir{ std::filesystem::current_path(),ec };

    for ( auto dirEntry : sdir )
    {
        if ( dirEntry.is_regular_file( ec ) )
            std::cout << dirEntry.path() << std::endl;
    }
}

参见在线编译器中的示例：https://gcc.godbolt.org/z/fb4qPE6Gf