模拟实现Promise,探究Promise原理

promise的功能

构造函数的实现

首先写一段promise的代码来分析一下构造函数的实现:

let pro = new Promise((resolve, reject) => {
  if (true) {
    resolve('the value')
  } else {
    reject('the reason')
  }
})

promise一共有pending(进行中)、fulfilled(已成功)、rejected(已失败)三种状态,一旦状态确定就不可更改。初始状态为pending,而且状态变更只有两种:

pending --> fulfilled

pending --> rejected

promise的构造函数,传了一个函数作为参数,这个函数接受promise返回的两个函数:resolve、reject,在构造函数中调用,调用后改变状态。

便可以写出如下构造函数:

const PENDING = 'pending' // 进行中
const FULFILLED = 'fulfilled' // 已成功
const REJECTED = 'rejected' // 已失败

class MyPromise {
  constructor(executor) {
    executor(this.resolve, this.reject)
  }
  // 初始状态PENDING
  status = PENDING

  resolve() {
    // 只有PENDING状态能变成FULFILLED状态
    if (this.status !== PENDING) return
    // 状态改变为成功
    this.status = FULFILLED
  }
  reject() {
    // 只有PENDING状态能变成REJECTED状态
    if (this.status !== PENDING) 
    // 状态变为失败
    this.status = REJECTED
  }
}
 

then方法的实现

调用一下上述写的promise

pro.then(res => {
    console.log(res)
}, (reason) => {
    console.log(reason)
})

promise原型上有一个then方法。then方法接受成功回调和失败回调两个参数。then方法内部判断状态, 如果是成功就调用成功回调函数,如果是失败就调用失败回调函数。

构造函数如果调用resolve会传递一个值,作为then方法中成功回调的参数;如果调用reject会传递一个值,作为then方法失败回调的参数。

便可以写出如下代码:

const PENDING = 'pending' // 进行中
const FULFILLED = 'fulfilled' // 已成功
const REJECTED = 'rejected' // 已失败

class MyPromise {
  constructor(executor) {
    executor(this.resolve, this.reject)
  }
  // 初始状态PENDING
  status = PENDING
  // 成功之后的值
  value = undefined
  // 失败之后的原因
  reason = undefined

  resolve = (value) => {
    // 只有PENDING状态能变成FULFILLED状态
    if (this.status !== PENDING) return
    // 状态改变为成功
    this.status = FULFILLED

    this.value = value
  }
  reject = (reason) => {
    // 只有PENDING状态能变成REJECTED状态
    if (this.status !== PENDING) 
    // 状态变为失败
    this.status = REJECTED

    this.reason = reason
  }
  then = (successCallback, failCallback) => {
    if (this.status === FULFILLED) {
      successCallback(this.value)
    } else {
      failCallback(this.reason)
    }
  }
}

tips: 每一步可以自行调用,验证是否已实现相关功能。

完善then支持异步

很明显我们的方法是不支持异步的,接下来实现异步功能:

let pro = new MyPromise((resolve, reject) => {
  setTimeout(() => {
    resolve(100)
  }, 1000)
})

pro.then((res) => {
  console.log(res)
})

很明显,我们需要在上述setTimeout内部执行的时候调用resolve,去改变状态。那then方法就会先调用,所以这时候我们应该在then的时候增加一种pending状态判断,这时候要把then传入的函数缓存起来。等到resolve调用的时候再去执行。实现如下:

 // 缓存成功回调函数
  successCallback = null
  // 缓存失败回调函数
  failCallback = null

  resolve = (value) => {
    // 只有PENDING状态能变成FULFILLED状态
    if (this.status !== PENDING) return
    // 状态改变为成功
    this.status = FULFILLED

    this.value = value
    // 执行成功回调
    if (this.successCallback) this.successCallback(this.value)
  }
  reject = (reason) => {
    // 只有PENDING状态能变成REJECTED状态
    if (this.status !== PENDING) 
    // 状态变为失败
    this.status = REJECTED

    this.reason = reason
    // 执行失败回调
    if (this.failCallback) this.failCallback(this.reason)
  }
  then = (successCallback, failCallback) => {
    if (this.status === FULFILLED) {
      successCallback(this.value)
    } else if (this.status === REJECTED) {
      failCallback(this.reason)
    } else {
       /**
       * pending状态 暂存成功回调和失败回调
       * 异步情况处理
       * **/
      this.successCallback = successCallback
      this.failCallback = failCallback
    }
  }
 

链式调用

promise是支持链式调用的,并且上一步的返回值会作为下一步的参数传入:

