如何更好的編寫async函數_ZenDei技術網路在線

2018年已經到了5月份，node的4.x版本也已經停止了維護我司的某個服務也已經切到了8.x，目前正在做koa2.x的遷移將之前的generator全部替換為async 但是，在替換的過程中，發現一些濫用async導致的時間上的浪費所以來談一下，如何優化async代碼，更充分的利用非同步事件流 ...

2018年已經到了5月份，node的4.x版本也已經停止了維護
我司的某個服務也已經切到了8.x，目前正在做koa2.x的遷移
將之前的generator全部替換為async
但是，在替換的過程中，發現一些濫用async導致的時間上的浪費所以來談一下，如何優化async代碼，更充分的利用非同步事件流 杜絕濫用async

首先，你需要瞭解Promise

Promise是使用async/await的基礎，所以你一定要先瞭解Promise是做什麼的
Promise是幫助解決回調地獄的一個好東西，能夠讓非同步流程變得更清晰。
一個簡單的Error-first-callback轉換為Promise的例子：

const fs = require('fs')

function readFile (fileName) {
  return new Promise((resolve, reject) => {
    fs.readFile(fileName, (err, data) => {
      if (err) reject(err)

      resolve(data)
    })
  })
}

readFile('test.log').then(data => {
  console.log('get data')
}, err => {
  console.error(err)
})

我們調用函數返回一個Promise的實例，在實例化的過程中進行文件的讀取，當文件讀取的回調觸髮式，進行Promise狀態的變更，resolved或者rejected
狀態的變更我們使用then來監聽，第一個回調為resolve的處理，第二個回調為reject的處理。

async與Promise的關係

async函數相當於一個簡寫的返回Promise實例的函數，效果如下：

function getNumber () {
  return new Promise((resolve, reject) => {
    resolve(1)
  })
}
// =>
async function getNumber () {
  return 1
}

兩者在使用上方式上完全一樣，都可以在調用getNumber函數後使用then進行監聽返回值。以及與async對應的await語法的使用方式：

getNumber().then(data => {
  // got data
})
// =>
let data = await getNumber()

await的執行會獲取表達式後邊的Promise執行結果，相當於我們調用then獲取回調結果一樣。 P.S. 在async/await支持度還不是很高的時候，大家都會選擇使用generator/yield結合著一些類似於co的庫來實現類似的效果

async函數代碼執行是同步的，結果返回是非同步的

async函數總是會返回一個Promise的實例 這點兒很重要
所以說調用一個async函數時，可以理解為裡邊的代碼都是處於new Promise中，所以是同步執行的
而最後return的操作，則相當於在Promise中調用resolve：

async function getNumber () {
  console.log('call getNumber()')

  return 1
}

getNumber().then(_ => console.log('resolved'))
console.log('done')

// 輸出順序：
// call getNumber()
// done
// resolved

Promise內部的Promise會被消化

也就是說，如果我們有如下的代碼：

function getNumber () {
  return new Promise(resolve => {
    resolve(Promise.resolve(1))
  })
}

getNumber().then(data => console.log(data)) // 1

如果按照上邊說的話，我們在then裡邊獲取到的data應該是傳入resolve中的值，也就是另一個Promise的實例。
但實際上，我們會直接獲得返回值：1，也就是說，如果在Promise中返回一個Promise，實際上程式會幫我們執行這個Promise，併在內部的Promise狀態改變時觸發then之類的回調。
一個有意思的事情：

function getNumber () {
  return new Promise(resolve => {
    resolve(Promise.reject(new Error('Test')))
  })
}

getNumber().catch(err => console.error(err)) // Error: Test

如果我們在resolve中傳入了一個reject，則我們在外部則可以直接使用catch監聽到。
這種方式經常用於在async函數中拋出異常 如何在async函數中拋出異常：

async function getNumber () {
  return Promise.reject(new Error('Test'))
}
try {
  let number = await getNumber()
} catch (e) {
  console.error(e)
}

一定不要忘了await關鍵字

如果忘記添加await關鍵字，代碼層面並不會報錯，但是我們接收到的返回值卻是一個Promise

let number = getNumber()
console.log(number) // Promise

所以在使用時一定要切記await關鍵字

let number = await getNumber()
console.log(number) // 1

不是所有的地方都需要添加await

