ES(2017)學習筆記(十二)【Async】

風靈使發表於2019-04-06

async 函式

含義

ES2017 標準引入了 async 函式,使得非同步操作變得更加方便。

async 函式是什麼,一句話,它就是 Generator 函式的語法糖。

前文有個 Generator 函式,依次讀取兩個檔案。

const fs = require('fs');

const readFile = function (fileName) {
  return new Promise(function (resolve, reject) {
    fs.readFile(fileName, function(error, data) {
      if (error) return reject(error);
      resolve(data);
    });
  });
};

const gen = function* () {
  const f1 = yield readFile('/etc/fstab');
  const f2 = yield readFile('/etc/shells');
  console.log(f1.toString());
  console.log(f2.toString());
};

寫成 async函式,就是下面這樣

const asyncReadFile = async funtion() {
    const f1 = await readFile('/etc/fstab/');
    const f2 = await readFile('/etc/shells');
    console.log(f1.toString());
    console.log(f2.toString());
}

一比較就會發現,async函式就是將 Generator 函式的星號(*)替換成async,將yield替換成await,僅此而已。

async函式對 Generator 函式的改進,體現在以下四點。

(1)內建執行器。

Generator 函式的執行必須靠執行器,所以才有了co模組,而async函式自帶執行器。也就是說,async函式的執行,與普通函式一模一樣,只要一行。

asyncReadFile();

上面的程式碼呼叫了asyncReadFile函式,然後它就會自動執行,輸出最後結果。這完全不像 Generator 函式,需要呼叫next方法,或者用co模組,才能真正執行,得到最後結果。

(2)更好的語義。

asyncawait,比起星號和yield,語義更清楚了。async表示函式裡有非同步操作,await表示緊跟在後面的表示式需要等待結果。

(3)更廣的適用性。

co模組約定,yield命令後面只能是 Thunk 函式或 Promise 物件,而async函式的await命令後面,可以是 Promise 物件和原始型別的值(數值、字串和布林值,但這時等同於同步操作)。

(4)返回值是 Promise。

async函式的返回值是 Promise 物件,這比 Generator 函式的返回值是 Iterator 物件方便多了。你可以用then方法指定下一步的操作。

進一步說,async函式完全可以看作多個非同步操作,包裝成的一個 Promise 物件,而await命令就是內部then命令的語法糖。


基本用法

async函式返回一個 Promise 物件,可以使用 then方法新增回撥函式。當函式執行的時候,一旦遇到 await就會先返回等到非同步操作完成,再接著執行函式體內後面的語句。

下面是一個例子

async function getStockPriceByName(name) {
  const symbol = await getStockSymbol(name);
  const stockPrice = await getStockPrice(symbol);
  return stockPrice;
}

getStockPriceByName('goog').then(function (result) {
  console.log(result);
});

上面程式碼是一個獲取股票報價的函式,函式前面的async關鍵字,表明該函式內部有非同步操作。呼叫該函式時,會立即返回一個Promise物件。

下面是另一個例子,指定多少毫秒後輸出一個值。

function timeout(ms) {
  return new Promise((resolve) => {
    setTimeout(resolve, ms);
  });
}

async function asyncPrint(value, ms) {
  await timeout(ms);
  console.log(value);
}

asyncPrint('hello world', 50);

上面程式碼指定 50 毫秒以後,輸出hello world

由於async函式返回的是 Promise 物件,可以作為await命令的引數。所以,上面的例子也可以寫成下面的形式。

async function timeout(ms) {
  await new Promise((resolve) => {
    setTimeout(resolve, ms);
  });
}

async function asyncPrint(value, ms) {
  await timeout(ms);
  console.log(value);
}

asyncPrint('hello world', 50);

async 函式有多種使用形式。

// 函式宣告
async function foo() {}

// 函式表示式
const foo = async function () {};

// 物件的方法
let obj = { async foo() {} };
obj.foo().then(...)