pro.then((res) => {
  console.log(res) // 100
  return 1000
}).then(res => {
  console.log(res) // 1000
  return 10000
}).then(res => {
  console.log(res) // 10000
}) 
//
pro.then().then((res) => console.log(res)) // 100

要实现链式调用,那then调用的时候还是要返回promise对象,因为then方法在promise上, 暂且将返回的promise对象叫做promise2,then方法就叫做then2。

let promise2 = new MyPromise((resolve, reject) => {})

这个返回的promise2调用resolve、reject的时机应该根据then2传入的successCallback、failCallback来决定。

这里要分为成功、失败和异步两种情况

成功、失败状态:

判断result的值是普通值还是promise对象:

如果是普通值 直接调用resolve

如果是promise对象 查看promise对象返回的结果,再根据promise对象返回的结果 决定调用resolve还是调用reject

异步:

暂存成功回调和失败回调

这时候我们要把多次调用的回调函数缓存起来,缓存到数组,resolve调用时,再逐一调用缓存的回调

当链式调用没有传递任何参数时,要向下传递值,所以我们在then内部,开始的时候判断, 如果没有成功或者失败回调,要直接返回值、或者失败原因:

/**
 * then不传递参数的时候不参数
 * 逐级传递参数
 ***/ 
 successCallback = successCallback ? successCallback:value => value
 failCallback = failCallback ? failCallback:reason => { throw reason }

具体实现如下:

  // 缓存成功回调函数
  successCallback = []
  // 缓存失败回调函数
  failCallback = []

  resolve = (value) => {
    // 只有PENDING状态能变成FULFILLED状态
    if (this.status !== PENDING) return
    // 状态改变为成功
    this.status = FULFILLED

    this.value = value
    if (this.successCallback.length) this.successCallback.shift()(this.value)
  }
  reject = (reason) => {
    // 只有PENDING状态能变成REJECTED状态
    if (this.status !== PENDING) 
    // 状态变为失败
    this.status = REJECTED

    this.reason = reason

    if (this.failCallback.length) this.failCallback.shift()(this.reason)
  }
  then = (successCallback, failCallback) => {
    /**
     * then不传递参数的时候不参数
     * 逐级传递参数
     * **/ 
    successCallback = successCallback ? successCallback:value => value
    failCallback = failCallback ? failCallback:reason => { throw reason }
    const promise2 = new MyPromise((resolve, reject) => {
      if (this.status === FULFILLED) {
        /**
         * 判断result的值是普通值还是promise对象:
         * 如果是普通值 直接调用resolve
         * 如果是promise对象 查看promise对象返回的结果
         * 再根据promise对象返回的结果 决定调用resolve还是调用reject
         * **/
        let result = successCallback(this.value)
        checkPromise(result, resolve, reject)
      } else if (this.status === REJECTED) {
        let result = failCallback(this.reason)
        checkPromise(result, resolve, reject)
      } else {
        /**
         * pending状态 暂存成功回调和失败回调
         * 异步情况处理
         * **/
        this.successCallback.push(() => {
          /**
           * 判断result的值是普通值还是promise对象:
           * 如果是普通值 直接调用resolve
           * 如果是promise对象 查看promise对象返回的结果
           * 再根据promise对象返回的结果 决定调用resolve还是调用reject
           * **/
          let result = successCallback(this.value)
          checkPromise(result, resolve, reject)
        })
        this.failCallback.push(() => {
          /**
           * 判断result的值是普通值还是promise对象:
           * 如果是普通值 直接调用resolve
           * 如果是promise对象 查看promise对象返回的结果
           * 再根据promise对象返回的结果 决定调用resolve还是调用reject
           * **/
          let result = failCallback(this.value)
          checkPromise(result, resolve, reject)
        })
      }
    })
    return promise2
  }

到这里 我们已经实现了promise的核心方法,下面再来补充promise 的一些调用方法。

catch实现

首先,为了代码的健壮性,我们要try-catch异常捕获。

在构造函数的时候,当捕获到异常要直接调用reject方法。

try {
   executor(this.resolve, this.reject)
} catch (error) {
   this.reject(error)
}

http://www.bijianshuo.com 软文发稿平台

在then方法中, 调用成功或失败回调的时候,如果捕获到异常,就直接走reject方法了。

try {
	let result = successCallback(this.value)
	checkPromise(result, resolve, reject)
} catch(error) {
	reject(error)
}

观察一下:

  pro.then()
     .catch(e => {
         
     })

catch 是promise原型上的,方法接收一个失败回调,实现如下:

 catch = (failCallback) => {
    this.then(null, failCallback)
  }
原文地址:https://www.cnblogs.com/qianxiaox/p/14982691.html