遍歷目錄是操作文件時(shí)的一個(gè)常見需求,比如寫一個(gè)程序,需要找到并處理指定目錄下的所有JS文件時(shí),就需要遍歷整個(gè)目錄。NodeJS遍歷目錄可以使用遞歸算法、遍歷算法,遍歷算法又分為同步遍歷、異步遍歷兩種，本文介紹NodeJS使用遞歸算法和遍歷算法來(lái)遍歷目錄的方法。

NodeJS遞歸算法

遍歷目錄時(shí)一般使用遞歸算法，否則就難以編寫出簡(jiǎn)潔的代碼。遞歸算法與數(shù)學(xué)歸納法類似，通過(guò)不斷縮小問(wèn)題的規(guī)模來(lái)解決問(wèn)題。以下示例說(shuō)明了這種方法。

function factorial(n) {
    if (n === 1) {
        return 1;
    } else {
        return n * factorial(n - 1);
    }
}

上邊的函數(shù)用于計(jì)算N的階乘（N!）?？梢钥吹剑?dāng)N大于1時(shí)，問(wèn)題簡(jiǎn)化為計(jì)算N乘以N-1的階乘。當(dāng)N等于1時(shí)，問(wèn)題達(dá)到最小規(guī)模，不需要再簡(jiǎn)化，因此直接返回1。

陷阱： 使用遞歸算法編寫的代碼雖然簡(jiǎn)潔，但由于每遞歸一次就產(chǎn)生一次函數(shù)調(diào)用，在需要優(yōu)先考慮性能時(shí)，需要把遞歸算法轉(zhuǎn)換為循環(huán)算法，以減少函數(shù)調(diào)用次數(shù)。

NodeJS遍歷算法

目錄是一個(gè)樹狀結(jié)構(gòu)，在遍歷時(shí)一般使用深度優(yōu)先+先序遍歷算法。深度優(yōu)先，意味著到達(dá)一個(gè)節(jié)點(diǎn)后，首先接著遍歷子節(jié)點(diǎn)而不是鄰居節(jié)點(diǎn)。先序遍歷，意味著首次到達(dá)了某節(jié)點(diǎn)就算遍歷完成，而不是最后一次返回某節(jié)點(diǎn)才算數(shù)。因此使用這種遍歷方式時(shí)，下邊這棵樹的遍歷順序是A > B > D > E > C > F。

          A
         / \
        B   C
       / \   \
      D   E   F

NodeJS同步遍歷

了解了必要的算法后，我們可以簡(jiǎn)單地實(shí)現(xiàn)以下目錄遍歷函數(shù)。

function travel(dir, callback) {
    fs.readdirSync(dir).forEach(function (file) {
        var pathname = path.join(dir, file);

        if (fs.statSync(pathname).isDirectory()) {
            travel(pathname, callback);
        } else {
            callback(pathname);
        }
    });
}

可以看到，該函數(shù)以某個(gè)目錄作為遍歷的起點(diǎn)。遇到一個(gè)子目錄時(shí)，就先接著遍歷子目錄。遇到一個(gè)文件時(shí)，就把文件的絕對(duì)路徑傳給回調(diào)函數(shù)?；卣{(diào)函數(shù)拿到文件路徑后，就可以做各種判斷和處理。因此假設(shè)有以下目錄：

- /home/user/
    - foo/
        x.js
    - bar/
        y.js
    z.css

使用以下代碼遍歷該目錄時(shí)，得到的輸入如下。

travel('/home/user', function (pathname) {
    console.log(pathname);
});

------------------------
/home/user/foo/x.js
/home/user/bar/y.js
/home/user/z.css

NodeJS異步遍歷

如果讀取目錄或讀取文件狀態(tài)時(shí)使用的是異步API，目錄遍歷函數(shù)實(shí)現(xiàn)起來(lái)會(huì)有些復(fù)雜，但原理完全相同。travel函數(shù)的異步版本如下。

function travel(dir, callback, finish) {
    fs.readdir(dir, function (err, files) {
        (function next(i) {
            if (i < files.length) {
                var pathname = path.join(dir, files[i]);

                fs.stat(pathname, function (err, stats) {
                    if (stats.isDirectory()) {
                        travel(pathname, callback, function () {
                            next(i + 1);
                        });
                    } else {
                        callback(pathname, function () {
                            next(i + 1);
                        });
                    }
                });
            } else {
                finish && finish();
            }
        }(0));
    });
}

到此這篇關(guān)于NodeJS使用遞歸算法和遍歷算法來(lái)遍歷目錄的方法的文章就介紹到這了,更多相關(guān)NodeJS遍歷目錄內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論