手写promise

// 判断变量是否为function
const isFunction = variable => typeof variable === 'function'
// 三种状态
const PENDING = 'PENDING'
const FULFILLED = 'FULFILLED'
const REJECTED = 'REJECTED'
// 定义Promise的三种状态常量
class MyPromise {
  constructor(handle) {
    if (!isFunction(handle)) {
      throw new Error('MyPromise must accept a function as a parameter')
    }
    // 添加状态
    this._status = PENDING
    // 添加状态
    this._value = undefined
    // 添加成功回调函数队列
    this._fulfilledQueues = []
    // 添加失败回调函数队列
    this._rejectedQueues = []
    // 执行handle
    try {
      handle(this._resolve.bind(this), this._reject.bind(this))
    } catch (err) {
      this._reject(err)
    }
  }
  // 添加resolve时执行的函数
  _resolve(val) {
    const run = () => {
      if(this._status !== PENDING) return
      // 依次执行成功队列中的函数,并清空队列
      const runFulfilled = (value) => {
        let cb;
        while(cb = this._fulfilledQueues.shift()) {
          cb(value)
        }
      }
      // 依次执行失败队列中的函数,并清空队列
      const runRejected = (error) => {
        let cb;
        while(cb = this._rejectedQueues.shift()) {
          cb(error)
        }
      }
      /**
      * 如果resolve的参数为Promise对象,则必须等待该Promise对象状态改变后,
      * 当前Promise的状态才会改变,且状态取决于参数Promise对象的状态
      */ 
      if(val instanceof MyPromise) {
        val.then(value => {
          this._value = value
          this._status = FULFILLED
          runFulfilled(value)
        }, err => {
          this._value = err
          this._status = REJECTED
          runRejected(err)
        })
      } else {
        this._value = val
        this._status = FULFILLED
        runFulfilled(val)
      }
    }
    // 为了支持同步的Promise, 这里采用异步调用
    setTimeout(() => run(), 0)
  }
  // 添加reject时执行的函数
  _reject(err) {
    if(this._status !== PENDING) return
    const run = () => {
      this._status = REJECTED
      this._value = err
      let cb;
      while(cb = this._rejectedQueues.shift()) {
        cb(err)
      }
    }
    setTimeout(run, 0)
  }

  // 添加then方法
  then(onFulfilled, onRejected) {
    const {_value, _status } = this
    // 返回一个新的Promise对象
    return new Promise((onFulfilledNext, onRejectedNext) => {
      // 封装一个成功时执行的函数
      let fulfilled = value => {
        try {
          if(!isFunction(onFulfilled)) {
            onFulfilledNext(value)
          } else {
            let res = onRejected(value);
            if(res instanceof MyPromise) {
              // 如果当前回调函数返回MyPromise对象,必须等待其状态改变后再执行下一个回调
              res.then(onFulfilledNext, onRejectedNext)
            } else {
              onFulfilledNext(res)
            }
          }
        } catch (err) {
          onRejectedNext(err)
        }
      }
    // 封装一个失败时执行的函数
    let rejected = error => {
      try {
        if(!isFunction(onRejected)){
          onRejectedNext(error)
        } else {
          let res = onRejected(error)
          if(res instanceof MyPromise) {
            // 如果当前回调函数返回MyPromise对象, 必须等待其状态改变后再执行下一个回调
            onFulfilledNext(res)
          }
        }
      } catch (error) {
        // 如果函数执行出错,新的Promise方法回调加入执行队列等待执行
        onRejectedNext(error)
      }
    }
    switch (_status) {
      // 当状态为pending时,将then方法回调函数加入执行队列等待执行
      case PENDING:
        this._fulfilledQueues.push(onFulfilled)
        this._rejectedQueues.push(onRejected)
        break
      case FULFILLED:
        onFulfilled(_value)
        break
      case REJECTED:
        onRejected(_value)
        break
      }
    })
  }

  // 添加catch方法
  catch(onRejected) {
    return this.then(undefined, onRejected)
  }

  // 添加静态resolve方法
  static resolve(value) {
    // 如果参数是MyPromise实例,直接返回这个实例
    if(value instanceof MyPromise) return value
    return new MyPromise(resolve => resolve(value))
  }

  // 静态reject方法
  static reject (value) {
    return new MyPromise((resolve, reject) => reject(value))
  }

  // 静态all方法
  static all (list) {
    return new MyPromise((resolve, reject) => {
      /** 
       * 返回值的集合
      */
      let values = []
      let count = 0
      for(let [i, p] of list.entries()) {
        // 数组参数如果不是MyPromise实例,先调用MyPromise.resolve
        this.resolve(p).then(res => {
          values[i] = res
          count++
          if(count === list.length) resolve(values)
        }, err => {
          rejected(err)
        })
      }
    })
  }

  // 添加静态race方法
  static race (list) {
    return new MyPromise((resolve, reject) => {
      for(let p of list) {
        // 只要有一个实例率先改变状态,新的MyPromise的状态就跟着改变
        this.resolve(p).then(res => {
          resolve(res)
        }, err => {
          reject(err)
        })
      }
    })
  }

  finally(cb) {
    return this.then(
      value => MyPromise.resolve(cb()).then(() => value),
      reason => MyPromise.resolve((cb)).then(() => { throw reason})
    )
  }

}
const name = new MyPromise((resolve, reject) => {
  console.log(1)
  resolve()
})
name.then((res, rej) => {
  console.log(2)
})
上一篇:手把手一行一行代码教你“手写Promise“,完美通过 Promises/A+ 官方872个测试用例


下一篇:手写一个MyPromise构造函数