C++17 STL Стандартная библиотека шаблонов

В этом примере мы реализуем небольшой инструмент, который рекурсивно итерирует по заданному каталогу. В процессе он определяет, файлы с какими расширениями находятся в данном каталоге, сколько файлов присутствует для каждого расширения, а также их средний размер.

1. Сначала включим необходимые заголовочные файлы и объявим об использовании пространств имен std и filesystem:

#include

#include

#include

#include

#include

using namespace std;

using namespace filesystem;

2. Функция size_string была полезна в предыдущих примерах. Она преобразует размеры файлов в читабельные строки:

static string size_string(size_t size)

{

  stringstream ss;

  if (size >= 1000000000) {

    ss <<   (size / 1000000000) << 'G';

  } else if (size >= 1000000) {

    ss <<   (size / 1000000) << 'M';

  } else if (size >= 1000) {

    ss <<   (size / 1000) << 'K';

  } else { ss << size << 'B'; }

return ss.str();

}

3. Затем реализуем вспомогательную функцию, которая принимает объект пути и итерирует по всем файлам данного пути. При этом она собирает всю информацию в ассоциативный массив, в котором соотносятся расширения файлов и пары, содержащие общее количество файлов и их суммарный размер.

static map> ext_stats(const path &dir)

{

  map> m;

  for (const auto &entry :

      recursive_directory_iterator{dir}) {

4. Если запись каталога тоже является каталогом, то ее можно опустить. Это не значит, что мы не будем рекурсивно спускаться в каталог. Итератор recursive_ directory_iterator все равно совершит данное действие, но мы не хотим сами заходить в эти подкаталоги.

    const path p {entry.path()};

    const file_status fs {status(p)};

    if (is_directory(fs)) { continue; }

5. Далее извлекаем расширения из строки, представляющей запись каталога. Если у нее нет расширения, то просто опускаем ее:

    const string ext {p.extension().string()};

    if (ext.length() == 0) { continue; }

6. Подсчитаем размер текущего файла. Затем найдем агрегатный объект в ассоциативном массиве для этого расширения. Если такого объекта нет, то неявно создадим его. Просто увеличим счетчик количества файлов и добавим размер файла в переменную-аккумулятор:

    const size_t size {file_size(p)};

    auto &[size_accum, count] = m[ext];

    size_accum += size;

    count      += 1;

  }

7. После этого вернем ассоциативный массив:

  return m;

}

8. В функции main примем путь, предоставленный пользователем в командной строке. Конечно, нужно проверить, существует ли он, поскольку в противном случае продолжать не имеет смысла.

int main(int argc, char *argv[])

{

  path dir {argc > 1 ? argv[1] : "."};

  if (!exists(dir)) {

    cout << "Path " << dir << " does not exist.\n";

    return 1;

  }

9. Можно мгновенно проитерировать по ассоциативному массиву, предоставляемому ext_stats. Поскольку элементы типа accum_size этого массива содержат сумму всех файлов с одинаковым расширением, разделим эту сумму на общее количество таких файлов и выведем ее на экран:

  for (const auto &[ext, stats] : ext_stats(dir)) {

    const auto &[accum_size, count] = stats;

    cout << setw(15) << left << ext << ": "

         << setw(4) << right << count

         << " items, avg size "

         << setw(4) << size_string(accum_size / count)

         << '\n';

  }

}

10. Компиляция и запуск программы дадут следующий результат. Я предоставил ей в качестве аргумента командной строки каталог, содержащий офлайн-справку по С++.

$ ./file_type ~/Documents/cpp_reference/

.css   :    2 items, avg size  41K

.gif   :    7 items, avg size 902B

.html  : 4355 items, avg size  38K

.js    :    3 items, avg size   4K

.php   :    1 items, avg size 739B

.png   :   34 items, avg size   2K

.svg   :   53 items, avg size   6K