代码视界

Hanpeng Chen的个人博客

一起手写实现符合Promise/A+规范的Promise源码

本文于 1097 天之前发表,文中内容可能已经过时。

作者:Hanpeng_Chen

公众号:前端极客技术

Promise是JavaScript中异步编程的核心内容,也是前端面试的高频问题。关于Promise的基本用法这里不再详细介绍,接下来我们一起来实现一个符合 Promise/A+ 规范的Promise。

在开始写代码之前,我们要知道Promise/A+规范都有哪些内容。

Promise/A+ 规范

https://promisesaplus.com/

上面是Promise/A+规范的官方地址,是一个英文版本,大家可以自行点击阅读。下面我把规范中比较关键的点进行说明:

术语

  1. promise 是一个具有then方法的对象或函数,它的行为符合该规范。

  2. thenable 是一个定义了then方法的对象或函数。

  3. value 可以是任何一个合法的JavaScript的值,包括undefined、thenable、promise。

  4. exception 是一个异常,是在Promise里面可以用throw语句抛出的值。

  5. reason 是一个Promise里reject之后返回的拒绝原因。

状态

    1. 一个Promise有三种状态:pending、fulfilled、rejected。
    1. 当状态为pending时,可以转换为 fulfilled 或者 rejected。
    1. 当状态为 fulfilled 时,就不能再变为其他状态,必须返回一个不能再改变的值。
    1. 当状态为 rejected 时,就不能再变为其他状态,必须有一个promise被reject的原因,原因值也不能改变。

then方法

Promise必须拥有一个then方法,来访问最终的结果。

then方法有两个参数:

1
promise.then(onFulfilled, onRejected)

onFulfilled 和 onRejected
onFulfilled和onRejected两个都是可选参数,两者都必须是函数类型。

如果onFulfilled是函数,必须在promise变成fulfilled时,调用 onFulfilled,参数是promise的value,onFulfilled只能被调用一次。

如果onRejected是函数,必须在promise变成rejected时,调用 onRejected,参数是promise的reason,onRejected只能被调用一次。

then方法可以被多次调用
then方法可以被一个Promise调用多次,且必须返回一个Promise对象

如果promise变成了 fulfilled态,所有的onFulfilled回调都需要按照then的顺序执行

如果promise变成了 rejected态,所有的onRejected回调都需要按照then的顺序执行

Promise/A+规范的主要部分就是上面这些,接下来我们一起动手实现一个Promise。

实现 Promise

构造函数

Promise构造函数接受一个 executor 函数,executor函数执行完同步或异步操作后,调用它的两个参数resolve和reject。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
const PENDING = "pending";
const FULFILLED = "fulfilled";
const REJECTED = "rejected";


function Promise(executor) {
let self = this;
self.status = PENDING;
self.data = undefined; // Promise的值
self.onResolvedCallback = []; // Promise resolve时的回调函数
self.onRejectedCallback = []; // Promise rejected的回调函数集

function resolve(value) {

}
function reject(reason) {

}

try {
executor(resolve, reject)
} catch(e) {
reject(e)
}
}

Promise构造函数中定义了resolve和reject两个函数,下面一起来看看这两个函数如何实现?

从规范中我们知道:resolve和reject函数主要是返回对应状态的值value或者reason,并将Promise内部的状态从pending变成对应的状态,并且在这个状态改变之后是不可逆的。

那么这两个函数要如何实现,可以看下面这段代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
function Promise(executor) {
...

function resolve(value) {
if (self.status === PENDING) {
self.status = FULFILLED;
self.data = value;
for(let i = 0; i < self.onResolvedCallback.length; i++) {
self.onResolvedCallback[i](value)
}
}
}
function reject(reason) {
if (self.status === PENDING) {
self.status = REJECTED;
self.data = reason;
for(let i = 0; i < self.onRejectedCallback.length; i++) {
self.onRejectedCallback[i](reason);
}
}
}
...
}

then方法的实现

then方法接收两个参数,分别是Promise成功的回调 onFulfilled 和 失败的回调 onRejected。

如果调用then时,promise已经成功,则执行onFulfilled,并将value值作为参数传递进去;

如果promise已经失败,执行onRejected,将失败的原因reason作为参数传递进去。

如果promise的状态是pending,需要将onFulfilled和onRejected函数存放起来,等待状态确定后,再依次将对应的函数执行(发布订阅)

promise 可以then多次,promise 的then 方法返回一个 promise

