为了更好的理解和实践 promise,我尝试自己写一个实现 promise 所有功能的类,并基于此去做一些扩展,达到可以在生产环境使用的程度;并且为了便于维护和理解,代码全部使用 typescript 编写。
01. 目录
- 02.自下而上
- 03.如何实现
- 04.Promise/A+规范
- 05.更多优化
- 06.源码
- 07.小结
- 08.其他参考
02.自下而上
02.01 基本概念
- 首先我们来整理一些
Promise
基本的概念,包括私有状态,内部方法,静态方法等等。
私有属性
- 私有属性包括状态和值
PromisState
PromiseResult
,这些属性外部无法访问。 - 状态属性有以下三种:
-
pending
初始化状态 -
fulfilled
兑现(完成) -
rejected
拒绝
-
- 值属性,由
resolve
或reject
处理来决定。
实例方法
then
catch
finally
静态方法
Promise.reject
Promise.resolve
Promise.race
Promise.all
Promise.allSettled
Promise.any
03.如何实现
03.01 基础类
-
在罗列所有的状态和方法之后,我们首先来实现一个最基础的
Promise
类。 -
最基础的类,包括以下核心几点:
- 拥有私有状态,也有着能够改变私有状态的私有方法。
- 同时接收一个执行器函数作为参数,执行器函数内部则是预先定义好的私有方法。
- 私有状态一旦改变(兑现或拒绝)后不可逆。
/** * Promise 内部状态的枚举 */ enum PROMISE_STATES { PENDING = ‘pending‘, FULFILLED = ‘fulfilled‘, REJECTED = ‘rejected‘ } type PromiseStates = PROMISE_STATES.PENDING | PROMISE_STATES.FULFILLED | PROMISE_STATES.REJECTED; export const isFunction = (fn: any):boolean => typeof fn === ‘function‘; export const isObject = (obj: any):boolean => typeof obj === ‘object‘; class PromiseLike { protected PromiseState: PromiseStates; protected PromiseResult: any; constructor(executor) { this.PromiseState = PROMISE_STATES.PENDING; this.PromiseResult = undefined; executor(this._resolve, this._reject) } _resolve = (value?: any) => { if (this.PromiseState !== PROMISE_STATES.PENDING) { return; } this.PromiseState = PROMISE_STATES.FULFILLED; this.PromiseResult = value; } _reject = (value?: any) => { if (this.PromiseState !== PROMISE_STATES.PENDING) { return; } this.PromiseState = PROMISE_STATES.REJECTED; this.PromiseResult = value; } } 复制代码
resolve 和 reject
-
上述代码比较好理解, 我们定义了状态,定义了执行器函数以及相关的两个参数,这两个参数对应的方法分别修改了对应的状态。
-
但是差点忘了,
Promise
是异步的,意味着这两个函数处理也应当是异步的;这里可以使用setTimeout
来模拟异步进程。这部分还可以优化,后面我们会提到。class PromiseLike { /** * 使状态变更为 fulfilled * 调用注册的事件,注意调用后进行清除 * @param value * @returns */ _resolve = (value?: any) => { const resolveCb = () => { if (this.PromiseState !== PROMISE_STATES.PENDING) { return; } this.PromiseState = FULFILLED; this.PromiseResult = value; } // 使任务变成异步的 setTimeout(resolveCb, 0); } /** * 使状态变更为 rejected * @param value */ _reject = (value?: any) => { const rejectCb = () => { if (this.PromiseState !== PROMISE_STATES.PENDING) { return; } this.PromiseState = REJECTED; this.PromiseResult = value; } setTimeout(rejectCb, 0); } } 复制代码
-
我们可以接着实现相关的静态方法,因为它们所做的事很简单,就是修改当前的内部状态,于是完全可以直接调用当前类实例化来处理。
-
重复代码不再罗列,下面是新增的静态方法:
class PromiseLike { // ...sth static resolve(value?: any) { return new PromiseLike((resolve) => resolve(value)); } static reject(value?: any) { return new PromiseLike((resolve, reject) => reject(value)); } } 复制代码
-
一个简单的基础类就这样完成了。不过先不要着急,当前的实现显然有许多要完善的地方,甚至也许有错误,让我们进一步来梳理。
03.02 原型方法
Promise.prototype.then
-
相信对
Promise
有所了解的都知道Promise
的then
方法以及它的链式调用。本质上,**它是对Thenable
接口的具体实现。**这句话很重要,后面会用到。 -
让我们先来回顾一下
then
的用法:Promise.resolve(29).then(function fulfilled(res) { console.info(res); return res; }, function rejected(err) { console.error(err); }); 复制代码
-
then
方法接收两个参数,分别用来处理resolve
和reject
的结果,称之为完成回调和拒绝回调。默认情况下,同时注册这两个回调方法,一次只可能会调用到其中一个。即使在前一个函数中抛出了异常,第二个异常捕获函数也无法立即捕获。- 完成回调,接收先前
promise
的resolve
值作为默认参数,处理对应数据,并返回一个值,作为下一个then
内部函数调用的默认参数。 - 让我们再仔细想想,
then
注册事件的调用次数是否和注册次数相同?是的。假如使用then
注册了多个回调函数,则它们会依次执行。这意味着我们得在原先的基础上加上相应的事件队列。 - 另外别忘了,
then
方法支持链式调用,我们这里先使用return this
的方式来简单实现。
- 完成回调,接收先前
-
现在我们对上面的基础类进行改进和修复。
- 定义两个数组,分别用来保存完成回调和拒绝回调。
- 下面罗列核心代码:
export interface ICallbackFn { (value?: any): any; } type CallbackParams = ICallbackFn | null; export interface IExecutorFn { (resolve: ICallbackFn, reject: ICallbackFn): any; } class PromiseLike { protected PromiseState: PromiseStates; protected PromiseResult: any; resolveCallbackQueues: Array<ICallbackFn>; rejectCallbackQueues: Array<ICallbackFn>; constructor(executor: IExecutorFn) { if (!isFunction(executor)) { throw new Error(‘Promise resolver undefined is not a function‘); } this.PromiseState = PENDING; this.PromiseResult = undefined; // 分别用于两个注册事件保存的数组 this.resolveCallbackQueues = []; this.rejectCallbackQueues = []; executor(this._resolve, this._reject); } /** * 使状态变更为 fulfilled * 调用注册的事件,注意调用后进行清除 * @param value * @returns */ _resolve = (value: any) => { const resolveCb = () => { if (this.PromiseState !== PROMISE_STATES.PENDING) { return; } while (this.resolveCallbackQueues.length) { const fn = this.resolveCallbackQueues.shift(); fn && fn(value); } this.PromiseState = FULFILLED; this.PromiseResult = value; } // 使任务变成异步的 setTimeout(resolveCb, 0); } /** * 使状态变更为 rejected * @param value */ _reject = (value: any) => { const rejectCb = () => { if (this.PromiseState !== PROMISE_STATES.PENDING) { return; } while (this.rejectCallbackQueues.length) { const fn = this.rejectCallbackQueues.shift(); fn && fn(value); } this.PromiseState = REJECTED; this.PromiseResult = value; } setTimeout(rejectCb, 0); } /** * 根据当前不同状态来执行对应逻辑 * 如果在默认状态就是注册对应事件 * 如果状态变化则是执行对应事件 * @param onFulfilled * @param onRejected * @returns */ then = (onFulfilled, onRejected) => { switch (this.PromiseState) { case PENDING: isFunction(onFulfilled) && this.resolveCallbackQueues.push(onFulfilled); isFunction(onRejected) && this.rejectCallbackQueues.push(onRejected); case FULFILLED: isFunction(onFulfilled) && onFulfilled(this.PromiseResult); break; case REJECTED: isFunction(onRejected) && onRejected(this.PromiseResult); break; } return this; } } 复制代码
-
我们丰富了
then
方法。但是你我都知道,return this
看起来并不太可靠。 -
让我们来回顾一点,
Promise
的私有状态一旦改变后不可逆。如果在这个then
方法里抛出异常,promise
显然会变成拒绝状态,而同一实例的状态在改变后是不能够再次修改的。所以,then
的链式调用本质上是每次都会生成一个新的实例。 -
也许再贴一个使用
then
的例子会让我们有一些启发。const p = Promise.resolve(123); const p1 = p.then(); const p2 = p1.then((val) => val + 123)) const p3 = p2.then(console.info)); const p4 = p3.then(() => { throw new Error(‘Oops!‘); }); // 分别打印 p1 p2 p3 p4 // Promise {<fulfilled>: 123} // Promise {<fulfilled>: 246} // Promise {<fulfilled>: undefined} // Promise {<rejected>: Error: Oops! 复制代码
-
这段代码的输出,有助于让我们进一步理解
then
内部所做的事。- p1: 在没有传入回调函数的时候,它仅仅是将值传递,也就是内部会初始化一个默认的处理函数,这个处理函数只会乖乖地传递值。
- p2: 存在完成回调时,可以获取值并进行处理,这个新的值通过返回的形式继续往后传递。
- p3: 如果传入完成回调函数,但没有显式返回值,则最终的
promise
的值是undefined
. - p4:
promise
状态已经变更成rejected
, 意味着是新的promise
. 符合我们的预期。
-
带着上述理解,我们来改进
then
方法。 -
首先,需要处理参数异常的情况,也就是传入参数不是函数,或者未传的情况,就给定默认处理函数。
- 完成回调负责传递参数。
- 拒绝回调负责抛出异常。
then = (onFulfilled?: CallbackParams, onRejected?: CallbackParams) => { // 默认处理!!! onFulfilled = isFunction(onFulfilled) ? onFulfilled : value => value; onRejected = isFunction(onRejected) ? onRejected : err => { throw err }; } 复制代码
-
我们把这两个兼容处理放在函数内的顶部,这样有助于理解,也可以简化后续的逻辑。
-
下面是具体的内容,其中核心改动已注释说明。
class PromiseLike { /** * 根据当前不同状态来执行对应逻辑 * 如果在默认状态就是注册对应事件 * 如果状态变化则是执行对应事件 * @param onFulfilled * @param onRejected * @returns */ then = (onFulfilled: CallbackParams, onRejected: CallbackParams) => { // 默认处理!!! onFulfilled = isFunction(onFulfilled) ? onFulfilled : value => value; onRejected = isFunction(onRejected) ? onRejected : err => { throw err }; return new PromiseLike((resolve, reject) => { /** * 封装完成回调函数 * @param val */ const handleFulfilled = (val: any) => { try { const res = onFulfilled(val); resolve(res); } catch (error) { // 如果当前执行逻辑内发生异常,则抛出异常 reject(error); } }; /** * 封装错误回调函数 * @param val */ const handleRejected = (val: any) => { try { const res = onRejected(val); reject(res); } catch (error) { reject(error); } } switch (this.PromiseState) { case PROMISE_STATES.PENDING: this.resolveCallbackQueues.push(handleFulfilled); this.rejectCallbackQueues.push(handleRejected); break; case PROMISE_STATES.FULFILLED: handleFulfilled(this.PromiseResult); break; case PROMISE_STATES.REJECTED: handleRejected(this.PromiseResult); break; } }); } } 复制代码
-
这个
then
方法的处理已经接近完善,不过在Promise
里有一点容易被人遗忘。- 在
Promise
中处理Promise
,内部处理会将其展开来获取其中的值。 - 下面这个例子出来你就理解了。
Promise.resolve(41) === Promise.resolve(Promise.resolve(41)); // false 复制代码
- 在
-
不好意思,走错片场。js 中每个单独定义的引用类型都是不相等的。
- 再来一次。
const p = Promise.resolve(41); Promise.resolve(p) === p; // true 复制代码
-
没错,如果我们给
promise
一个promise
值,内部机制会将其展开。这个过程是递归的,这里我们先不展开探讨,但记住有这样的场景需要处理。 -
可以先定义一个静态方法判断是否是
Promise
实例,方便后续的判断。class PromiseLike { /** * 判断是否是当前类的实例 * @param promise * @returns */ static is(promise: PromiseType) { return promise instanceof PromiseLike; } } 复制代码
-
有了这个方法,我们可以进一步完善上面的
then
方法。注意观察其中的变化,有注释说明。 -
为方便阅读,只展示核心方法(只有这里改动)。
/** * 封装完成回调函数 * @param val */ const handleFulfilled = (val) => { try { const res = onFulfilled(val); if (PromiseLike.is(res)) { // 如果参数是 Promise 实例,直接可以把 promise 实例进行传递 res.then(resolve, reject); } else { resolve(res); } } catch (error) { // 如果当前执行逻辑内发生异常,则抛出异常 reject(error); } }; 复制代码
Promise.prototype.catch
-
在实现
then
方法之后,其实catch
的实现是你想象不到的简单。 -
因为本质上
catch
方法是then
第二个参数也就是错误回调函数的语法糖。照着这个理解,实现起来就比较容易。class PromiseLike { /** * 错误处理 * https://developer.mozilla.org/zh-CN/docs/Web/JavaScript/Reference/Global_Objects/Promise/catch * @param rejectedCb * @returns */ catch = (rejectedCb: CallbackParams) => { return this.then(null, rejectedCb); } } 复制代码
Promise.prototype.finally
-
实现
finally
需要我们理解几个点。- 前面的状态只要不是
pending
, 则一定会进入执行。 - 类似于
then
, 它可以注册多个回调,每个回调函数会依次执行。 - 回调函数内无法获取内部值。
- 除非在回调函数内抛出异常会把状态变成
rejected
,否则它所做的仅仅是把状态和值传递。
- 前面的状态只要不是
-
了解上述几点之后,我们可以复用
then
方法,并自定义回调函数传入来实现。class PromiseLike { /** * https://developer.mozilla.org/zh-CN/docs/Web/JavaScript/Reference/Global_Objects/Promise/finally * @param finallyCb * @returns */ finally = (finallyCb: CallbackParams) => { return this.then( // 完成回调时,执行注册函数,并且将原来的值传递下去 // 封装 Promise 类,再调用 then 方法传递 val => PromiseLike.resolve(finallyCb && finallyCb()).then(() => val), // 异常回调时,执行注册函数,并且抛出异常 err => PromiseLike.resolve(finallyCb && finallyCb()).then(() => { throw err }) ); } } 复制代码
-
写到这里,几个核心的原型方法我们就实现完毕了。
-
心急的伙伴可以直接实例化一个对象来尝试,不过
Promise
当然还不止于此,接下来我们来实现对应的静态方法。
03.03 静态方法
- 前面已经实现了一个自定义的
Promise.is
方法来判断实例。这个工具类函数简单实用,可以留着。 - 还有两个快速实例化
Promise
类的方法我们也进行了实现:Promise.resolve
和Promise.reject
. 下面来做一点改进。
Promise.resolve
-
既然我们定义好了
Promise.is
方法,加上对Promise
的理解进一步加深,知道了如果传入的已经是Promise
实例,则不必再进行处理。所以这个方法需要做一点兼容处理。class PromiseLike { /** * 直接实例化 proimse * https://developer.mozilla.org/zh-CN/docs/Web/JavaScript/Reference/Global_Objects/Promise/resolve * @param value * @returns */ static resolve(value?: any) { if (PromiseLike.is(value)) { return value; } return new PromiseLike((resolve) => resolve(value)); } } 复制代码
-
现在我们可以尝试实现
Promise
提供的剩下两个类方法Promise.all
,Promise.race
.
Promise.all
-
该方法是接收一个由
Promise
实例组成的数组,并返回Promise
实例,其值是所有Promise
实例的resolve
的值组成的数组。- 当其中任意一个
Promise
有reject
的值时,Promise.all
会返回最先rejected
的值。 - 等到所有
Promise
resolve
之后,Promise
.all 才会返回结果。 -
Promise.all
也是支持链式调用的。
- 当其中任意一个
-
大白话也许有些晦涩,我们直接看案例。
Promise.all([Promise.resolve(1), Promise.reject(2)]); // Promise {<rejected>: 2} Promise.all([Promise.resolve(1), Promise.resolve(2)]); // Promise {<fulfilled>: Array(2)} [1, 2] Promise.all([]); // Promise {<fulfilled>: Array(2)} 复制代码
-
其中,第三个表达式的结果对理解
Promise.race
和Promise.all
的区别很重要。这点后面会谈。除此之外,结果是显而易见的。 -
Promise.all
返回的结果是传入数组的参数的顺序,也可以理解为顺序执行,并填入对应的位置。基于这几点,要实现它就有思路了。- 顺序执行所有
Promise
,并把结果保存到数组的对应位置,同时统计已执行的数量;当该数量等同于传入的数组长度时,返回由结果组成的数组。
class PromiseLike { /** * https://developer.mozilla.org/zh-CN/docs/Web/JavaScript/Reference/Global_Objects/Promise/all * @param promises 严格意义上来说,参数是可迭代对象,为了简化实现这里统一成数组 * @returns */ static all(promises: Array<ICallbackFn>) { // 支持链式调用 return new PromiseLike((resolve, reject) => { const len = promises.length; let resolvedPromisesCount = 0; let resolvedPromisesResult = <any>[]; for (let i = 0; i < len; i++) { const currentPromise = promises[i]; // 如果不是 Promise 实例,则需要包装一份; // 但因为直接包装 Promise 类的效果是幂等的,所以这里不需要判断,直接处理即可 PromiseLike.resolve(currentPromise) .then((res: any) => { resolvedPromisesCount++; resolvedPromisesResult[i] = res; // 当所有值都 resolve 之后, 返回对应数组 if (resolvedPromisesCount === len) { resolve(resolvedPromisesResult); } }) // 如果有任意一个异常,则直接推出 .catch((err: any) => { reject(err); }); } }); } } 复制代码
- 顺序执行所有
-
如同在方法注释里说明的一样,其实
Promise.all
和Promise.race
方法接收的参数都是可迭代对象,并不仅仅是数组。这里为了方便实现,使用数组替代。可迭代对象不在这篇文章的核心讨论范围之内,感兴趣的可以点进上面的链接继续了解。
Promise.race
-
Promise.race
和Promise.all
有些相似,至少就参数而言,都接收可迭代对象作为参数,也可以链式调用,意味着它也返回一个新的Promise
实例。 -
不同的是,
Promise.race
将会返回第一个Promise.resolve
的值,或是第一个 reject 的值,而且这个值并不是数组。 -
了解到这两点之后,实现起来就有清晰的思路了。
- 遍历顺序执行所有 Promise 并取出第一个 resolve 的值。
class PromiseLike { /** * https://developer.mozilla.org/zh-CN/docs/Web/JavaScript/Reference/Global_Objects/Promise/race * @param promises * @returns */ static race(promises: Array<ICallbackFn>) { return new PromiseLike((resolve, reject) => { for (let i = 0; i < promises.length; i++) { const currentPromise = promises[i]; PromiseLike.resolve(currentPromise) .then((res: any) => { resolve(res); }) .catch((err: any) => { reject(err); }); } }); } } 复制代码
-
再运行这样一段代码,得到的结果应该并不会让你意外。
Promise.race([]); // Promise {<pending>} 与 Promise.all 的结果不同 复制代码
-
至此,目前已广泛兼容的两个核心方法我们都已经实现了。这是不是意味着可以愉快的玩耍了呢,当然可以。不过,既然都走到这一步了,我们顺带可以实现更多的
Promise
方法,一来锻炼动手能力,二来证明学以致用。
03.04 其他静态方法
Promise.allSettled
-
这个方法和
Promise.all
非常相似,执行所有的Promise
实例并返回所有的结果,不论结果如何,都在返回的数组里塞回一个对象。- 每个对象只有两个属性
status
和value
或reason
;如果当前proimse
是fulfilled
则属性是status
和value
, 如果当前是rejected
则属性是status
和reason
.
- 每个对象只有两个属性
-
对
Promise.all
稍加改动就可以实现。- 判断计数的逻辑在两个回调函数中都进行,并且对返回值加一层包装。
/** * https://developer.mozilla.org/zh-CN/docs/Web/JavaScript/Reference/Global_Objects/Promise/allSettled * @param promises 严格意义上来说,参数是可迭代对象,为了简化实现这里统一成数组 * @returns */ static allSettled(promises: Array<IPromiseType>) { // 支持链式调用 return new PromiseLike((resolve, reject) => { const len = promises.length; const startTime = Date.now(); let resolvedPromisesCount = 0; let resolvedPromisesResult = <any>[]; for (let i = 0; i < len; i++) { const currentPromise = promises[i]; // 如果不是 Promise 实例,则需要包装一份; // 但因为直接包装 Promise 类的效果是幂等的,所以这里不需要判断,直接处理即可 PromiseLike.resolve(currentPromise) .then((res: any) => { resolvedPromisesCount++; resolvedPromisesResult[i] = { status: PROMISE_STATES.FULFILLED, value: res }; // 当所有 promises 完成后,返回数组;多封装了一个属性用于显示执行时间 if (resolvedPromisesCount === len) { resolvedPromisesResult.duringTime = Date.now() - startTime + ‘ms‘; resolve(resolvedPromisesResult); } }) // 如果有任意一个异常,则直接推出 .catch((err: any) => { resolvedPromisesCount++; resolvedPromisesResult[i] = { status: PROMISE_STATES.REJECTED, reason: err }; if (resolvedPromisesCount === len) { resolvedPromisesResult.duringTime = Date.now() - startTime + ‘ms‘; resolve(resolvedPromisesResult); } }); } }); } 复制代码
Promise.any
-
这是今年(2021)刚刚落定草案的新 API。定义和
Promise.race
很相似,接收可迭代对象作为参数,可以链式调用。 -
不同的是,它会返回第一个落定的,也就是
resolve
的值;如果传入的promise
全都都进入拒绝状态,则它会等到所有拒绝状态都完成后,再返回一个由拒绝错误组成的对象。这个对象是新定义的类型 AggregateError,这里暂且先不展开,直接使用它。 -
从定义上来看,它和
Promise.race
相似,不过从实现上观察,却和Promise.all
更加相似。- 只需要把计算数量的逻辑搬到错误回调中,并将其返回错误对象即可。
class PromiseLike { /** * 2021 年刚纳入规范的 any * https://developer.mozilla.org/zh-CN/docs/Web/JavaScript/Reference/Global_Objects/Promise/any * @param promises * @returns */ static any(promises: Array<ICallbackFn>) { return new PromiseLike((resolve, reject) => { const len = promises.length; let rejectedPromisesCount = 0; let rejectedPromisesResult = <any>[]; for (let i = 0; i < promises.length; i++) { const currentPromise = promises[i]; PromiseLike.resolve(currentPromise) .then((res: any) => { resolve(res); }) .catch((err: any) => { rejectedPromisesCount++; rejectedPromisesResult[i] = err; if (rejectedPromisesCount === len) { // 如果浏览器支持,则直接抛出这个新对象,否则则直接抛出异常 if (isFunction(AggregateError)) { throw new AggregateError(rejectedPromisesResult, ‘All promises were rejected‘); } else { throw (rejectedPromisesResult); } } }); } }) } } 复制代码
04.Promise/A+规范
04.01 promises-aplus-tests 验证
-
这个库 promises-aplus-tests 可以用来验证我们实现的
Promise
是否遵循 Promise/A+规范 。 -
使用方式比较简单,注入一个方法即可,这个方法返回的对象包含
Promise/resolve/reject
. -
由于我们使用类的方式编写,所以直接新增一个静态函数即可。
class PromiseLike { /** * 三方库验证 * @returns */ static deferred() { let defer: any = {}; defer.promise = new PromiseLike((resolve, reject) => { defer.resolve = resolve; defer.reject = reject; }); return defer; } } 复制代码
-
需要注意的是,要用
commonjs
规范的方式来导出,否则会出现报错。module.exports = PromiseLike; 复制代码
-
运行
npx promises-aplus-tests 目录名
进行验证。
04.02 并不完美(兼容修复)
- 运行结果显示,有部分 case 没有通过。糟透了!下面一一提取。
‘Chaining cycle detected for promise‘
-
这个异常显示,我们不能在
promise
中使用自身,否则会造成死循环。 -
举个例子:
const p = Promise.resolve(1).then(() => p); 复制代码
-
运行这段代码,就会得到上述报错。
-
解决办法并不难,定义变量来保存
then
函数的返回值,同时在内部方法返回的位置进行兼容处理,如果相等就抛出异常。const res = onFulfilled(val); // 返回的 promise 不可以是当前的 promise 否则会造成死循环 if (newPromise === res) { throw new TypeError(‘Chaining cycle detected for promise #<Promise>‘); } 复制代码
2.3.3: Otherwise, if
x is an object or function
-
再次执行,发现
Promise
规范对传入参数是对象和函数类型也有着特殊的处理。我们并没有处理,所以出现了上述报错。规范里 有所定义,我们可以简单理解为如果传入的参数是Thenable
的,则需要调用其中的then
方法,也就是将其展开调用。上文中自己有提到,终究是逃不过。 -
之前我们仅仅对
Promise
的实例进行了特殊处理,现在意识到还需要处理Thenable
接口的对象。但因为Promise
实例本身就是实现Thenable
接口的特殊对象。(typeof Promise.resolve(1); // object
),所以实现了对Thenable
接口的处理,自然也能涵盖原有的逻辑。 -
新定义一个单独的方法来实现,以提高可读性。
- 这个函数有些复杂,但每一条逻辑都可以在规范里追溯。
/** * 该实现遵循 Promise/A+ 规范 * https://github.com/promises-aplus/promises-spec * @param promise * @param x * @param resolve * @param reject * @returns */ const resolvePromise = (promise: any, x: any, resolve: ICallbackFn, reject: ICallbackFn) => { // 返回的 promise 不可以是当前的 promise 否则会造成死循环 if (newPromise === x) { reject(new TypeError(‘Chaining cycle detected for promise #<Promise>‘)); } // 对可能是 thenable 接口实现的对象判断 if (isObject(x) || isFunction(x)) { if (x === null) { return resolve(x); } let thenCb; try { thenCb = x.then; } catch (error) { return reject(error); } // 如果是 thenable 的对象,则调用其 then 方法 // 这一步涵盖了 Promise 实例的可能性 if (isFunction(thenCb)) { let isCalled = false; try { thenCb.call( x, // 指向当前函数或对象 (y: any) => { // 如果 resolvePromise 和 rejectPromise 都可能被调用 // 则只需调用第一次(resolvePromise 或 rejectPromise),后续无需再执行 if (isCalled) return; isCalled = true; // 传入当前函数,以实现递归展开调用 resolvePromise(promise, y, resolve, reject); }, (r: any) => { // 对应前面任意的调用之后,就不再只需后续逻辑 if (isCalled) return; isCalled = true; reject(r); } ) } catch (error) { if (isCalled) return; reject(error); } } else { resolve(x); } } else { resolve(x); } } 复制代码
-
在原先处理数据的地方,换成
resolvePromise
函数就可以了。 -
这下是可算是完整通过测试了。
872 passing (14s) 复制代码
05.更多优化
05.01 queueMicrosoft
- 学习过程中,意外发现
queueMicrosoft
这个方法,用于将任务转换成微任务。我们知道setTimeout
虽然可以实现异步的效果,但它属于宏任务,与Promise
所属的微任务不符。所以可以用queueMicrosoft
来替换。 - 有关使用方式,可以查看这里
05.02 typescript 完善
-
前面的例子里已经定义许多接口。这里举个例子完善一哈,更多详细内容可以查看下文的源码。
export interface IPromiseType { then: IExecutorFn; catch: ICallbackFn; finally: ICallbackFn; } class PromiseLike implements IPromiseType {} 复制代码
-
由于自己的 typescript 实践仍在学习中,可能源码中还存在许多值得改进和优化的地方,可以在评论或 issue 中指出,合理的改进建议一定会采纳并实践。
-
使用版本:
"typescript": "^4.3.5"
05.03 花里胡哨的变种方法
Promise.last
-
定义一个函数,返回最后一个完成的
promise
, 并且可以选择是否需要rejected
的promise
./** * 返回最后一个完成的值,可以自行决定是否忽略异常 * 如果不忽略,异常优先抛出 * 如果忽略,返回完成值 * @param promises * @param ignoreRejected * @returns */ static last(promises: Array<IPromiseType>, ignoreRejected: boolean = false) { return new PromiseLike((resolve, reject) => { const len = promises.length; const startTime = Date.now(); let resolvedPromisesCount = 0; for (let i = 0; i < len; i++) { const currentPromise = promises[i]; PromiseLike.resolve(currentPromise) .then((res: any) => { resolvedPromisesCount++; // 当所有 promises 完成后,返回最后一个值;封装一个属性用于显示执行时间 if (resolvedPromisesCount === len) { isObject(res) && (res.duringTime = Date.now() - startTime + ‘ms‘); resolve(res); } }) // 如果有任意一个异常,则直接推出 .catch((err: any) => { if (ignoreRejected) { resolvedPromisesCount++; } else { reject(err) } }); } }); } 复制代码
-
还可以实现一个它的变种,即返回最后一个更新的值,不论是
fulfilled
或者rejected
状态.源码有展示,这里不再赘述。
Promise.wrap
-
该方法可以将原来的普通异步请求包装成
Promise
实例,便于链式调用等。 -
假设有这样一个请求处理函数。
function fn(url, cb) { ajax(url, cb); } 复制代码
-
想让它变成可以使用链式调用,使用方式见注释。
/** * 把不是 promise 实例的函数包装成 promise 实例 * 例如 ajax 请求 * const request = Promise.wrap(ajax); * request.then(callback); * @param fn * @returns */ static wrap(fn: any) { if (!isFunction(fn)) { return fn; } return function () { const args: any[] = Array.prototype.slice.call(arguments); return new PromiseLike((resolve) => { fn.apply(null, args.concat(function (res: any, err: any) { res && resolve(res); err && resolve(err); })); }) } } 复制代码
Promise.sequence
-
链式调用的能力可以结合数组的
reduce
完成串行操作,把函数传入组合成新的函数。- 这里的参数不涉及
Promise
实例,使用链式调用来实现。
/** * 返回一个函数来执行 * @param fns * @returns */ static sequence(fns: Array<ICallbackFn>) { return (x: number) => fns.reduce((acc, fn: ICallbackFn) => { if (!isFunction(fn)) { fn = x => x; } return acc.then(fn).catch((err: any) => { throw err }); }, PromiseLike.resolve(x)); } 复制代码
- 这里的参数不涉及
-
假设有多个函数,我们可以通过这样的操作来将它们组合,组合的内容是处理函数。
function addThree(x) { return x + 3; } function addFive(x) { return x + 5; } const addEight = ProimseLike.sequence([addThree, addFive]); addEight(2); // 10 复制代码
-
上面的函数其实已经实现了串行;还可以做一些改动把每个值按顺序保存下来。
Promise.sequenceByOrder
-
该方法顺序执行函数,并返回按完成顺序排列的值。
/** * 串行执行所有 promises,并返回按返回顺序排列的数组 * 注意接收的参数是返回 promise 实例的函数组成的数组 * @param promises * @returns */ static sequenceByOrder(promises: Array<ICallbackFn>) { return new PromiseLike((resolve) => { let promiseResults: any = []; const reduceRes = promises.reduce((prevPromise, currentPromise: ICallbackFn, currentIndex: number) => { return prevPromise.then((val: any) => { promiseResults.push(val); const newVal = currentPromise(val); // 最后一次循环时保存,并剔除第一个值(默认 undefined) if (currentIndex === promises.length - 1) { promiseResults.unshift(); } return newVal; }); }, PromiseLike.resolve()); reduceRes.then((val: any) => { promiseResults.push(val); resolve(promiseResults); }); }); } 复制代码
Promise.map
-
定义一个可以处理所有
promise
值的函数,类似数组的map
方法。/** * 对每个 promise 的值进行特定的处理 * Promise.map([p1, p2, p3], (val, resolve) => { * resolve(val + 1); * }) * @param promises * @param fn * @returns */ static map(promises: Array<IPromiseType>, fn: any) { return PromiseLike.all(promises.map((currentPromise) => { return new PromiseLike((resolve) => { if (!isFunction(fn)) { fn = (val:any, resolve: ICallbackFn) => resolve(val); } fn(currentPromise, resolve); }) })); } 复制代码
Promise.observe
-
定义这样一个函数,用于清理
promise
相关的副作用,通常用在Promise.race
中。假设我们使用Promise.race
来设定超时,但仍然希望超时的场景里能够处理数据。/** * Promise.race([Promise.observe(p, cleanup // 处理函数), timeoutFn // 超时函数]) * @param promise * @param fn * @returns */ static observe(promise: IPromiseType, fn: ICallbackFn) { promise .then((res: any) => { PromiseLike.resolve(res).then(fn); }, (err) => { PromiseLike.resolve(err).then(fn); }); return promise; } 复制代码
06.源码
06.01 部分源码
-
所有源码虽不多,全部张贴出来也比较占版面。下面是
then
的完整实现。class PromiseLike { /** * 根据当前不同状态来执行对应逻辑 * 如果在默认状态就是注册对应事件 * 如果状态变化则是执行对应事件 * @param onFulfilled * @param onRejected * @returns */ then = (onFulfilled?: CallbackParams, onRejected?: CallbackParams) => { // 默认处理!!! onFulfilled = isFunction(onFulfilled) ? onFulfilled : value => value; onRejected = isFunction(onRejected) ? onRejected : err => { throw err }; /** * 该实现遵循 Promise/A+ 规范 * https://github.com/promises-aplus/promises-spec * @param promise * @param x * @param resolve * @param reject * @returns */ const resolvePromise = (promise: IPromiseType, x: any, resolve: ICallbackFn, reject: ICallbackFn) => { // 返回的 promise 不可以是当前的 promise 否则会造成死循环 if (newPromise === x) { reject(new TypeError(‘Chaining cycle detected for promise #<Promise>‘)); } // 对可能是 thenable 接口实现的对象判断 if (isObject(x) || isFunction(x)) { if (x === null) { return resolve(x); } let thenCb; try { thenCb = x.then; } catch (error) { return reject(error); } // 如果是 thenable 的对象,则调用其 then 方法 // 这一步涵盖了 Promise 实例的可能性 if (isFunction(thenCb)) { let isCalled = false; try { thenCb.call( x, // 指向当前函数或对象 (y: any) => { // 如果 resolvePromise 和 rejectPromise 都可能被调用 // 则只需调用第一次(resolvePromise 或 rejectPromise),后续无需再执行 if (isCalled) return; isCalled = true; // 传入当前函数,以实现递归展开调用 resolvePromise(promise, y, resolve, reject); }, (r: any) => { // 对应前面任意的调用之后,就不再只需后续逻辑 if (isCalled) return; isCalled = true; reject(r); } ) } catch (error) { if (isCalled) return; reject(error); } } else { resolve(x); } } else { resolve(x); } } // 定义变量,用于传参进行比较 const newPromise = new PromiseLike((resolve, reject) => { /** * 封装完成回调函数 * @param val */ const handleFulfilled = (val: any) => { try { const x = onFulfilled && onFulfilled(val); resolvePromise(newPromise, x, resolve, reject); } catch (error) { // 如果当前执行逻辑内发生异常,则抛出异常 reject(error); }; }; /** * 封装错误回调函数 * @param val */ const handleRejected = (val: any) => { try { const x = onRejected && onRejected(val); resolvePromise(newPromise, x, resolve, reject); } catch (error) { reject(error); } } switch (this.PromiseState) { case PROMISE_STATES.PENDING: this.resolveCallbackQueues.push(handleFulfilled); this.rejectCallbackQueues.push(handleRejected); break; case PROMISE_STATES.FULFILLED: handleFulfilled(this.PromiseResult); break; case PROMISE_STATES.REJECTED: handleRejected(this.PromiseResult); break; } }); return newPromise; } } 复制代码
06.02 全部源码
07.小结
- 在这次尝试实现
Promise
的过程中,自己也在边写边学。这是我理解和实践的整个思路,并不一定适用其他人;希望它能作为一种参考,启发或者影响到他人。在写这篇文章之前,我没想到会投入了整整两天时间,却也只是弄懂了些皮毛。 而Promise
内部显然还有许多值得探讨的地方,涉及的微任务,async/await
相关,迭代器和生成器;只是目前精力所限,先止于此。后面也许会解析迭代器和生成器的内容。 - 整个实践,也是自己练习
typescipt
的过程,这里我使用类的方式编写,主要是便于自己理解;但它也完全可以用函数实现。typescript
编译后的代码就是函数的实现,而且是 js. 可以直接查看下面的地址了解。另外,其中内容的编译转换也是值得探索的。