从零实现一个 Promise

Promise 作为由社区提出和实现的异步编程解决方案,ES6 将其写进了语言标准,统一了用法,原生提供了 Promise 对象。本文将剖析 Promise 内部标准,根据 Promises/A+ 规范从零实现一个 Promise。

Promise 构造函数

在 Promise 构造函数中,主要操作是初始化状态和数据以及执行函数参数。

首先需要将状态初始化为 pending,然后定义 Promise 的值以及回调函数集。

const PENDING = 'pending'
const RESOLVED = 'resolved'
const REJECTED = 'rejected'

function MyPromise(executor) {
  let self = this
  self.status = PENDING // Promise 状态,初始状态为 pending
  self.data = undefined  // Promise 的值
  self.onResolvedCallback = [] // Promise resolve 时的回调函数集
  self.onRejectedCallback = [] // Promise reject 时的回调函数集

  // 待完善,resolve 和 reject 函数
  // 待完善,执行 executor 函数
}

在构造函数中,还需要执行由外部传进来的 executor 函数,executor 函数中有两个函数参数,分别为 resolve 和 reject 函数。

function MyPromise(executor) {
  let self = this
  self.status = PENDING // Promise 状态,初始状态为 pending
  self.data = undefined  // Promise 的值
  self.onResolvedCallback = [] // Promise resolve 时的回调函数集
  self.onRejectedCallback = [] // Promise reject 时的回调函数集

  function resolve(value) {
    // 当状态为 pending 时,改变状态为 resolved,存储 Promise 值以及执行回调函数集
    if (self.status === PENDING) {
      self.status = RESOLVED
      self.data = value
      self.onResolvedCallback.map(cb => cb(value))
    }
  }

  function reject(reason) {
    // 当状态为 pending 时,改变状态为 rejected,存储 Promise 值以及执行回调函数集
    if (self.status === PENDING) {
      self.status = REJECTED
      self.data = reason
      self.onRejectedCallback.map(cb => cb(reason))
    }
  }

  try {
    executor(resolve, reject)
  } catch (e) {
    // executor 函数执行中抛出错误时该 Promise 应该被 reject
    reject(e)
  }
}

executor 函数需要使用 try catch 包裹执行的原因则是在 executor 函数执行中可能会抛出错误,当抛出错误时则该 Promise 应该被 reject,如下情况:

// 该 Promise 应该被 reject
new Promise(function(resolve, reject) {
  throw 2
})
then 方法

then 方法主要是根据 Promise 当前状态处理相应的逻辑,返回一个新的 Promise,新 Promise 的状态由原先 Promise 的状态和 then 方法函数参数中的返回值决定。

当 then 方法执行时,该 Promise 的状态是不确定的,所以需要对 Promise 的状态进行判断然后执行不同的操作,then 方法会返回一个新的 Promise。

MyPromise.prototype.then = function (onResolved, onRejected) {
  let self = this
  let promise2

  // 如果 then 的参数不是 function,则我们需要赋予默认函数实现值的透传
  onResolved = typeof onResolved === 'function' ? onResolved : value => value
  onRejected = typeof onRejected === 'function' ? onRejected : reason => { throw reason }

  if (self.status === RESOLVED) {
    return promise2 = new MyPromise((resolve, reject) => {
      
    })
  }

  if (self.status === REJECTED) {
    return promise2 = new MyPromise((resolve, reject) => {
      
    })
  }

  if (self.status === PENDING) {
    return promise2 = new MyPromise((resolve, reject) => {
      
    })
  }
}

then 方法会返回一个新的 Promise 后,新 Promise 的状态由原先 Promise 的状态和 then 方法函数参数中的返回值决定。