// Class 的方法
class Storage {
  constructor() {
    this.cachePromise = caches.open('avatars');
  }

  async getAvatar(name) {
    const cache = await this.cachePromise;
    return cache.match(`/avatars/${name}.jpg`);
  }
}

const storage = new Storage();
storage.getAvatar('jake').then();

// 箭頭函式
const foo = async () => {};

語法

async函式的語法規則總體上比較簡單,難點是錯誤處理機制

返回 Promise 物件

async函式返回一個 Promise 物件。

async函式內部return語句返回的值,會成為then方法回撥函式的引數。

async function f(){
    return 'hello world';
}
f().then( v => console.log(v));

上面程式碼中,函式f內部return命令返回的值,會被then方法回撥函式接收到。

async函式內部丟擲錯誤,會導致返回的 Promise 物件變為reject狀態。丟擲的錯誤物件會被catch方法回撥函式接收到。

async function f() {
  throw new Error('出錯了');
}

f().then(
  v => console.log(v),
  e => console.log(e)
)
// Error: 出錯了

Promise 物件的狀態變化

async函式返回的 Promise 物件,必須等到內部所有await命令後面的 Promise 物件執行完,才會發生狀態改變,除非遇到return語句或者丟擲錯誤。也就是說,只有async函式內部的非同步操作執行完,才會執行then方法指定的回撥函式。

下面是一個例子。

async function getTitle(url){
    let response = await fetch(url);
    let html = await response.text();
    return html.match(/<title>([\s\S]+)<\/title>/i)[1];
}
getTitle('https://tc39.github.io/ecma262/').then(console.log)
// ECMAScript® 2019 Language&nbsp;Specification

await 命令

正常情況下,await命令後面是一個 Promise 物件。如果不是,會被轉成一個立即resolve的 Promise 物件。

async funtion f(){
    return await 123;
}
f().then(v => console.log(v));

上面程式碼中,await命令的引數是數值123,它被轉成 Promise 物件,並立即resolve

await命令後面的 Promise 物件如果變成 reject狀態,則 reject的引數會被 catch方法的回撥函式接收到。

async funtion f(){
    await Promise.reject('出錯了');
}
f().then(v => console.log(v))
   .catch(e => console.log(e))
// 出錯

注意,上面程式碼中,await語句前面沒有return,但是reject方法的引數依然傳入了catch方法的回撥函式。這裡如果在await前面加上return,效果是一樣的。

只要一個await語句後面的 Promise 變為reject,那麼整個async函式都會中斷執行。

async function f() {
  await Promise.reject('出錯了');
  await Promise.resolve('hello world'); // 不會執行
}

上面程式碼中,第二個await語句是不會執行的,因為第一個await語句狀態變成了reject

有時,我們希望即使前一個非同步操作失敗,也不要中斷後面的非同步操作。這時可以將第一個await放在try...catch結構裡面,這樣不管這個非同步操作是否成功,第二個await都會執行。

async function f() {
  try {
    await Promise.reject('出錯了');
  } catch(e) {
  }
  return await Promise.resolve('hello world');
}

f()
.then(v => console.log(v))
// hello world

另一種方法是await後面的 Promise 物件再跟一個catch方法,處理前面可能出現的錯誤。

async function f() {
  await Promise.reject('出錯了')
    .catch(e => console.log(e));
  return await Promise.resolve('hello world');
}

f()
.then(v => console.log(v))
// 出錯了
// hello world

錯誤處理

如果await後面的非同步操作出錯,那麼等同於async函式返回的 Promise 物件被reject

async function f() {
  await new Promise(function (resolve, reject) {
    throw new Error('出錯了');
  });
}

f()
.then(v => console.log(v))
.catch(e => console.log(e))
// Error:出錯了

上面程式碼中,async函式f執行後,await後面的 Promise 物件會丟擲一個錯誤物件,導致catch方法的回撥函式被呼叫,它的引數就是丟擲的錯誤物件。具體的執行機制,可以參考後文的“async 函式的實現原理”。

