入门Promise的正确姿势中我们已经了解到Promise的基本用法。那么现在给你一个需求:根据Promise的用法和Promise/A+规范,实现一个自己的Promise函数。

1、Promise构造函数的结构

根据Promise的用法我们知道:在new一个Promise函数的时候,Promise函数必须接受一个函数作为参数。我们暂且把这个参数函数称为执行器。

执行器提供两个参数(resolve,reject),且内部有状态机制(pending,resolved,rejected)。Promise构造函数的原型上有then方法。

我们再来看一下PromiseA+标准中是怎样规定的:

  1. Promise对象初始状态为 Pending,在被 resolve 或 reject 时,状态变为 Fulfilled或 Rejected
  2. resolve接收成功的数据,reject接收失败或错误的数据
  3. Promise对象必须有一个 then 方法,且只接受两个函数参数 onFulfilledonRejected

由以上标准就容易就能实现这个类的大致结构。

开始撸:

var PENDING = 'pending',
RESOLVED = 'resolved',
REJECTED = 'rejected',
function Promise(executor) {
//养成良好的编程习惯,先对参数做一个容错处理
if (executor && typeof executor != 'function') {
throw new Error('Promise is not a function');
}
let _this = this;//缓存this
_this.status = PENDING;//当前Promise的状态
_this.value = undefined;//Promise成功执行时要传递给回调的数据,默认为undefined
_this.reason = undefined;//Promise失败执行时传递给毁掉的原因,默认为undefined
function resolve(value) { //内置一个resolve方法
if (_this.status == PENDING) {
_this.status = RESOLVED;//当调用resolve时,将Promise的状态改为resoled
_this.value = value;//保存成功调用时传递进来的数据
}
}
function reject(reason) {
if (_this.status == PENDING) {
_this.status = REJECTED;//当调用reject时,将Promise的状态改为rejected
_this.reason = reason;//保存失败调用时传递进来的原因
}
}
executor(resolve,reject); }
//then方法可以接受两个函数参数,分别表示当前Promise执行成功时的调用onFulfilled和执行失败时调用onRejected
Promise.prototype.then = function(onFulfilled,onRejected) {
let _this = this,//缓存this,保不齐后面会用到,当然如果你不想缓存this,也可以在后面使用箭头函数
if (_this.status == RESOLVED) {
onFulfilled(_this.value);
}
if (_this.status == REJECTED) {
onFulfilled(_this.reason);
}
}

看起来不错,但回调函数是立即执行的,也就是说上面实现的Promise只支持同步代码,而无法进行异步操作,比如这样是不行的

let p = new Promise(function(resolve, reject){
setTimeout(function(){
resolve('成功执行了!');
}, 1000)
})
p.then(function(data){
console.log('成功', data)
},function(err){
console.log('失败', err)
})
// 不会输出任何代码

原因是:我们在then函数中只对成功态和失败态进行了判断,而实例被new时,执行器中的代码会立即执行,但setTimeout中的代码将稍后执行,也就是说,then方法执行时,Promise的状态没有被改变依然是pending态。

所以我们要对pending态也做判断,而由于代码可能是异步的,所以回调函数就不该被立即执行,所以我们就要想办法把回调函数进行缓存,当状态改变后(由pending态变成resolved或rejected态),再执行相应的函数。

那么状态在什么时候改变呢:很显然,在执行resolve或rejected函数的时候,状态会发生改变。

并且,then方法是可以多次使用的,所以要能存多个回调,那么这里我们用一个数组来存储多个回调函数。

1.1、实现异步

在实例上挂载两个参数

_this.onResolvedCallbacks = []; // 存放then成功的回调
_this.onRejectedCallbacks = []; // 存放then失败的回调

我们再给then方法加一个pending时的判断

if(_this.status === 'pending'){
// 每一次then时,如果是等待态,就把回调函数push进数组中,什么时候改变状态什么时候再执行
_this.onResolvedCallbacks.push(function(){ // 这里用一个函数包起来,是为了后面加入新的逻辑进去
onFulfilled(_this.value)
})
_this.onRejectedCallbacks.push(function(){ // 同理
onRjected(_this.reason)
})
}

下一步要分别在resolve和reject方法里加入执行数组中存放的函数的方法,修改一下上面的resolve和reject方法

function resolve(value) {
if (_this.status === 'pending') {
_this.status = 'resolved'
_this.value = value
_this.onResolvedCallbacks.forEach(function(fn){ // 当成功的函数被调用时,之前缓存的回调函数会被一一调用
fn()
})
}
}
function reject(reason) {
if (_this.status === 'pending') {
_this.status = 'rejected'
_this.reason = reason
_this.onRejectedCallbacks.forEach(function(fn){// 当失败的函数被调用时,之前缓存的回调函数会被一一调用
fn()
})
}
}

