快捷導(dǎo)航

理解Koa2中的async&await的用法

更新時(shí)間：2018年02月05日 11:45:50 作者：Z_Xu

這篇文章主要介紹了理解Koa2中的async&await的用法，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧

Koa是一款非常著名的Node服務(wù)端框架，有1.x版本和2.x版本。前者使用了generator來(lái)進(jìn)行異步操作，后者則用了最新的async/await方案

一開始使用這種寫法的時(shí)候，我遇到一個(gè)問(wèn)題，代碼如下：

const Koa = require('koa');
const app = new Koa();

const doSomething = time => {
  return new Promise(resolve => {
    setTimeout(() => {
      resolve('task done!')
    }, time)
  })
}

// 用來(lái)打印請(qǐng)求信息
app.use((ctx, next) => {
  console.log(`${ctx.method}:::${ctx.url}`)
  next()
})

app.use(async ctx => {
  const result = await doSomething(3000)
  console.log(result);
  ctx.body = result
})

app.listen(3000);

讓我們測(cè)試一下：curl http://localhost:3000

期望結(jié)果：

(3秒后...)task done!

然而現(xiàn)實(shí)卻是：

(立即)
Not Found

什么鬼？為什么沒(méi)有按照預(yù)期執(zhí)行？這就需要我們來(lái)理解下Koa中中間件是如何串聯(lián)起來(lái)的了。翻一下源碼，將middlewares串聯(lián)起來(lái)的代碼如下：

function compose (middleware) {
 return function (context, next) {
  // 這個(gè)index用來(lái)計(jì)數(shù)，防止next被多次調(diào)用
  let index = -1
  // 執(zhí)行入口
  return dispatch(0)
  
  function dispatch (i) {
   // 如果next被多次調(diào)用，報(bào)異常
   if (i <= index) return Promise.reject(new Error('next() called multiple times'))
   index = i
   // 取出第一個(gè)middleware
   let fn = middleware[i]
   // 將最初傳入的next作為最后一個(gè)函數(shù)執(zhí)行
   if (i === middleware.length) fn = next
   if (!fn) return Promise.resolve()
   try {
    /**
    這里就是關(guān)鍵了,Promise.resolve是什么意思呢？
     Promise.resolve方法有下面三種形式：
     
     Promise.resolve(value);
     Promise.resolve(promise);
     Promise.resolve(theanable);
     
    這三種形式都會(huì)產(chǎn)生一個(gè)新的Promise。其中：

    第一種形式提供了自定義Promise的值的能力，它與Promise.reject(reason)對(duì)應(yīng)。兩者的不同，在于得到的Promise的狀態(tài)不同。

    第二種形式，提供了創(chuàng)建一個(gè)Promise的副本的能力。

    第三種形式，是將一個(gè)類似Promise的對(duì)象轉(zhuǎn)換成一個(gè)真正的Promise對(duì)象。它的一個(gè)重要作用是將一個(gè)其他實(shí)現(xiàn)的Promise對(duì)象封裝成一個(gè)當(dāng)前實(shí)現(xiàn)的Promise對(duì)象。例如你正在用bluebird，但是現(xiàn)在有一個(gè)Q的Promise，那么你可以通過(guò)此方法把Q的Promise變成一個(gè)bluebird的Promise。第二種形式可以歸在第三種里面
    
    **/
    return Promise.resolve(fn(context, function next () {
     // 執(zhí)行下一個(gè)middleware，返回結(jié)果也是一個(gè)Promise
     return dispatch(i + 1)
    }))
   } catch (err) {
    return Promise.reject(err)
   }
  }
 }
}

有了以上基礎(chǔ)，我們?cè)賮?lái)看一下之前的問(wèn)題，為什么response沒(méi)有等到第二個(gè)middleware執(zhí)行完成就立即返回了呢？

因?yàn)榈谝粋€(gè)middleware并不是一個(gè)異步函數(shù)啊。

由于每次next方法的執(zhí)行，實(shí)際上都是返回了一個(gè)Promise對(duì)象，所以如果我們?cè)谀硞€(gè)middleware中執(zhí)行了異步操作，要想等待其完成，就要在執(zhí)行這個(gè)middleware之前添加await

那我們來(lái)改寫一下之前的代碼

app.use(async (ctx, next) => {
  console.log(`${ctx.method}:::${ctx.url}`)
  await next()
})

app.use(async ctx => {
  const result = await doSomething(3000)
  console.log(result);
  ctx.body = result
})

好了，沒(méi)有問(wèn)題，一切如期望執(zhí)行:clap:

錯(cuò)誤處理

借助了Promise強(qiáng)大的功力，配合async/await語(yǔ)法，我們只需要把try/catch的操作寫在最外層的middleware中，就可以捕獲到之后所有中間件的異常！

app.use(async (ctx, next) => {
  try{
    await next()
  }catch(err){
    console.log(err)
  }
})

app.use(async (ctx)=>{
  throw new Error('something wrong!')
  ctx.body = 'Hello'
})

基于中間件鏈的完全控制，并且基于 Promise 的事實(shí)使得一切都變得容易操作起來(lái)。不再是到處的 if (err) return next(err) 而只有 promise

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: