node.js - 带累加器的 Node.js 异步目录遍历

Question

我想获取包含在目录中的唯一文件名列表，包括使用节点的子目录，并且在组合每个回调的结果时遇到了一些麻烦。我想避免重复操作，如果我只是从fs.stat回调中记录文件名，就会发生这种情况。

var distinct = {};

function getNames(root) {
    fs.readdir(root, function(err, list) {
        list.forEach(function(file) {
            file = root + '/' + file; 
            fs.stat(file, function(err, stat) {
                if (!err && stat.isDirectory()) {
                    getNames(file);
                } else {
                    distinct[path.basename(file)] = true;
                }
            });
        });
    });
}

// perform various operations on unique filename list
console.log(Object.keys(distinct));

当然，这console.log()太早地调用了该函数并给出了不希望的结果。我怎样才能获得一组文件名来处理；有没有一种使用异步方法的好方法，即不必使用readdirSyncand statSync？

Answer 1

我可以建议使用承诺吗？

function getNames(root) {
    readdir = Q.nfbind(fs.readdir);
    stat = Q.nfbind(fs.stat);
    return readdir(root)
        .then(function(list) {
            files = [];
            for(var i = 0; i < list.length; i++) {
                file = root + '/' + list[i]; 
                files.push(stat(file)
                    .then(function(stat) {
                        if (stat.isDirectory()) {
                            return getNames(file);
                        } else {
                            return path.basename(file);
                        }
                     })
                );
            }
            return Q.all(files);
        });
}
getNames('someDir')
    .then(console.log)
    .catch(console.error);

Answer 2

异步模块将成为您的朋友。

var distinct = {};
function getNames(root, callback) {
    fs.readdir(root, function(err, list) {
        if (err) return callback(err);   

        async.each(list, function(file, done) {
            fs.stat(file, function(stat) {
                if (err) return done(err);
                if (stat.isDirectory()) return getNames(file, done);
                distinct[path.basename(file)] = true;
                done();
            });
        }, function(err) {
            // called when all done or error
            callback(err, distinct);
        })
    });
}

然后你可以像这样使用它

getNames('path/to/dir', function(distinct) {
    // logic
});