防止出錯的方法,也是將其放在try...catch程式碼塊之中。

async function f() {
  try {
    await new Promise(function (resolve, reject) {
      throw new Error('出錯了');
    });
  } catch(e) {
  }
  return await('hello world');
}

如果有多個await命令,可以統一放在try...catch結構中。

async function main() {
  try {
    const val1 = await firstStep();
    const val2 = await secondStep(val1);
    const val3 = await thirdStep(val1, val2);

    console.log('Final: ', val3);
  }
  catch (err) {
    console.error(err);
  }
}

下面的例子使用try...catch結構,實現多次重複嘗試。

const superagent = require('superagent');
const NUM_RETRIES = 3;

async function test() {
  let i;
  for (i = 0; i < NUM_RETRIES; ++i) {
    try {
      await superagent.get('http://google.com/this-throws-an-error');
      break;
    } catch(err) {}
  }
  console.log(i); // 3
}

test();

上面程式碼中,如果await操作成功,就會使用break語句退出迴圈;如果失敗,會被catch語句捕捉,然後進入下一輪迴圈。

使用注意點

第一點,前面已經說過,await命令後面的Promise物件,執行結果可能是rejected,所以最好把await命令放在try...catch程式碼塊中。

async function myFunction(){
    try{
        await somethingThatReturnsAPromise();
    } catch(e){
        console.log(e);
    }
}
// 另一種寫法

async function myFunction(){
    await somethingThatReturnAPromise()
        .catch(function(e){
        console.log(e);
    })
}

第二點,多個await命令後面的非同步操作,如果不存在繼發關係,最好讓它們同時觸發。

let foo = await getFoo();
let bar = await getBar();

上面程式碼中,getFoogetBar是兩個獨立的非同步操作(即互不依賴),被寫成繼發關係。這樣比較耗時,因為只有getFoo完成以後,才會執行getBar,完全可以讓它們同時觸發。

// 寫法一
let [foo, bar] = await Promise.all([getFoo(), getBar()]);

// 寫法二
let fooPromise = getFoo();
let barPromise = getBar();
let foo = await fooPromise;
let bar = await barPromise;

上面兩種寫法,getFoogetBar都是同時觸發,這樣就會縮短程式的執行時間。

第三點,await命令只能用在async函式之中,如果用在普通函式,就會報錯。

async function dbFuc(db) {
  let docs = [{}, {}, {}];

  // 報錯
  docs.forEach(function (doc) {
    await db.post(doc);
  });
}

上面程式碼會報錯,因為await用在普通函式之中了。但是,如果將forEach方法的引數改成async函式,也有問題。

function dbFuc(db) { //這裡不需要 async
  let docs = [{}, {}, {}];

  // 可能得到錯誤結果
  docs.forEach(async function (doc) {
    await db.post(doc);
  });
}

上面程式碼可能不會正常工作,原因是這時三個db.post操作將是併發執行,也就是同時執行,而不是繼發執行。正確的寫法是採用for迴圈。

async function dbFuc(db) {
  let docs = [{}, {}, {}];

  for (let doc of docs) {
    await db.post(doc);
  }
}

如果確實希望多個請求併發執行,可以使用Promise.all方法。當三個請求都會resolved時,下面兩種寫法效果相同。

async function dbFuc(db) {
  let docs = [{}, {}, {}];
  let promises = docs.map((doc) => db.post(doc));

  let results = await Promise.all(promises);
  console.log(results);
}

// 或者使用下面的寫法

async function dbFuc(db) {
  let docs = [{}, {}, {}];
  let promises = docs.map((doc) => db.post(doc));

  let results = [];
  for (let promise of promises) {
    results.push(await promise);
  }
  console.log(results);
}

目前,esm模組載入器支援頂層await,即await命令可以不放在 async 函式裡面,直接使用。

// async 函式的寫法
const start = async () => {
  const res = await fetch('google.com');
  return res.text();
};