在代碼的執行過程中，有時候，並不是所有的非同步都要添加await的。比如下邊的對文件的操作：
我們假設fs所有的API都被我們轉換為了Promise版本

async function writeFile () {
  let fd = await fs.open('test.log')
  fs.write(fd, 'hello')
  fs.write(fd, 'world')
  return fs.close(fd)
}

就像上邊說的，Promise內部的Promise會被消化，所以我們在最後的close也沒有使用await
我們通過await打開一個文件，然後進行兩次文件的寫入。
但是註意了，在兩次文件的寫入操作前邊，我們並沒有添加await關鍵字。
因為這是多餘的，我們只需要通知API，我要往這個文件裡邊寫入一行文本，順序自然會由fs來控制。
然後最後再進行close，因為如果我們上邊在執行寫入的過程還沒有完成時，close的回調是不會觸發的，
也就是說，回調的觸發就意味著上邊兩步的write已經執行完成了。

合併多個不相干的async函數調用

如果我們現在要獲取一個用戶的頭像和用戶的詳細信息（而這是兩個介面 雖說一般情況下不太會出現）

async function getUser () {
  let avatar = await getAvatar()
  let userInfo = await getUserInfo()

  return {
    avatar,
    userInfo
  }
}

這樣的代碼就造成了一個問題，我們獲取用戶信息的介面並不依賴於頭像介面的返回值。
但是這樣的代碼卻會在獲取到頭像以後才會去發送獲取用戶信息的請求。
所以我們對這種代碼可以這樣處理：

async function getUser () {
  let [avatar, userInfo] = await Promise.all([getAvatar(), getUserInfo()])

  return {
    avatar,
    userInfo
  }
}

這樣的修改就會讓getAvatar與getUserInfo內部的代碼同時執行，同時發送兩個請求，在外層通過包一層Promise.all來確保兩者都返回結果。

讓相互沒有依賴關係的非同步函數同時執行

一些迴圈中的註意事項

forEach

當我們調用這樣的代碼時：

async function getUsersInfo () {
  [1, 2, 3].forEach(async uid => {
    console.log(await getUserInfo(uid))
  })
}

function getuserInfo (uid) {
  return new Promise(resolve => {
    setTimeout(_ => resolve(uid), 1000)
  })
}

await getUsersInfo()

這樣的執行好像並沒有什麼問題，我們也會得到1、2、3三條log的輸出，
但是當我們在await getUsersInfo()下邊再添加一條console.log('done')的話，就會發現：
我們會先得到done，然後才是三條uid的log，也就是說，getUsersInfo返回結果時，其實內部Promise並沒有執行完。
這是因為forEach並不會關心回調函數的返回值是什麼，它只是運行回調。

不要在普通的for、while迴圈中使用await

使用普通的for、while迴圈會導致程式變為串列：

for (let uid of [1, 2, 3]) {
  let result = await getUserInfo(uid)
}

這樣的代碼運行，會在拿到uid: 1的數據後才會去請求uid: 2的數據

關於這兩種問題的解決方案：

目前最優的就是將其替換為map結合著Promise.all來實現：

await Promise.all([1, 2, 3].map(async uid => await getUserInfo(uid)))

這樣的代碼實現會同時實例化三個Promise，並請求getUserInfo

P.S. 草案中有一個`await*`，可以省去`Promise.all`

await* [1, 2, 3].map(async uid => await getUserInfo(uid))

P.S. 為什麼在使用`Generator`+`co`時沒有這個問題

在使用koa1.x的時候，我們直接寫yield [].map是不會出現上述所說的串列問題的
看過co源碼的小伙伴應該都明白，裡邊有這麼兩個函數（刪除了其餘不相關的代碼）：

function toPromise(obj) {
  if (Array.isArray(obj)) return arrayToPromise.call(this, obj);
  return obj;
}

function arrayToPromise(obj) {
  return Promise.all(obj.map(toPromise, this));
}

co是幫助我們添加了Promise.all的處理的（膜拜TJ大佬）。

總結

總結一下關於async函數編寫的幾個小提示：

使用return Promise.reject()在async函數中拋出異常
讓相互之間沒有依賴關係的非同步函數同時執行
不要在迴圈的回調中/for、while迴圈中使用await，用map來代替它

參考資料

async-function-tips

如何更好的編寫async函數