使用承诺和递归迭代文件目录

Iterating file directory with promises and recursion

本文关键字：迭代 文件目录 递归 承诺      更新时间：2023-09-26

我知道我将在以下函数中提前返回，如何将递归承诺链接到我的结果？

我的目标是获取目录中的文件列表数组及其所有子目录。数组是单维的，我在这个例子中使用了 concat。

function iterate(body) {
    return new Promise(function(resolve, reject){
        var list = [];
        fs.readdir(body.path, function(error, list){
            list.forEach(function(file){
                file = path.resolve(body.path, file);
                fs.stat(file, function(error, stat){
                    console.log(file, stat.isDirectory());
                    if(stat.isDirectory()) {
                        return iterate({path: file})
                            .then(function(result){
                                list.concat(result);
                            })
                            .catch(reject);
                    } else {
                        list.push(file);
                    }
                })
            });
            resolve(list);
        });
    });
};

您的代码中有许多错误。 部分列表：

  1. .concat()返回一个新数组，因此list.concat(result)本身实际上不会执行任何操作。

  2. 您正在同步调用resolve()，而不是等待所有异步操作完成。

  3. 您正在尝试从几个嵌套异步回调的深处递归返回。 你不能这么做。 这不会在任何地方获得结果。

我发现通过使用 fs 模块的 prodid 版本，这更容易使用。 我使用蓝鸟来创建它，然后你可以这样做：

const path = require('path');
var Promise = require('bluebird');
const fs = Promise.promisifyAll(require('fs'));
function iterate(dir) {
    return fs.readdirAsync(dir).map(function(file) {
        file = path.resolve(dir, file);
        return fs.statAsync(file).then(function(stat) {
            if (stat.isDirectory()) {
                return iterate(file);
            } else {
                return file;
            }
        })
    }).then(function(results) {
        // flatten the array of arrays
        return Array.prototype.concat.apply([], results);
    });
}

注意：我更改了iterate()，只采用初始路径，因此它更通用。 您可以先将body.path传递给它以适应。

以下是使用通用 ES6 承诺的版本：

const path = require('path');
const fs = require('fs');
fs.readdirAsync = function(dir) {
    return new Promise(function(resolve, reject) {
        fs.readdir(dir, function(err, list) {
            if (err) {
                reject(err);
            } else {
                resolve(list);
            }
        });
    });
}
fs.statAsync = function(file) {
    return new Promise(function(resolve, reject) {
        fs.stat(file, function(err, stat) {
            if (err) {
                reject(err);
            } else {
                resolve(stat);
            }
        });
    });
}

function iterate2(dir) {
    return fs.readdirAsync(dir).then(function(list) {
        return Promise.all(list.map(function(file) {
            file = path.resolve(dir, file);
            return fs.statAsync(file).then(function(stat) {
                if (stat.isDirectory()) {
                    return iterate2(file);
                } else {
                    return file;
                }
            });
        }));
    }).then(function(results) {
        // flatten the array of arrays
        return Array.prototype.concat.apply([], results);
    });
}
iterate2(".").then(function(results) {
    console.log(results);
});

这是一个添加可自定义过滤器功能的版本：

function iterate2(dir, filterFn) {
    // default filter function accepts all files
    filterFn = filterFn || function() {return true;}
    return fs.readdirAsync(dir).then(function(list) {
        return Promise.all(list.map(function(file) {
            file = path.resolve(dir, file);
            return fs.statAsync(file).then(function(stat) {
                if (stat.isDirectory()) {
                    return iterate2(file, filterFn);
                } else {
                    return filterFn(file)? file : "";
                }
            });
        })).then(function(results) {
            return results.filter(function(f) {
                return !!f;
            });
        });
    }).then(function(results) {
        // flatten the array of arrays
        return Array.prototype.concat.apply([], results);
    });
}
// example usage
iterate2(".", function(f) {
    // filter out 
    return !(/(^|'/)'.[^'/'.]/g).test(f);
}).then(function(results) {
    console.log(results);
});
相关文章：