start().then(console.log);

// 頂層 await 的寫法
const res = await fetch('google.com');
console.log(await res.text());

上面程式碼中,第二種寫法的指令碼必須使用esm載入器,才會生效。


async 函式的實現原理

async 函式的實現原理,就是將 Generator 函式和自動執行器,包裝在一個函式裡。

async function fn(args){
    // ...
}

// 等同於
function fn(args){
    return span(function* (){
        // ...
    })
}

所有的async函式都可以寫成上面的第二種形式,其中的spawn函式就是自動執行器。

下面給出spawn函式的實現,基本就是前文自動執行器的翻版。

function spawn(genF) {
  return new Promise(function(resolve, reject) {
    const gen = genF();
    function step(nextF) {
      let next;
      try {
        next = nextF();
      } catch(e) {
        return reject(e);
      }
      if(next.done) {
        return resolve(next.value);
      }
      Promise.resolve(next.value).then(function(v) {
        step(function() { return gen.next(v); });
      }, function(e) {
        step(function() { return gen.throw(e); });
      });
    }
    step(function() { return gen.next(undefined); });
  });
}

與其他非同步處理方法的比較

我們通過一個例子,來看 async 函式與 Promise、Generator 函式的比較。

假定某個 DOM 元素上面,部署了一系列的動畫,前一個動畫結束,才能開始後一個。如果當中有一個動畫出錯,就不再往下執行,返回上一個成功執行的動畫的返回值。

首先是 Promise 的寫法。

function chainAnimationsPromise(elem, animations) {

  // 變數ret用來儲存上一個動畫的返回值
  let ret = null;

  // 新建一個空的Promise
  let p = Promise.resolve();

  // 使用then方法,新增所有動畫
  for(let anim of animations) {
    p = p.then(function(val) {
      ret = val;
      return anim(elem);
    });
  }

  // 返回一個部署了錯誤捕捉機制的Promise
  return p.catch(function(e) {
    /* 忽略錯誤,繼續執行 */
  }).then(function() {
    return ret;
  });

}

雖然 Promise 的寫法比回撥函式的寫法大大改進,但是一眼看上去,程式碼完全都是 Promise 的 API (thencatch等等),操作本身的語義反而不容易看出來。

接著是 Generator 函式的寫法

function chainAnimationsGenerator(elem, animations){
    return spawn(function* (){
        let ret = null;
        try{
            for( let anim of animations){
                ret = yield anim(elem);
            }
        } catch(e){
            /* 忽略錯誤,繼續執行 */
        }
        return ret;
    });
}

上面程式碼使用 Generator 函式遍歷了每個動畫,語義比 Promise 寫法更清晰,使用者定義的操作全部都出現在spawn函式的內部。這個寫法的問題在於,必須有一個任務執行器,自動執行 Generator 函式,上面程式碼的spawn函式就是自動執行器,它返回一個 Promise 物件,而且必須保證yield語句後面的表示式,必須返回一個 Promise。

最後是 async 函式的寫法。

async function chainAnimationsAsync(elem, animations){
    let ret = null;
    try{
        for(let anim of animations){
            ret = await anim(elem);
        }
    } catch(e){
        /* 忽略錯誤,繼續執行 */
    }
    return ret;
}

可以看到 Async 函式的實現最簡潔,最符合語義,幾乎沒有語義不相關的程式碼。它將 Generator 寫法中的自動執行器,改在語言層面提供,不暴露給使用者,因此程式碼量最少。如果使用 Generator 寫法,自動執行器需要使用者自己提供。


例項:按順序完成非同步操作

實際開發中,經常遇到一組非同步操作,需要按照順序完成。比如,依次遠端讀取一組 URL,然後按照讀取的順序輸出結果。

Promise 的寫法如下。

