ES6 系列之非同步處理實戰

冴羽發表於2019-02-16

前言

我們以查詢指定目錄下的最大檔案為例,感受從

回撥函式 -> Promise -> Generator -> Async

非同步處理方式的改變。

API 介紹

為了實現這個功能,我們需要用到幾個 Nodejs 的 API,所以我們來簡單介紹一下。

fs.readdir

readdir 方法用於讀取目錄,返回一個包含檔案和目錄的陣列。

fs.stat

stat 方法的引數是一個檔案或目錄,它產生一個物件,該物件包含了該檔案或目錄的具體資訊。此外,該物件還有一個 isFile() 方法可以判斷正在處理的到底是一個檔案,還是一個目錄。

思路分析

我們基本的實現思路就是:

  1. fs.readdir 獲取指定目錄的內容資訊
  2. 迴圈遍歷內容資訊,使用 fs.stat 獲取該檔案或者目錄的具體資訊
  3. 將具體資訊儲存起來
  4. 當全部儲存起來後,篩選其中的是檔案的資訊
  5. 遍歷比較,找出最大檔案
  6. 獲取並返回最大檔案

然後我們直接上程式碼吧。

回撥函式

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

function findLargest(dir, cb) {
    // 讀取目錄下的所有檔案
    fs.readdir(dir, function(er, files) {
        if (er) return cb(er);

        var counter = files.length;
        var errored = false;
        var stats = [];

        files.forEach(function(file, index) {
            // 讀取檔案資訊
            fs.stat(path.join(dir, file), function(er, stat) {

                if (errored) return;

                if (er) {
                    errored = true;
                    return cb(er);
                }

                stats[index] = stat;

                // 事先算好有多少個檔案,讀完 1 個檔案資訊,計數減 1,當為 0 時,說明讀取完畢,此時執行最終的比較操作
                if (--counter == 0) {

                    var largest = stats
                        .filter(function(stat) { return stat.isFile() })
                        .reduce(function(prev, next) {
                            if (prev.size > next.size) return prev
                            return next
                        })

                    cb(null, files[stats.indexOf(largest)])
                }
            })
        })
    })
}

使用方式為:

// 查詢當前目錄最大的檔案
findLargest(`./`, function(er, filename) {
    if (er) return console.error(er)
    console.log(`largest file was:`, filename)
});

Promise

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

var readDir = function(dir) {
    return new Promise(function(resolve, reject) {
        fs.readdir(dir, function(err, files) {
            if (err) reject(err);
            resolve(files)
        })
    })
}

var stat = function(path) {
    return new Promise(function(resolve, reject) {
        fs.stat(path, function(err, stat) {
            if (err) reject(err)
            resolve(stat)
        })
    })
}

function findLargest(dir) {
    return readDir(dir)
        .then(function(files) {
            let promises = files.map(file => stat(path.join(dir, file)))
            return Promise.all(promises).then(function(stats) {
                return { stats, files }
            })
        })
        .then(data => {

            let largest = data.stats
                .filter(function(stat) { return stat.isFile() })
                .reduce((prev, next) => {
                    if (prev.size > next.size) return prev
                    return next
                })

            return data.files[data.stats.indexOf(largest)]
        })

}

使用方式為:

findLargest(`./`)
.then(function(filename) {
    console.log(`largest file was:`, filename);
})
.catch(function() {
    console.log(error);
});

Generator

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

var co = require(`co`)

var readDir = function(dir) {
    return new Promise(function(resolve, reject) {
        fs.readdir(dir, function(err, files) {
            if (err) reject(err);
            resolve(files)
        })
    })
}

var stat = function(path) {
    return new Promise(function(resolve, reject) {
        fs.stat(path, function(err, stat) {
            if (err) reject(err)
            resolve(stat)
        })
    })
}

function* findLargest(dir) {
    var files = yield readDir(dir);
    var stats = yield files.map(function(file) {
        return stat(path.join(dir, file))
    })

    let largest = stats
        .filter(function(stat) { return stat.isFile() })
        .reduce((prev, next) => {
            if (prev.size > next.size) return prev
            return next
        })

    return files[stats.indexOf(largest)]

}

使用方式為:

co(findLargest, `./`)
.then(function(filename) {
    console.log(`largest file was:`, filename);
})
.catch(function() {
    console.log(error);
});

Async

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

var readDir = function(dir) {
    return new Promise(function(resolve, reject) {
        fs.readdir(dir, function(err, files) {
            if (err) reject(err);
            resolve(files)
        })
    })
}

var stat = function(path) {
    return new Promise(function(resolve, reject) {
        fs.stat(path, function(err, stat) {
            if (err) reject(err)
            resolve(stat)
        })
    })
}

async function findLargest(dir) {
    var files = await readDir(dir);

    let promises = files.map(file => stat(path.join(dir, file)))
    var stats = await Promise.all(promises)

    let largest = stats
        .filter(function(stat) { return stat.isFile() })
        .reduce((prev, next) => {
            if (prev.size > next.size) return prev
            return next
        })

    return files[stats.indexOf(largest)]

}

使用方式為:

findLargest(`./`)
.then(function(filename) {
    console.log(`largest file was:`, filename);
})
.catch(function() {
    console.log(error);
});

ES6 系列

ES6 系列目錄地址:https://github.com/mqyqingfeng/Blog

ES6 系列預計寫二十篇左右,旨在加深 ES6 部分知識點的理解,重點講解塊級作用域、標籤模板、箭頭函式、Symbol、Set、Map 以及 Promise 的模擬實現、模組載入方案、非同步處理等內容。

如果有錯誤或者不嚴謹的地方,請務必給予指正,十分感謝。如果喜歡或者有所啟發,歡迎 star,對作者也是一種鼓勵。

相關文章