我们把then方法实现的思路过了一遍,下面来看其实现代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
Promise.prototype.then = function(onFulfilled, onRejected) {
let self = this;
let promise2;

// 根据规范,如果then的参数不是function,则需要忽略它
onFulfilled = typeof onFulfilled === 'function' ? onFulfilled : value => value;
onRejected = typeof onRejected === 'function' ? onRejected : reason => { throw reason };

if (self.status === FULFILLED) {
// 如果promise状态确定为fulfilled,调用onFulFilled,但代码执行中可能会抛出,所以将其包裹在try/catch代码块中
return promise2 = new Promise(function(resolve, reject) {
try {
let x = onFulfilled(self.data)
if (x instanceof Promise) {
x.then(resolve, reject)
}
resolve(x)
} catch(e) {
reject(e)
}
})
}
if (self.status === REJECTED) {
return promise2 = new Promise(function(resolve, reject) {
try {
let x = onRejected(self.data);
if (x instanceof Promise) {
x.then(resolve, reject)
}
} catch(e) {
reject(e)
}
})
}
if (self.status === PENDING) {
return promise2 = new Promise(function(resolve, reject) {
self.onResolvedCallback.push(function(value) {
try {
let x = onFulfilled(self.data);
if (x instanceof Promise) {
x.then(resolve, reject)
}
} catch(e) {
reject(e)
}
})
self.onRejectedCallback.push(function(reason) {
try {
let x = onRejected(self.data)
if (x instanceof Promise) {
x.then(resolve, reject)
}
} catch(e) {
reject(e)
}
})
})
}
}

到这里一个符合Promise/A+规范的then方法基本实现了。

优化

在Promise/A+规范中,onFulfilled和onRejected两个函数需要异步调用。

并且规范中提到,要支持不同的Promise进行交互,关于不同的Promise交互详细说明点击下方链接查看:

https://promisesaplus.com/#point-46

基于上面两点,我们对上面实现的then方法中执行Promise的方法进行优化,在处理Promise进行resolve或reject时,加上 setTimeout(fn, 0)

优化后的完整代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
const PENDING = "pending";
const FULFILLED = "fulfilled";
const REJECTED = "rejected";

function Promise(executor) {
let self = this;
self.status = PENDING;
self.data = undefined; // Promise的值
self.onResolvedCallback = []; // Promise resolve时的回调函数
self.onRejectedCallback = []; // Promise rejected的回调函数集

function resolve(value) {
if (value instanceof Promise) {
return value.then(resolve, reject);
}
setTimeout(function () {
if (self.status === PENDING) {
self.status = FULFILLED;
self.data = value;
for (let i = 0; i < self.onResolvedCallback.length; i++) {
self.onResolvedCallback[i](value);
}
}
}, 0);
}
function reject(reason) {
setTimeout(function () {
if (self.status === PENDING) {
self.status = REJECTED;
self.data = reason;
for (let i = 0; i < self.onRejectedCallback.length; i++) {
self.onRejectedCallback[i](reason);
}
}
}, 0);
}

try {
executor(resolve, reject);
} catch (e) {
reject(e);
}
}

Promise.prototype.then = function (onFulfilled, onRejected) {
let self = this;
let promise2;

// 根据规范,如果then的参数不是function,则需要忽略它
onFulfilled =
typeof onFulfilled === "function" ? onFulfilled : function(v) {
return v;
};
onRejected =
typeof onRejected === "function"
? onRejected
: function(r) {
throw r;
};

if (self.status === FULFILLED) {
// 如果promise状态确定为fulfilled,调用onFulFilled,但代码执行中可能会抛出,所以将其包裹在try/catch代码块中
return promise2 = new Promise(function (resolve, reject) {
setTimeout(function(){
try {
let x = onFulfilled(self.data);
resolvePromise(promise2, x, resolve, reject)
} catch (e) {
reject(e);
}
})
});
}
if (self.status === REJECTED) {
return promise2 = new Promise(function (resolve, reject) {
setTimeout(function(){
try {
let x = onRejected(self.data);
resolvePromise(promise2, x, resolve, reject)
} catch (e) {
reject(e);
}
})
});
}
if (self.status === PENDING) {
return promise2 = new Promise(function (resolve, reject) {
self.onResolvedCallback.push(function (value) {
try {
let x = onFulfilled(self.data);
resolvePromise(promise2, x, resolve, reject)
} catch (e) {
reject(e);
}
});
self.onRejectedCallback.push(function (reason) {
try {
let x = onRejected(self.data);
resolvePromise(promise2, x, resolve, reject)
} catch (e) {
reject(e);
}
});
});
}
};


function resolvePromise(promise2, x, resolve, reject) {
let then;
let thenCalledOrThrow = false;
if (promise2 === x) {
return reject(new TypeError('Chaining cycle detected for promise'))
}
if (x instanceof Promise) {
if (x.status === PENDING) {
x.then(function(v) {
resolvePromise(promise2, v, resolve, reject)
}, reject)
} else {
x.then(resolve, reject)
}
return
}
if ((x !== null) && (typeof x === 'object' || typeof x === 'function')) {
try {
then = x.then;
if (typeof then === 'function') {
then.call(x, function(y) {
if (thenCalledOrThrow) return; // 已经调用过
thenCalledOrThrow = true;
return resolvePromise(promise2, y, resolve, reject)
}, function(r) {
if (thenCalledOrThrow) return;
thenCalledOrThrow = true;
return reject(r);
})
} else {
resolve(x)
}
} catch(e) {
if (thenCalledOrThrow) return;
thenCalledOrThrow = true;
return reject(e);
}
} else {
resolve(x)
}
}

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

module.exports = Promise;

测试