现在可以执行异步任务了,也可以给一个Promise函数多次then了。

1.2、错误处理

上面的代码看似很完美,前提是整个Promise函数能够正确执行,那样就没reject函数什么事了。但我们必须要考虑的一种情况是:当Promise函数执行出错时怎么办,所以一个健壮的Promise函数是应该有错误处理机制的。所以我们应该在Promise的代码体中加入try catch,如果出现异常,则捕捉错误交给reject。

我们实现一下,思路很简单,在执行器执行时进行try catch

try{
executor(resolve, reject)
}catch(e){ // 如果捕获发生异常,直接调失败,并把参数穿进去
reject(e)
}

2、实现then的链式调用(难点)

上面说过了,then可以链式调用,也是这一点让Promise十分好用,当然这部分源码也比较复杂

我们知道jQuery实现链式调用是return了一个this,但Promise不行,为什么不行?

因为then函数内返回的是一个新的Promise对象。我们看一下标准是如何定义的:

标准中规定:

  1. then 方法必须返回一个新的 Promise实例(ES6中的标准,Promise/A+中没有明确说明)

  2. 为了保证 then中回调的执行顺序,onFulfilled 或 onRejected 必须异步调用

所以我们要做的就是:在then方法中先定义一个新的Promise,取名为promise2,然后在三种状态下分别用promise2包装一下,在调用onFulfilled时用一个变量x接收返回值,try catch一下代码,没错就调resolve传入x,有错就调reject传入错误,最后再把promise2给return出去,就可以进行链式调用了

//修改then
Promise.prototype.then = function(onFulfilled,onRejected) {
onFulfilled = typeof onFulfilled === 'function' ? onFulfilled : function(value) {return vaule};
onRejected = typeof onRejected === 'function' ? onFulfilled : function(err) {throw err}; let _this = this;//缓存this,保不齐后面会用到,当然如果你不想缓存this,也可以在后面使用箭头函数
let promise2;
if (_this.status === RESOLVED) {
promise2 = new Promise(function(resolve,reject) {
try {
//onFulfilled函数的执行情况要考虑多种情况,后面会细说
let x = onFulfilled(_this.value);//将执行onFulfilled的返回值传给x,这里需要注意的是执行过程中有可能会出错
resolve(x);
} catch(e) {
reject(e);
}
}) }
if (_this.status === REJECTED) {
promise2 = new Promise(function(resolve,reject) {
try {
//onFulfilled函数的执行情况要考虑多种情况,后面会细说
let x = onRejected(_this.reason);//将执行onFulfilled的返回值传给x(即这次then函数执行的返回值),这里需要注意的是执行过程中有可能会出错
resolve(x);//这个x会被下一次的then函数接收到
} catch(e) {
reject(e);
}
}) } if(_this.status === PENDING){
promise2 = new Promise(function(resolve,reject) {
// 每一次then时,如果是等待态,就把回调函数push进数组中,什么时候改变状态什么时候再执行
_this.onResolvedCallbacks.push(function(){ // 这里用一个函数包起来,是为了后面加入新的逻辑进去
try {
let x = onFulfilled(_this.value);
resolve(x);
} catch(e) {
reject(e);
}
})
_this.onRejectedCallbacks.push(function(){ // 同理
try {
let x = onRejected(_this.reason);
resolve(x);
} catch(e) {
reject(e);
}
})
}) }
}

上面的实现虽然能用,但是很粗糙。

在接下的分析之前,我希望大家能够清晰的明白x的值代表什么:x表示的是上一次的then函数或promise函数执行结果的返回值,这个x的值会被resolve(x),作为下一次then函数调用时的参数。

明确了x的值以后,接下来对onFulfilled和onRejected函数可能出现的情况做一个列举,这也是为了我们的Promise函数能做到最大的容错率:

  1. 前一次then返回一个普通值,字符串数组对象这些东西,都没问题,只需传给下一个then,刚才的方法就够用。
  2. 前一次then返回的是一个Promise,是正常的操作,也是Promise提供的语法糖,我们要想办法判断到底返回的是啥。
  3. 前一次then返回的是一个Promise,其中有异步操作,也是理所当然的,那我们就要等待他的状态改变,再进行下面的处理。
  4. 前一次then返回的是自己本身这个Promise
var p1 = p.then(function(){// 这里得用var,let由于作用域的原因会报错undefined
return p1
})

  5.前一次then返回的是一个别人自己随便写的Promise,这个Promise可能是个有then的普通对象,比如{then:'哈哈哈'},也有可能在then里故意抛错(这种蛋疼的 操作我们也要考虑进去)。比如他这样写

let promise = {}
Object.defineProperty(promise,'then',{
value: function(){
throw new Error('报错气死你')
}
})
// 如果返回这东西,我们再去调then方法就肯定会报错了

  6.调resolve的时候再传一个Promise下去,我们还得处理这个Promise。