MyPromise.prototype.then = function (onResolved, onRejected) {
  let self = this
  let promise2

  // 如果 then 的参数不是 function,则我们需要赋予默认函数实现值的透传
  onResolved = typeof onResolved === 'function' ? onResolved : value => value
  onRejected = typeof onRejected === 'function' ? onRejected : reason => { throw reason }

  if (self.status === RESOLVED) {
    return promise2 = new MyPromise((resolve, reject) => {
      try {
        // 执行 onResolved 函数并获取返回值。若返回值是 Promise 对象,则取它的结果作为 promise2 的结果,否则以返回值作为 promise2 的结果
        var x = onResolved(self.data)
        if (x instanceof MyPromise) {
          x.then(resolve, reject)
        }
        resolve(x)
      } catch (e) {
        // 抛出错误则以捕获到的错误作为 promise2 的结果
        reject(e)
      }
    })
  }

  if (self.status === REJECTED) {
    return promise2 = new MyPromise((resolve, reject) => {
      try {
        // 执行 onRejected 函数并获取返回值。若返回值是 Promise 对象,则取它的结果作为 promise2 的结果,否则以返回值作为 promise2 的结果
        var x = onRejected(self.data)
        if (x instanceof MyPromise) {
          x.then(resolve, reject)
        }
        reject(x)
      } catch (e) {
        // 抛出错误则以捕获到的错误作为 promise2 的结果
        reject(e)
      }
    })
  }

  if (self.status === PENDING) {
    return promise2 = new MyPromise((resolve, reject) => {
      // 将回调函数存进回调函数集
      self.onResolvedCallback.push((value) => {
        try {
          var x = onResolved(self.data)
          if (x instanceof MyPromise) {
            x.then(resolve, reject)
          }
          resolve(x)
        } catch (e) {
          reject(e)
        }
      })
      self.onRejectedCallback.push((reason) => {
        try {
          var x = onRejected(self.data)
          if (x instanceof MyPromise) {
            x.then(resolve, reject)
          } 
          reject(x)
        } catch (e) {
          reject(e)
        }
      })
    })
  }
}

在 then 方法中根据 Promise 的当前状态分别执行了不同的操作。当状态为 resolved 时,执行 onResolved 函数(then 方法第一个函数参数)并根据返回值确定 promise2 的状态;当状态为 rejected 时,执行 onRejected 函数(then 方法第二个函数参数)并根据返回值确定 promise2 的状态;当状态为 pending 时,则需要将 onResolved 和 onRejected 函数先存进回调函数集中,等到 Promise 状态改变后再执行。

而在代码注释中说明,如果 then 的参数不是 function,则我们需要赋予默认函数实现值的透传

当传进 then 方法中 onResolved 或 onRejected 函数参数为空时,则应该赋予它们一个默认函数,该默认函数 return 或 throw 原先的参数值,这样才能正确实现 then 方法的链式调用,如下:

new MyPromise((resolve, reject) => { resolve(1) })
  .then()
  .then()
  .then((value) => {
    console.log(value)
  })

至此,我们便完成了一个符合 Promises/A+ 规范的 Promise 基础版,同原生 Promise 一样,可以通过如下方式使用:

let myPromise = new MyPromise((resolve, reject) => {
  if (/* 异步操作成功 */) {
    resolve(value);
  } else {
    reject(error);
  }
});

myPromise.then((value) => {
  console.log(value)
}, (err) => {
  console.log(err)
})
Promise 终极版

上述的代码已经能够实现一个基本 Promise 的功能,而在实际使用过程中,我们还需要根据 Promises/A+ 规范继续完善它。

需要完善的主要有以下两点:

  1. 不同 Promise 之间的兼容;
  2. 异步调用操作;

在实际中,有多种不同的 Promise 实现,关于不同 Promise 间的交互, Promises/A+ 规范已经做了详细的说明,其中详细指定了如何通过 then 的实参的返回值来决定 promise2 的状态,我们只需要按照标准将内容转成代码即可。

function resolvePromise(promise2, x, resolve, reject) {
  var then
  var 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(value) {
        resolvePromise(promise2, value, 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 rs(y) { 
          if (thenCalledOrThrow) return
          thenCalledOrThrow = true
          return resolvePromise(promise2, y, resolve, reject) 
        }, function rj(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)
  }
}

所以,在 then 方法中,我们不再需要判断返回值 x 的类型,然后再根据 x 的类型去决定 promise2 的状态,只需要将其传入 resolvePromise 函数即可。

// self.status === RESOLVED 部分更改,其余两个状态更改同理
var x = onResolved(self.data)
if (x instanceof MyPromise) {
  x.then(resolve, reject)
}
resolve(x)
=>
var x = onResolved(self.data)
resolvePromise(promise2, x, resolve, reject)

最后,在标准中,说明了某些地方需要使用异步调用,在我们的实现中,我们需要在 resolve、reject、onResolved、onRejected 加上异步调用的代码,这里我们使用 setTimeout(fn, 0) 来实现。

至此,我们实现了一个符合 Promises/A+ 规范的终极版 Promise,如下:

const PENDING = 'pending'
const RESOLVED = 'resolved'
const REJECTED = 'rejected'

function MyPromise(executor) {
  let self = this
  self.status = PENDING // Promise 状态,初始状态为 pending
  self.data = undefined  // Promise 的值
  self.onResolvedCallback = [] // Promise resolve 时的回调函数集
  self.onRejectedCallback = [] // Promise reject 时的回调函数集

  function resolve(value) {
    setTimeout(() => { // 异步回调
      // 当状态为 pending 时,改变状态为 resolved,存储 Promise 值以及执行回调函数集
      if (self.status === PENDING) {
        self.status = RESOLVED
        self.data = value
        self.onResolvedCallback.map(cb => cb(value))
      }
    }, 0)
  }

  function reject(reason) {
    setTimeout(() => { // 异步回调
      // 当状态为 pending 时,改变状态为 rejected,存储 Promise 值以及执行回调函数集
      if (self.status === PENDING) {
        self.status = REJECTED
        self.data = reason
        self.onRejectedCallback.map(cb => cb(reason))
      }
    }, 0)
  }

  try {
    executor(resolve, reject)
  } catch (e) {
    // executor 函数执行中抛出错误时该 Promise 应该被 reject
    reject(e)
  }
}

MyPromise.prototype.then = function (onResolved, onRejected) {
  let self = this
  let promise2

  // 如果 then 的参数不是 function,则我们需要赋予默认函数实现值的透传
  onResolved = typeof onResolved === 'function' ? onResolved : value => value
  onRejected = typeof onRejected === 'function' ? onRejected : reason => { throw reason }

  if (self.status === RESOLVED) {
    return promise2 = new MyPromise((resolve, reject) => {
      setTimeout(() => { // 异步回调
        try {
          // 执行 onResolved 函数并获取返回值。若返回值是 Promise 对象,则取它的结果作为 promise2 的结果,否则以返回值作为 promise2 的结果
          var x = onResolved(self.data)
          resolvePromise(promise2, x, resolve, reject)
        } catch (e) {
          // 抛出错误则以捕获到的错误作为 promise2 的结果
          reject(e)
        }
      }, 0)
    })
  }

  if (self.status === REJECTED) {
    return promise2 = new MyPromise((resolve, reject) => {
      setTimeout(() => { // 异步回调
        try {
          // 执行 onRejected 函数并获取返回值。若返回值是 Promise 对象,则取它的结果作为 promise2 的结果,否则以返回值作为 promise2 的结果
          var x = onRejected(self.data)
          resolvePromise(promise2, x, resolve, reject)
        } catch (e) {
          // 抛出错误则以捕获到的错误作为 promise2 的结果
          reject(e)
        }
      }, 0)
    })
  }

  if (self.status === PENDING) {
    return promise2 = new MyPromise((resolve, reject) => {
      // 将回调函数存进回调函数集
      self.onResolvedCallback.push((value) => {
        try {
          var x = onResolved(self.data)
          resolvePromise(promise2, x, resolve, reject)
        } catch (e) {
          reject(e)
        }
      })
      self.onRejectedCallback.push((reason) => {
        try {
          var x = onRejected(self.data)
          resolvePromise(promise2, x, resolve, reject)
        } catch (e) {
          reject(e)
        }
      })
    })
  }
}

function resolvePromise(promise2, x, resolve, reject) {
  var then
  var 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 (value) {
        resolvePromise(promise2, value, 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 rs(y) {
          if (thenCalledOrThrow) return
          thenCalledOrThrow = true
          return resolvePromise(promise2, y, resolve, reject)
        }, function rj(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)
  }
}

参考文章:
https://github.com/xieranmaya/blog/issues/2

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,470评论 6 501
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,393评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,577评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,176评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,189评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,155评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,041评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,903评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,319评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,539评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,703评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,417评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,013评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,664评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,818评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,711评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,601评论 2 353

推荐阅读更多精彩内容