Агрегация 2 — JS: Деревья

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

const tree = fsTrees.mkdir('/', [
  fsTrees.mkdir('etc', [
    fsTrees.mkdir('apache'),
    fsTrees.mkdir('nginx', [
      fsTrees.mkfile('nginx.conf'),
    ]),
  ]),
  fsTrees.mkdir('consul', [
    fsTrees.mkfile('config.json'),
    fsTrees.mkfile('file.tmp'),
    fsTrees.mkdir('data'),
  ]),
  fsTrees.mkfile('hosts'),
  fsTrees.mkfile('resolve'),
])

console.log(getSubdirectoriesInfo(tree))
// => [['etc', 1], ['consul', 2]]

Внутри себя эта задача распадается на две:

• Реализация функции подсчёта файлов внутри директории
• Вызов данной функции для каждой из поддиректорий

Начнём с подсчёта количества файлов. Это классическая задача на агрегацию:

const getFilesCount = (node) => {
  if (fsTrees.isFile(node)) {
    return 1
  }

  const children = fsTrees.getChildren(node)
  const descendantCounts = children.map(getFilesCount)
  return _.sum(descendantCounts)
}

Следующий шаг заключается в том, чтобы извлечь всех детей из исходного узла и к каждому из них применить подсчёт:

const getSubdirectoriesInfo = (tree) => {
  const children = fsTrees.getChildren(tree)
  const result = children
    // Нас интересуют только директории
    .filter(fsTrees.isDirectory)
    // Запускаем подсчёт для каждой директории
    .map(child => [fsTrees.getName(child), getFilesCount(child)])

  return result
}

То есть мы обратились к детям напрямую сначала отфильтровав их, а затем выполнили отображение на необходимый массив, содержащий для каждой директории имя и количество файлов в нем.