假設(shè)一個場景街佑,需要讀取兩個文件愉棱,然后把兩個文件的內(nèi)容倒序合并后輸出翎蹈,而且不能阻塞進程分别。
直接用回調(diào)來實現(xiàn)
var fs = require('fs');
var path1 = __dirname + '/package.json';
var path2 = __dirname + '/app.js';
console.log('start');
fs.readFile(path1, 'utf8', (err, content1) => {
if (err) throw err;
fs.readFile(path2, 'utf8', (err, content2) => {
if (err) throw err;
console.log(path2 + '\
' + content2 + '\
' + path1 + '\
' + content1);
});
});
console.log('end');
如果有 n 個文件要讀取遍愿,那就是回調(diào)地獄(Callback Hell)。
使用 Promise 實現(xiàn)
直接使用原生 Promise
var fs = require('fs');
var co = require('co');
var path1 = __dirname + '/package.json';
var path2 = __dirname + '/app.js';
console.log('start');
var readFilePromise = (path) => {
return new Promise((resolve, reject) => {
fs.readFile(path, 'utf8', (err, content) => {
if (err) reject(err);
resolve(content);
});
});
};
Promise.resolve().then(() => {
return readFilePromise(path1).then((content1) => {
return content1;
});
}).then((result)=>{
return readFilePromise(path2).then((content2)=>{
return path2 + '\
' + content2 + '\
' + path1 + '\
' + result;
});
}).then((result)=>{
console.log(result);
});
console.log('end');
readFilePromise 把 readfile 的 callback 轉(zhuǎn)換為Promise耘斩,下面就可以直接用 then 方法了沼填。
這里直接用 bluebird
var fs = require('fs');
var Promise = require('bluebird');
var path1 = __dirname + '/package.json';
var path2 = __dirname + '/app.js';
console.log('start');
Promise.promisifyAll(fs);
Promise.resolve().then(() => {
return fs.readFileAsync(path1, 'utf8').then((content) => {
return content;
});
}).then((result) => {
return fs.readFileAsync(path2, 'utf8').then((content) => {
return path2 + '\
' + content + '\
' + path1 + '\
' + result;
});
}).then((result) => {
console.log(result);
});
console.log('end');
這里的 fs.readFileAsync 方法已經(jīng)通過 bluebrid 轉(zhuǎn)換成了 Promise。
如果有 n 個文件需要煌往,那就需要加 n 個 then 方法倾哺。
改進的 Promise 實現(xiàn)方式
使用 Promise.all 方法
原生 Promise
var fs = require('fs');
var path1 = __dirname + '/package.json';
var path2 = __dirname + '/app.js';
console.log('start');
var readFilePromise = (path) => {
return new Promise((resolve, reject) => {
fs.readFile(path, 'utf8', (err, content) => {
if (err) reject(err);
resolve(content);
});
});
};
var files = [path1, path2];
var promises = [];
files.forEach((filePath) => {
promises.push(readFilePromise(filePath));
});
Promise.all(promises).then((result) => {
var contents = result.reverse();
var out=[];
files.reverse().forEach((filePath,index)=>{
out.push(filePath);
out.push(contents[index]);
});
console.log(out.join('\
'));
});
console.log('end');
bluebird 方式
var fs = require('fs');
var Promise = require('bluebird');
var path1 = __dirname + '/package.json';
var path2 = __dirname + '/app.js';
console.log('start');
Promise.promisifyAll(fs);
var files = [path1, path2];
var fileReadPromise = [];
files.forEach((filePath) => {
fileReadPromise.push(fs.readFileAsync(filePath, 'utf8'));
});
Promise.all(fileReadPromise).then((result) => {
var contents = result.reverse();
var out=[];
files.reverse().forEach((filePath,index)=>{
out.push(filePath);
out.push(contents[index]);
});
console.log(out.join('\
'));
});
console.log('end');
這樣的話,不管有多少文件刽脖,只要放入 files 數(shù)組即可。
兩種 Promise 實現(xiàn)是有區(qū)別的忌愚,第一種方式是順序讀取文件曲管,讀完一個再讀另外一個,第二種方式是同時開始讀取硕糊,全部完成后返回院水。
使用Generator實現(xiàn)
var fs = require('fs');
var path1 = __dirname + '/package.json';
var path2 = __dirname + '/app.js';
console.log('start');
var thunkReadFile = function (path) {
return function (cb) {
fs.readFile(path, 'utf8', (err, content) => {
cb(err, content);
});
};
};
function* readAllFile() {
var content1 = yield thunkReadFile(path1);
var content2 = yield thunkReadFile(path2);
console.log(path2 + '\
' + content2 + '\
' + path1 + '\
' + content1);
}
var rf = readAllFile();
var fn1 = rf.next().value;
fn1(function (err, con1) {
if (err) rf.throw(err);
var fn2 = rf.next(con1).value;
fn2(function (err, con2) {
if (err) rf.throw(err);
rf.next(con2);
});
});
console.log('end');
這樣寫不就還是 Callback Hell, 這是為了了解Generator.
function* 定義了一個 Generator function, 那么 rf 就是 Generator腊徙,它會有 next 方法,返回一個{value : (object),done: (boolean)}的對象檬某,其中的 value 就是 yield 后面返回的值撬腾。每次調(diào)用 next 就會向下執(zhí)行一個 yield,直到?jīng)]有 yield 后返回的對象里 done就為true恢恼,而每次調(diào)用 next 的參數(shù)會作為 yield 的返回值民傻。
以上面的代碼來說,
var fn = rf.next().value;
這里執(zhí)行后就到了第一個 yield 停住
var content1 = yield thunkReadFile(path1);
那么 next 返回的對象中的 value 就是 thunkReadFile(path1)场斑,這里已經(jīng)賦值給了 fn1漓踢。這里要注意 yield 還沒有返回,content1還沒有值漏隐。
第二次執(zhí)行 next 的時候
var fn2 = rf.next(con1).value;
得到了 thunkReadFile(path2)喧半,同時將第一個 yield 的值通過 next 的參數(shù)返回,content1現(xiàn)在是第一個文件的內(nèi)容.
第三次執(zhí)行 next 的時候
rf.next(con2);
將第二個 yield 的值返回給 content2,同時整個 Generator function 執(zhí)行完畢青责,輸出結(jié)果挺据。
當(dāng)然也可以用 Promise 來實現(xiàn)這個過程。但是有更簡單的方法脖隶,那就是 CO 庫吴菠。
CO 實現(xiàn)
var fs = require('fs');
var co = require('co');
var path1 = __dirname + '/package.json';
var path2 = __dirname + '/app.js';
console.log('start');
var trunkReadFile = function (path) {
return function (cb) {
fs.readFile(path, 'utf8', (err, content) => {
cb(err, content);
});
};
};
co(function* readAllFile() {
var content1 = yield trunkReadFile(path1);
var content2 = yield trunkReadFile(path2);
console.log(path2 + '\
' + content2 + '\
' + path1 + '\
' + content1);
});
console.log('end');
這樣有 n 個文件同樣也都寫入數(shù)組處理即可。fn1那部分其實就是 CO 的基本原理浩村,但是 CO 是用 Promise 實現(xiàn)的做葵。