p.then(function(data) {
return new Promise(function(resolve, reject) {
resolve(new Promise(function(resolve,reject){
resolve(1111)
}))
})
})

  7.可能既调resolve又调reject,得忽略后一个。

  8.光then,里面啥也不写。

问题8最好解决,我们只要手动添加一个onFulfilled和onRejected进去就好了。

对于问题1-7,我们可以采取统一的觉得方案,定义一个函数来判断和处理这一系列的情况,官方给出了一个叫做resolvePromise的函数。

Promise原理详解-LMLPHP

所以,我们进一步来完善then方法。

Promise.prototype.then = function(onFulfilled,onRejected) {
onFulfilled = typeof onFulfilled === 'function' ? onFulfilled : function(value) {return value};
onRejected = typeof onRejected === 'function' ? onRejected : function(err) {throw err};
   //这里正好解释了在正式的Promise函数中,我们为什么可以不写onRejected函数,
   //因为then方法内部会帮我们封装好一个onRejected函数,用来抛出上一次then或Promise执行出错的信息,这也是为什么可以在最后执行catch方法的原因 let _this = this;//缓存this,保不齐后面会用到,当然如果你不想缓存this,也可以在后面使用箭头函数
let promise2;
if (_this.status === RESOLVED) {
promise2 = new Promise(function(resolve,reject) {
try {
//onFulfilled函数的执行情况要考虑多种情况,后面会细说
let x = onFulfilled(_this.value);//将执行onFulfilled的返回值传给x,这里需要注意的是执行过程中有可能会出错
resolvePromise(promise2,x,resolve,reject);
} catch(e) {
reject(e);
}
}) }
if (_this.status === REJECTED) {
promise2 = new Promise(function(resolve,reject) {
try {
//onFulfilled函数的执行情况要考虑多种情况,后面会细说
let x = onRejected(_this.reason);//将执行onFulfilled的返回值传给x(即这次then函数执行的返回值),这里需要注意的是执行过程中有可能会出错
resolvePromise(promise2,x,resolve,reject);//这个x会被下一次的then函数接收到
} catch(e) {
reject(e);
}
}) } if(_this.status === PENDING){
promise2 = new Promise(function(resolve,reject) {
// 每一次then时,如果是等待态,就把回调函数push进数组中,什么时候改变状态什么时候再执行
_this.onResolvedCallbacks.push(function(){ // 这里用一个函数包起来,是为了后面加入新的逻辑进去
try {
let x = onFulfilled(_this.value);
resolvePromise(promise2,x,resolve,reject);
} catch(e) {
reject(e);
}
})
_this.onRejectedCallbacks.push(function(){ // 同理
try {
let x = onRejected(_this.reason);
resolvePromise(promise2,x,resolve,reject);
} catch(e) {
reject(e);
}
})
}) }
return promise2;
}

定义resolvePromise方法:

function resolvePromise(promise2, x, resolve, reject) {
// 接受四个参数: 新的Promise、返回值,成功和失败的回调
// 有可能这里返回的x是别人的promise
// 尽可能允许其他乱写
if (promise2 === x) { //这里应该报一个类型错误,来解决问题4
return reject(new TypeError('循环引用了'))
}
// 看x是不是一个promise,promise应该是一个对象
let called = false; // 表示是否调用过成功或者失败,用来解决问题7
//下面判断上一次then返回的是普通值还是函数,来解决问题1、2
if (x !== null && (typeof x === 'object' || typeof x === 'function')) {
// 可能是promise {},看这个对象中是否有then方法,如果有then我就认为他是promise了
try {
let then = x.then;// 保存一下x的then方法
if (typeof then === 'function') {
// 成功
//用call方法修改指针为x,否则this指向window
then.call(x, function (y) {//如果x是一个Promise对象,y参数表示x执行后的resolve值
//console.log(y);
if (called) return //如果调用过就return掉
called = true
// y可能还是一个promise,在去解析直到返回的是一个普通值
resolvePromise(promise2, y, resolve, reject)//递归调用,解决了问题6
}, function (err) { //失败时执行的函数
if (called) return
called = true
console.log('r');
reject(err);
})
} else {//如果x不是一个Promise对象,则直接resolve(x)
resolve(x)
}
} catch (e) {
if (called) return
called = true;
reject(e);
}
} else { // 说明是一个普通值
resolve(x); // 表示成功了
}
}

2.1、catch方法

由于catch方法是then(null, onRejected)的语法糖,所以这里也很好实现。

Promise.prototype.catch = function(onRejected) {
return this.then(null,onRejected);
}

至此,一个简易粗糙的Promise函数已经实现了。同学们可以自己用自己写的这个Promise函数验证一下是否是否可行。

05-11 21:57