function logInOrder(urls){
    // 遠端讀取所有URL
    const textPromises = urls.map(url =>{
        return fetch(url).then(response => response.text());
    });
    
    // 按次序輸出
    textPromises.reduce((chain, textPromise) =>{
        return chain.then(() => textPromise)
        .then(text => console.log(text));
    }, Promise.resolve());
}

上面程式碼使用fetch方法,同時遠端讀取一組 URL。每個fetch操作都返回一個 Promise 物件,放入textPromises陣列。然後,reduce方法依次處理每個 Promise 物件,然後使用then,將所有 Promise 物件連起來,因此就可以依次輸出結果。

這種寫法不太直觀,可讀性比較差。下面是 async 函式實現。

async function logInOrder(urls){
    for(const url of urls){
        const response = await fetch(url);
        console.log(await response.text());
    }
}

上面程式碼確實大大簡化,問題是所有遠端操作都是繼發。只有前一個 URL 返回結果,才會去讀取下一個 URL,這樣做效率很差,非常浪費時間。我們需要的是併發發出遠端請求。

sync function logInOrder(urls) {
  // 併發讀取遠端URL
  const textPromises = urls.map(async url => {
    const response = await fetch(url);
    return response.text();
  });

  // 按次序輸出
  for (const textPromise of textPromises) {
    console.log(await textPromise);
  }
}

上面程式碼中,雖然map方法的引數是async函式,但它是併發執行的,因為只有async函式內部是繼發執行,外部不受影響。後面的for..of迴圈內部使用了await,因此實現了按順序輸出。


非同步遍歷器

《遍歷器》一章說過,Iterator 介面是一種資料遍歷的協議,只要呼叫遍歷器物件的next方法,就會得到一個物件,表示當前遍歷指標所在的那個位置的資訊。next方法返回的物件的結構是{value, done},其中value表示當前的資料的值,done是一個布林值,表示遍歷是否結束。

這裡隱含著一個規定,next方法必須是同步的,只要呼叫就必須立刻返回值。也就是說,一旦執行next方法,就必須同步地得到valuedone這兩個屬性。如果遍歷指標正好指向同步操作,當然沒有問題,但對於非同步操作,就不太合適了。目前的解決方法是,Generator 函式裡面的非同步操作,返回一個 Thunk 函式或者 Promise 物件,即value屬性是一個 Thunk 函式或者 Promise 物件,等待以後返回真正的值,而done屬性則還是同步產生的。

ES2018 引入了”非同步遍歷器“(Async Iterator),為非同步操作提供原生的遍歷器介面,即valuedone這兩個屬性都是非同步產生。

非同步遍歷的介面

非同步遍歷器的最大的語法特點,就是呼叫遍歷器的next方法,返回的是一個 Promise 物件。

asyncIterator
  .next()
  .then(
    ({ value, done }) => /* ... */
  );

上面程式碼中,asyncIterator是一個非同步遍歷器,呼叫next方法以後,返回一個 Promise 物件。因此,可以使用then方法指定,這個 Promise 物件的狀態變為resolve以後的回撥函式。回撥函式的引數,則是一個具有valuedone兩個屬性的物件,這個跟同步遍歷器是一樣的。

我們知道,一個物件的同步遍歷器的介面,部署在Symbol.iterator屬性上面。同樣地,物件的非同步遍歷器介面,部署在Symbol.asyncIterator屬性上面。不管是什麼樣的物件,只要它的Symbol.asyncIterator屬性有值,就表示應該對它進行非同步遍歷。

下面是一個非同步遍歷器的例子。

const asyncIterable = createAsyncIterable(['a', 'b']);
const asyncIterator = asyncIterable[Symbol.asyncIterator]();

asyncIterator
.next()
.then(iterResult1 => {
  console.log(iterResult1); // { value: 'a', done: false }
  return asyncIterator.next();
})
.then(iterResult2 => {
  console.log(iterResult2); // { value: 'b', done: false }
  return asyncIterator.next();
})
.then(iterResult3 => {
  console.log(iterResult3); // { value: undefined, done: true }
});

上面程式碼中,非同步遍歷器其實返回了兩次值。第一次呼叫的時候,返回一個 Promise 物件;等到 Promise 物件resolve了,再返回一個表示當前資料成員資訊的物件。這就是說,非同步遍歷器與同步遍歷器最終行為是一致的,只是會先返回 Promise 物件,作為中介。

由於非同步遍歷器的next方法,返回的是一個 Promise 物件。因此,可以把它放在await命令後面。

async function f() {
  const asyncIterable = createAsyncIterable(['a', 'b']);
  const asyncIterator = asyncIterable[Symbol.asyncIterator]();
  console.log(await asyncIterator.next());
  // { value: 'a', done: false }
  console.log(await asyncIterator.next());
  // { value: 'b', done: false }
  console.log(await asyncIterator.next());
  // { value: undefined, done: true }
}

上面程式碼中,next方法用await處理以後,就不必使用then方法了。整個流程已經很接近同步處理了。

注意,非同步遍歷器的next方法是可以連續呼叫的,不必等到上一步產生的 Promise 物件resolve以後再呼叫。這種情況下,next方法會累積起來,自動按照每一步的順序執行下去。下面是一個例子,把所有的next方法放在Promise.all方法裡面。

const asyncIterable = createAsyncIterable(['a', 'b']);
const asyncIterator = asyncIterable[Symbol.asyncIterator]();
const [{value: v1}, {value: v2}] = await Promise.all([
  asyncIterator.next(), asyncIterator.next()
]);

console.log(v1, v2); // a b

另一種用法是一次性呼叫所有的next方法,然後await最後一步操作。

async function runner() {
  const writer = openFile('someFile.txt');
  writer.next('hello');
  writer.next('world');
  await writer.return();
}

runner();

for await…of

前面介紹過,for...of迴圈用於遍歷同步的 Iterator 介面。新引入的for await...of迴圈,則是用於遍歷非同步的 Iterator 介面。

async function f() {
  for await (const x of createAsyncIterable(['a', 'b'])) {
    console.log(x);
  }
}
// a
// b

上面程式碼中,createAsyncIterable()返回一個擁有非同步遍歷器介面的物件,for...of迴圈自動呼叫這個物件的非同步遍歷器的next方法,會得到一個 Promise 物件。await用來處理這個 Promise 物件,一旦resolve,就把得到的值(x)傳入for...of的迴圈體。

for await...of迴圈的一個用途,是部署了 asyncIterable 操作的非同步介面,可以直接放入這個迴圈。

let body = '';

async function f() {
  for await(const data of req) body += data;
  const parsed = JSON.parse(body);
  console.log('got', parsed);
}

上面程式碼中,req是一個 asyncIterable 物件,用來非同步讀取資料。可以看到,使用for await...of迴圈以後,程式碼會非常簡潔。

如果next方法返回的 Promise 物件被rejectfor await...of就會報錯,要用try...catch捕捉。

async function () {
  try {
    for await (const x of createRejectingIterable()) {
      console.log(x);
    }
  } catch (e) {
    console.error(e);
  }
}

注意,for await...of迴圈也可以用於同步遍歷器。

(async function () {
  for await (const x of ['a', 'b']) {
    console.log(x);
  }
})();
// a
// b

Node v10 支援非同步遍歷器,Stream 就部署了這個介面。下面是讀取檔案的傳統寫法與非同步遍歷器寫法的差異。

// 傳統寫法
function main(inputFilePath) {
  const readStream = fs.createReadStream(
    inputFilePath,
    { encoding: 'utf8', highWaterMark: 1024 }
  );
  readStream.on('data', (chunk) => {
    console.log('>>> '+chunk);
  });
  readStream.on('end', () => {
    console.log('### DONE ###');
  });
}

// 非同步遍歷器寫法
async function main(inputFilePath) {
  const readStream = fs.createReadStream(
    inputFilePath,
    { encoding: 'utf8', highWaterMark: 1024 }
  );

  for await (const chunk of readStream) {
    console.log('>>> '+chunk);
  }
  console.log('### DONE ###');
}

