新聞中心
koa 是一個非常輕量優(yōu)雅的 node 應用開發(fā)框架,趁著雙十一值班的空當閱讀了下其源代碼,其中一些比較有意思的地方整理成文與大家分享一下。
成都創(chuàng)新互聯(lián)公司專注于德化網(wǎng)站建設服務及定制,我們擁有豐富的企業(yè)做網(wǎng)站經(jīng)驗。 熱誠為您提供德化營銷型網(wǎng)站建設,德化網(wǎng)站制作、德化網(wǎng)頁設計、德化網(wǎng)站官網(wǎng)定制、成都微信小程序服務,打造德化網(wǎng)絡公司原創(chuàng)品牌,更為您提供德化網(wǎng)站排名全網(wǎng)營銷落地服務。
洋蔥型中間件機制的實現(xiàn)原理
我們經(jīng)常把 koa 中間件的執(zhí)行機制類比于剝洋蔥,這樣設計其執(zhí)行順序的好處是我們不再需要手動去管理 request 和 response 的業(yè)務執(zhí)行流程,且一個中間件對于 request 和 response 的不同邏輯能夠放在同一個函數(shù)中,可以幫助我們極大的簡化代碼。在了解其實現(xiàn)原理之前,先來介紹一下 koa 的整體代碼結(jié)構(gòu):
lib |-- application.js |-- context.js |-- request.js |-- response.js
application 是整個應用的入口,提供 koa constructor 以及實例方法屬性的定義。context 封裝了koa ctx 對象的原型對象,同時提供了對 response 和 request 對象下許多屬性方法的代理訪問,request.js 和 response.js 分別定義了ctx request 和 response 屬性的原型對象。
接下來讓我們來看 application.js中的一段代碼:
listen(...args) { debug('listen'); const server = http.createServer(this.callback()); return server.listen(...args); } callback() { const fn = compose(this.middleware); if (!this.listenerCount('error')) this.on('error', this.onerror); const handleRequest = (req, res) => { const ctx = this.createContext(req, res); return this.handleRequest(ctx, fn); }; return handleRequest; } handleRequest(ctx, fnMiddleware) { const res = ctx.res; res.statusCode = 404; const onerror = err => ctx.onerror(err); const handleResponse = () => respond(ctx); onFinished(res, onerror); return fnMiddleware(ctx).then(handleResponse).catch(onerror); }
上述代碼展示了 koa 的基本原理,在其實例方法 listen 中對 http.createServer 進行了封裝 ,然后在回調(diào)函數(shù)中執(zhí)行 koa 的中間件,在 callback 中,this.middleware 為業(yè)務定義的中間件函數(shù)所構(gòu)成的數(shù)組,compose 為 koa-compose 模塊提供的方法,它對中間件進行了整合,是構(gòu)建 koa 洋蔥型中間件模型的奧妙所在。從 handleRequest 方法中可以看出 compose 方法執(zhí)行返回的是一個函數(shù),且該函數(shù)的執(zhí)行結(jié)果是一個 promise。接下來我們就來一探究竟,看看 koa-compose 是如何做到這些的,其 源代碼和一段 koa 中間件應用示例代碼如下所示:
// compose源碼 function compose (middleware) { if (!Array.isArray(middleware)) throw new TypeError('Middleware stack must be an array!') for (const fn of middleware) { if (typeof fn !== 'function') throw new TypeError('Middleware must be composed of functions!') } return function (context, next) { // last called middleware # let index = -1 return dispatch(0) function dispatch (i) { if (i <= index) return Promise.reject(new Error('next() called multiple times')) index = i let fn = middleware[i] if (i === middleware.length) fn = next if (!fn) return Promise.resolve() try { return Promise.resolve(fn(context, dispatch.bind(null, i + 1))); } catch (err) { return Promise.reject(err) } } } } /* ** 中間件應用示例代碼 */ let Koa = require('koa') let app = new Koa() app.use(async function ware0 (ctx, next) { await setTimeout(function () { console.log('ware0 request') }, 0) next() console.log('ware0 response') }) app.use(function ware1 (ctx, next) { console.log('ware1 request') next() console.log('ware1 response') }) // 執(zhí)行結(jié)果 ware0 request ware1 request ware1 response ware0 response
從上述 compose 的源碼可以看出,每個中間件所接受的 next 函數(shù)入?yún)⒍际窃?compose 返回函數(shù)中定義的 dispatch 函數(shù),dispatch接受下一個中間件在 middlewares 數(shù)組中的索引作為入?yún)?,該索引就像一個游標一樣,每當 next 函數(shù)執(zhí)行后,游標向后移一位,以獲取 middlaware 數(shù)組中的下一個中間件函數(shù) 進行執(zhí)行,直到數(shù)組中最后一個中間件也就是使用 app.use 方法添加的最后一個中間件執(zhí)行完畢之后再依次 回溯執(zhí)行。整個流程實際上就是函數(shù)的調(diào)用棧,next 函數(shù)的執(zhí)行就是下一個中間件的執(zhí)行,只是 koa 在函數(shù)基礎上加了一層 promise 封裝以便在中間件執(zhí)行過程中能夠?qū)⒉东@到的異常進行統(tǒng)一處理。 以上述編寫的應用示例代碼作為例子畫出函數(shù)執(zhí)行調(diào)用棧示意圖如下:
整個 compose 方法的實現(xiàn)非常簡潔,核心代碼僅僅 17 行而已,還是非常值得圍觀學習的。
generator函數(shù)類型中間件的執(zhí)行
v1 版本的 koa 其中間件主流支持的是 generator 函數(shù),在 v2 之后改而支持 async/await 模式,如果依舊使用 generator,koa 會給出一個 deprecated 提示,但是為了向后兼容,目前 generator 函數(shù)類型的中間件依然能夠執(zhí)行,koa 內(nèi)部利用 koa-convert 模塊對 generator 函數(shù)進行了一層包裝,請看代碼:
function convert (mw) { // mw為generator中間件 if (typeof mw !== 'function') { throw new TypeError('middleware must be a function') } if (mw.constructor.name !== 'GeneratorFunction') { // assume it's Promise-based middleware return mw } const converted = function (ctx, next) { return co.call(ctx, mw.call(ctx, createGenerator(next))) } converted._name = mw._name || mw.name return converted } function * createGenerator (next) { return yield next() }
從上面代碼可以看出,koa-convert 在 generator 外部包裹了一個函數(shù)來提供與其他中間件一致的接口,內(nèi)部利用 co 模塊來執(zhí)行 generator 函數(shù),這里我想聊的就是 co 模塊的原理,generator 函數(shù)執(zhí)行時并不會立即執(zhí)行其內(nèi)部邏輯,而是返回一個遍歷器對象,然后通過調(diào)用該遍歷器對象的 next 方法來執(zhí)行,generator 函數(shù)本質(zhì)來說是一個狀態(tài)機,如果內(nèi)部有多個 yield 表達式,就需要 next 方法執(zhí)行多次才能完成函數(shù)體的執(zhí)行,而 co 模塊的能力就是實現(xiàn) generator 函數(shù)的 自動執(zhí)行,不需要手動多次調(diào)用 next 方法,那么它是如何做到的呢?co 源碼如下:
function co(gen) { var ctx = this; var args = slice.call(arguments, 1); // we wrap everything in a promise to avoid promise chaining, // which leads to memory leak errors. // see https://github.com/tj/co/issues/180 return new Promise(function(resolve, reject) { if (typeof gen === "function") gen = gen.apply(ctx, args); if (!gen || typeof gen.next !== "function") return resolve(gen); onFulfilled(); /** * @param {Mixed} res * @return {Promise} * @api private */ function onFulfilled(res) { var ret; try { ret = gen.next(res); } catch (e) { return reject(e); } next(ret); } /** * @param {Error} err * @return {Promise} * @api private */ function onRejected(err) { var ret; try { ret = gen.throw(err); } catch (e) { return reject(e); } next(ret); } /** * Get the next value in the generator, * return a promise. * * @param {Object} ret * @return {Promise} * @api private */ function next(ret) { if (ret.done) return resolve(ret.value); // toPromise是一個函數(shù),返回一個promise示例 var value = toPromise.call(ctx, ret.value); if (value && isPromise(value)) return value.then(onFulfilled, onRejected); return onRejected( new TypeError( "You may only yield a function, promise, generator, array, or object, " + 'but the following object was passed: "' + String(ret.value) + '"' ) ); } }); }
從 co 源碼來看,它先是手動執(zhí)行了一次onFulfilled 函數(shù)來觸發(fā) generator 遍歷器對象的 next 方法,然后利用promise的onFulfilled 函數(shù)去自動完成剩余狀態(tài)機的執(zhí)行,在onRejected 中利用遍歷器對象的 throw 方法拋出執(zhí)行上一次 yield 過程中遇到的異常,整個實現(xiàn)過程可以說是相當簡潔優(yōu)雅。
結(jié)語
通過上面的例子可以看出 promise 的能量是非常強大的,koa 的中間件實現(xiàn)和 co 模塊的實現(xiàn)都是基于 promise,除了應用于日常的異步流程控制,在開發(fā)過程中我們還可以大大挖掘其潛力,幫助我們完成一些自動化程序工作流的事情。
以上就是本文的全部內(nèi)容,希望對大家的學習有所幫助,也希望大家多多支持創(chuàng)新互聯(lián)。
分享名稱:koa源碼中promise的解讀
標題路徑:http://fisionsoft.com.cn/article/joogod.html