有专门的测试脚本可以测试我们写的Promise是否符合Promise/A+规范。

首先在Promise实现的代码中,暴露一个deferred方法:

1
2
3
4
5
6
7
8
Promise.defer = Promise.deferred = function(){
let dfd = {};
dfd.promise = new Promise(function(resolve, reject) {
dfd.resolve = resolve;
dfd.reject = reject;
});
return dfd;
}

在Promise源码的目录执行npm安装和执行测试命令:

1
2
3
npm i -g promises-aplus-tests

promises-aplus-tests Promise.js

promises-aplus-tests中共有872条测试用例。从上面的执行结果可以看出,我们手写的Promise通过所有用例。

Promise其他方法的实现

原生的Promise还提供了一些其他方法,比如:

  • Promise.resolve()
  • Promise.reject()
  • Promise.prototype.catch()
  • Promise.prototype.finally()
  • Promise.all()
  • Promise.race()

在面试中也经常碰到面试官要求手写实现这些方法,接下来我们一起看看这几个方法如何实现:

Promise.resolve

Promise.resolve(value)方法返回一个给定值解析后的Promise对象。

  • 如果 value 是个 thenable 对象,返回的promise会“跟随”这个thenable的对象,采用它的最终状态

  • 如果传入的value本身就是promise对象,那么Promise.resolve将不做任何修改、原封不动地返回这个promise对象。

  • 其他情况,直接返回以该值为成功状态的promise对象。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
Promise.resolve = function (value) {
if (value instanceof Promise) {
return value
}
return new Promise((resolve, reject) => {
if (value && typeof value === 'object' && typeof value.then === 'function') {
setTimeout(() => {
value.then(resolve, reject)
})
} else {
resolve(value)
}
})
}

Promise.reject

Promise.reject方法和Promise.resolve不同,Promise.reject()方法的参数,会原封不动地作为reject的理由,变成后续方法的参数。

1
2
3
4
5
Promise.reject = function(reason) {
return new Promise((resolve, reject) => {
reject(reason)
})
}

Promise.prototype.catch

Promise.prototype.catch 用于指定出错时的回调,是特殊的then方法,catch之后,可以继续.then

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

Promise.prototype.finally

不管成功还是失败,都会走到finally中,并且finally之后,还可以继续then。并且会将值原封不动的传递给后面的then.

1
2
3
4
5
6
7
8
9
10
11
Promise.prototype.finally = function(cb) {
return this.then((value) => {
return Promise.resolve(cb()).then(() => {
return value
})
}, (err) => {
return Promise.resolve(cb()).then(() => {
throw err;
})
})
}

Promise.all

Promise.all(promises) 返回一个promise对象

如果传入的参数是一个空的可迭代对象,那么此promise对象回调完成(resolve),只有此情况,是同步执行的,其它都是异步返回的。

如果传入的参数不包含任何 promise,则返回一个异步完成.

promises 中所有的promise都promise都“完成”时或参数中不包含 promise 时回调完成。

如果参数中有一个promise失败,那么Promise.all返回的promise对象失败

在任何情况下,Promise.all 返回的 promise 的完成状态的结果都是一个数组

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
Promise.all = function(promises) {
promises = Array.from(promises); // 将可迭代对象转换成数组
return new Promise((resolve, reject) => {
let index = 0;
let result = [];
if (promises.length === 0) {
resolve(result);
} else {
function processValue(i, value) {
result[i] = value;
if (++index === promises.length) {
resolve(result);
}
}
for (let i = 0; i < promises.length; i++) {
Promise.resolve(promises[i]).then((data) => {
processValue(i, data);
}, (err) => {
reject(err);
return;
})
}
}
})
}

Promise.race

Promise.race函数返回一个 Promise,它将与第一个传递的 promise 相同的完成方式被完成。它可以是完成( resolves),也可以是失败(rejects),这要取决于第一个完成的方式是两个中的哪个。

如果传的参数数组是空,则返回的 promise 将永远等待。

如果迭代包含一个或多个非承诺值和/或已解决/拒绝的承诺,则 Promise.race 将解析为迭代中找到的第一个值。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
Promise.race = function(promises) {
promises = Array.from(promises);
return new Promise((resolve, reject) => {
if (promises.length === 0) {
return;
} else {
for (let i = 0; i < promises.length; i++) {
Promise.resolve(promises[i]).then((data) => {
resolve(data);
return;
}, (err) => {
reject(err);
return;
})
}
}
})
}

本文所有源代码:handwriting-promise

总结

第一次写难免遇到各种问题,只要对照规范多写几次,自己总结一下,你也可以很快写出通过测试的Promise源码。

如果你能根据PromiseA+的规范,写出符合规范的源码,那么我想,对于面试中的Promise相关的问题,都能够给出比较完美的答案。

如果你觉得这篇内容对你有帮助的话:

1、点赞支持下吧,让更多的人也能看到这篇内容

2、关注公众号:前端极客技术,我们一起学习一起进步。

欢迎关注微信公众号: 『前端极客技术』『代码视界』
支付宝打赏 微信打赏

赞赏是不耍流氓的鼓励