非同步 Generator 函式

就像 Generator 函式返回一個同步遍歷器物件一樣,非同步 Generator 函式的作用,是返回一個非同步遍歷器物件。

在語法上,非同步 Generator 函式就是async函式與 Generator 函式的結合。

async function* gen() {
  yield 'hello';
}
const genObj = gen();
genObj.next().then(x => console.log(x));
// { value: 'hello', done: false }

上面程式碼中,gen是一個非同步 Generator 函式,執行後返回一個非同步 Iterator 物件。對該物件呼叫next方法,返回一個 Promise 物件。

非同步遍歷器的設計目的之一,就是 Generator 函式處理同步操作和非同步操作時,能夠使用同一套介面。

// 同步 Generator 函式
function* map(iterable, func) {
  const iter = iterable[Symbol.iterator]();
  while (true) {
    const {value, done} = iter.next();
    if (done) break;
    yield func(value);
  }
}

// 非同步 Generator 函式
async function* map(iterable, func) {
  const iter = iterable[Symbol.asyncIterator]();
  while (true) {
    const {value, done} = await iter.next();
    if (done) break;
    yield func(value);
  }
}

上面程式碼中,map是一個 Generator 函式,第一個引數是可遍歷物件iterable,第二個引數是一個回撥函式funcmap的作用是將iterable每一步返回的值,使用func進行處理。上面有兩個版本的map,前一個處理同步遍歷器,後一個處理非同步遍歷器,可以看到兩個版本的寫法基本上是一致的。

下面是另一個非同步 Generator 函式的例子。

async function* readLines(path) {
  let file = await fileOpen(path);

  try {
    while (!file.EOF) {
      yield await file.readLine();
    }
  } finally {
    await file.close();
  }
}

上面程式碼中,非同步操作前面使用await關鍵字標明,即await後面的操作,應該返回 Promise 物件。凡是使用yield關鍵字的地方,就是next方法停下來的地方,它後面的表示式的值(即await file.readLine()的值),會作為next()返回物件的value屬性,這一點是與同步 Generator 函式一致的。

非同步 Generator 函式內部,能夠同時使用awaityield命令。可以這樣理解,await命令用於將外部操作產生的值輸入函式內部,yield命令用於將函式內部的值輸出。

上面程式碼定義的非同步 Generator 函式的用法如下。

(async function () {
  for await (const line of readLines(filePath)) {
    console.log(line);
  }
})()

非同步 Generator 函式可以與for await...of迴圈結合起來使用。

async function* prefixLines(asyncIterable) {
  for await (const line of asyncIterable) {
    yield '> ' + line;
  }
}

非同步 Generator 函式的返回值是一個非同步 Iterator,即每次呼叫它的next方法,會返回一個 Promise 物件,也就是說,跟在yield命令後面的,應該是一個 Promise 物件。如果像上面那個例子那樣,yield命令後面是一個字串,會被自動包裝成一個 Promise 物件。

function fetchRandom() {
  const url = 'https://www.random.org/decimal-fractions/'
    + '?num=1&dec=10&col=1&format=plain&rnd=new';
  return fetch(url);
}

async function* asyncGenerator() {
  console.log('Start');
  const result = await fetchRandom(); // (A)
  yield 'Result: ' + await result.text(); // (B)
  console.log('Done');
}

const ag = asyncGenerator();
ag.next().then(({value, done}) => {
  console.log(value);
})

