nodejs 异步递归遍历文件

nodejs 同步递归遍历文件是非常简单的，异步递归遍历是就稍许复杂了一些

异步递归遍历可以采用类似迭代器的方式进行；首先定义一个方法，回调函数拥有三个参数分别是 err, file, next

function travelSync(dir, callback){
}
/**
 * 执行方法
 */
 travelSync('E:/runs', function (e, file, next) {
     //处理文件
    console.log(file);
    //获取下一个文件
    next();
});

当遍历完一个文件夹后，需要进行遍历下一个（next）文件，并且当前目录遍历完成后，若存在父级文件夹则遍历父级文件夹的未遍历的。那怎么判断已经遍历完成和遍历下一个文件？需要定位当前遍历位置，且当前目录遍历完后接着遍历下一个文件夹。可以在travelSync内部定义一个参数 i 用于记录当前位置，并在 travelSync 新增一个参数 finish 用于定位当前目录遍历完成进行遍历下一个父级目录文件


function travelSync(dir, callback, finish){
    fs.readdir(dir, function (e, files) {
            // i 用于定位当前遍历位置
            function next(i) {
                // 当i >= files 表示已经遍历完成，进行遍历下一个文件夹
                if(i < files.length){
                }else{
                    finish && finish();
                }
            };
    });
}

当前遍历对象是目录时进行递归调用，当前的位置通过封装在一个闭包里面传递到 travelSync 的 finish 参数中，当子目录遍历完成后，通过执行 finish 进行遍历下一个文件（finish中记录了遍历位置）。
遍历对象为文件时将文件对象（file）和当前遍历位置传递(next) 到回调方法 callback 中，并在回调方法 callback 处理完成后，执行 next 遍历下一个文件。
当 i >= files 时表示当前文件夹遍历完成，进行遍历父级目录执行 finish 方法，这里需要明白的是， finish 是递归调用时传进来的方法，实际finish 方法内部调用了 next(i) 方。因为首次调用时 finish 方法没有传入所以应判断 finish 是否存在。


function travelSync(dir, callback, finish){
    fs.readdir(dir, function (e, files) {
          // i 用于定位当前遍历位置
         (function next(i) {
              // 当i >= files 表示已经遍历完成，进行遍历下一个文件夹
             if(i < files.length){
                 var pathname = path.join(dir, files[i]);
                 if(fs.stat(pathname, function (e, stats) {
                     if(stats.isDirectory()){
                         travelSync(pathname, callback, function () {
                             next(i + 1);
                         });
                     }else{
                         callback(e, pathname, function () {
                             next(i + 1);
                         });
                     }
                 }));
             }else{
                 /**
                  * 当 i >= files.length 时，即表示当前目录已经遍历完了， 需遍历下一个文件夹
                  * 这里执行的时递归调用 传入的 方法 ， 方法里调用了 next(i) 记录了当前遍历位置
                  */
                 finish && finish();
             }

         })(0);
    });
}

 travelSync('E:/runs', function (e, file, next) {
     //处理文件
    console.log(file);
    //获取下一个文件
    next();
});

如此便可以进行正常的遍历了文件了，下面为完成的代码

var fs = require('fs');
var path = require('path');

function travelSync(dir, callback, finish){
    fs.readdir(dir, function (e, files) {
         if (e === null) {
            // i 用于定位当前遍历位置
            (function next(i) {
                 // 当i >= files 表示已经遍历完成，进行遍历下一个文件夹
                if(i < files.length){
                    var pathname = path.join(dir, files[i]);
                    if(fs.stat(pathname, function (e, stats) {
                        if(stats.isDirectory()){
                            travelSync(pathname, callback, function () {
                                next(i + 1);
                            });
                        }else{
                            callback(e, pathname, function () {
                                next(i + 1);
                            });
                        }
                    }));
                }else{
                    /**
                     * 当 i >= files.length 时，即表示当前目录已经遍历完了， 需遍历下一个文件夹
                     * 这里执行的时递归调用 传入的 方法 ， 方法里调用了 next(i) 记录了当前遍历位置
                     */
                    finish && finish();
                }

            })(0);
        }else{
            callback(e);
        }
    });
}
travelSync('E:/runs', function (e, file, next) {
    if(e !== null){
        console.log(e); 
    }
    console.log(file);
    //获取下一个文件 next 里面调用了 next(i) 记录了当前遍历位置
    next();
});

nodejs 异步递归遍历文件

猜你喜欢