nodejs 遍历目录 返回目录项数组

iteratorDir.js

function iteratorDir(dir, recursive) {
  // 去除尾部的'/'字符
  if (dir[dir.length - 1] == '/') {
    dir = dir.substring(0, dir.length - 1);
  }
 
  var dirList = [];
  var items = fs.readdirSync(dir);
  items.forEach((item, index) => {
    var path = dir + '/' + item;
    var stat = fs.statSync(path);

    if (stat.isFile()) {
      dirList.push(path);
    } else if (stat.isDirectory()) {
      if (recursive) {
        var subDirList = iteratorDir(path, recursive);
        dirList = dirList.concat(subDirList);
      } else {
        path = path + '/';
        dirList.push(path);
      }
    }
  });

  return dirList;
}