上面程式碼中,agasyncGenerator函式返回的非同步遍歷器物件。呼叫ag.next()以後,上面程式碼的執行順序如下。

  1. ag.next()立刻返回一個 Promise 物件。
  2. asyncGenerator函式開始執行,列印出Start
  3. await命令返回一個 Promise 物件,asyncGenerator函式停在這裡。
  4. A 處變成 fulfilled 狀態,產生的值放入result變數,asyncGenerator函式繼續往下執行。
  5. 函式在 B 處的yield暫停執行,一旦yield命令取到值,ag.next()返回的那個 Promise 物件變成 fulfilled 狀態。
  6. ag.next()後面的then方法指定的回撥函式開始執行。該回撥函式的引數是一個物件{value, done},其中value的值是yield命令後面的那個表示式的值,done的值是false

A 和 B 兩行的作用類似於下面的程式碼。

return new Promise((resolve, reject) => {
  fetchRandom()
  .then(result => result.text())
  .then(result => {
     resolve({
       value: 'Result: ' + result,
       done: false,
     });
  });
});

如果非同步 Generator 函式丟擲錯誤,會導致 Promise 物件的狀態變為reject,然後丟擲的錯誤被catch方法捕獲。

async function* asyncGenerator() {
  throw new Error('Problem!');
}

asyncGenerator()
.next()
.catch(err => console.log(err)); // Error: Problem!

注意,普通的 async 函式返回的是一個 Promise 物件,而非同步 Generator 函式返回的是一個非同步 Iterator 物件。可以這樣理解,async 函式和非同步 Generator 函式,是封裝非同步操作的兩種方法,都用來達到同一種目的。區別在於,前者自帶執行器,後者通過for await...of執行,或者自己編寫執行器。下面就是一個非同步 Generator 函式的執行器。

async function takeAsync(asyncIterable, count = Infinity) {
  const result = [];
  const iterator = asyncIterable[Symbol.asyncIterator]();
  while (result.length < count) {
    const {value, done} = await iterator.next();
    if (done) break;
    result.push(value);
  }
  return result;
}

上面程式碼中,非同步 Generator 函式產生的非同步遍歷器,會通過while迴圈自動執行,每當await iterator.next()完成,就會進入下一輪迴圈。一旦done屬性變為true,就會跳出迴圈,非同步遍歷器執行結束。

下面是這個自動執行器的一個使用例項。

async function f() {
  async function* gen() {
    yield 'a';
    yield 'b';
    yield 'c';
  }

  return await takeAsync(gen());
}

f().then(function (result) {
  console.log(result); // ['a', 'b', 'c']
})

非同步 Generator 函式出現以後,JavaScript 就有了四種函式形式:普通函式、async 函式、Generator 函式和非同步 Generator 函式。請注意區分每種函式的不同之處。基本上,如果是一系列按照順序執行的非同步操作(比如讀取檔案,然後寫入新內容,再存入硬碟),可以使用 async 函式;如果是一系列產生相同資料結構的非同步操作(比如一行一行讀取檔案),可以使用非同步 Generator 函式。

非同步 Generator 函式也可以通過next方法的引數,接收外部傳入的資料。

const writer = openFile('someFile.txt');
writer.next('hello'); // 立即執行
writer.next('world'); // 立即執行
await writer.return(); // 等待寫入結束

上面程式碼中,openFile是一個非同步 Generator 函式。next方法的引數,向該函式內部的操作傳入資料。每次next方法都是同步執行的,最後的await命令用於等待整個寫入操作結束。

最後,同步的資料結構,也可以使用非同步 Generator 函式。

async function* createAsyncIterable(syncIterable) {
  for (const elem of syncIterable) {
    yield elem;
  }
}

上面程式碼中,由於沒有非同步操作,所以也就沒有使用await關鍵字。

yield* 語句

yield*語句也可以跟一個非同步遍歷器。

async function* gen1() {
  yield 'a';
  yield 'b';
  return 2;
}

async function* gen2() {
  // result 最終會等於 2
  const result = yield* gen1();
}

上面程式碼中,gen2函式裡面的result變數,最後的值是2

與同步 Generator 函式一樣,for await...of迴圈會展開yield*

(async function () {
  for await (const x of gen2()) {
    console.log(x);
  }
})();
// a
// b

相關文章