Promise 源码详解

发布于 2023-05-02 10:43:35 字数 13336 浏览 43 评论 0

1.回调地狱

曾几何时，我们的代码是这样的，为了拿到回调的结果，不得不 callback hell ，这种环环相扣的代码可以说是相当恶心了

let fs = require('fs')
fs.readFile('./a.txt','utf8',function(err,data){
  fs.readFile(data,'utf8',function(err,data){
    fs.readFile(data,'utf8',function(err,data){
      console.log(data)
    })
  })
})

终于，我们的 盖世英雄 出现了，他身披金甲圣衣、驾着七彩祥云。好吧打岔儿了，没错他就是我们的 Promise ，那让我们来看看用了 Promise 之后，上面的代码会变成什么样吧

let fs = require('fs')
function read(url){
  return new Promise((resolve,reject)=>{
    fs.readFile(url,'utf8',function(error,data){
      error && reject(error)
      resolve(data)
    })
  })
}

read('./a.txt').then(data=>{
return read(data)
}).then(data=>{
return read(data)
}).then(data=>{
console.log(data)
})

如上所示真的是很方便，有木有？意中人可以说是 Swag 到变形了。那么言归正传，我们怎么才能自己写一个这么 Swag 的解决异步神器呢？

2.重点开始，小眼睛都看过来

2.1 Promise/A+

首先我们要知道自己手写一个 Promise ，应该怎么去写，谁来告诉我们怎么写，需要遵循什么样的规则。当然这些你都不用担心，其实业界都是通过一个规则指标来生产 Promise 的。让我们来看看是什么东西。传送门 Promise/A+

2.2 constructor

我们先声明一个类，叫做 Promise ，里面是构造函数。如果es6还有问题的可以去阮大大的博客上学习一下（传送门 es6）

class Promise{
  constructor(executor){
    //控制状态，使用了一次之后，接下来的都不被使用
    this.status = 'draft'
    this.value = undefined
    this.reason = undefined
//定义resolve函数
let resolve = (data)=&gt;{
  //这里draft，主要是为了防止executor中调用了两次resovle或reject方法，而我们只调用一次
  if(this.status==='draft'){
    this.status = 'resolve'
    this.value = data
  } 
}
//定义reject函数
let reject = (data)=&gt;{
  if(this.status==='draft'){
    this.status = 'reject'        
    this.reason = data
  } 
}
//executor方法可能会抛出异常，需要捕获
try{
  //将resolve和reject函数给使用者      
  executor(resolve,reject)      
}catch(e){
  //如果在函数中抛出异常则将它注入reject中
  reject(e)
}
}
}

那么接下来我会分析上面代码的作用，原理

executor： 这是实例 Promise 对象时在构造器中传入的参数，一般是一个 function(resolve,reject){}
status：``Promise 的状态，一开始是默认的draft状态，每当调用道resolve和reject方法时，就会改变其值，在后面的then方法中会用到
value： resolve回调成功后，调用resolve方法里面的参数值
reason： reject回调成功后，调用reject方法里面的参数值
resolve： 声明resolve方法在构造器内，通过传入的executor方法传入其中，用以给使用者回调
reject： 声明reject方法在构造器内，通过传入的executor方法传入其中，用以给使用者回调

2.3 then

then方法是 Promise 中最为重要的方法，他的用法大家都应该已经知道，就是将 Promise 中的resolve或者reject的结果拿到，那么我们就能知道这里的then方法需要两个参数，成功回调和失败回调，上代码！

then(onFufilled,onRejected){  
  if(this.status === 'resolve'){
    onFufilled(this.value)
  }
  if(this.status === 'reject'){
    onRejected(this.reason)
  }
}

这里主要做了将构造器中resolve和reject的结果传入 onFufilled 和 onRejected 中，注意这两个是使用者传入的参数，是个方法。所以你以为这么简单就完了？要想更 Swag 的应对各种场景，我们必须得再完善。继续往下走！

3.异步的 Promise

之前我们只是处理了同步情况下的 Promise，简而言之所有操作都没有异步的成分在内。那么如果是异步该怎么办？

3.1 callback

最早处理异步的方法就是callback，就相当于我让你帮我扫地，我会在给你发起任务时给你一个手机，之后我做自己的事情去，不用等你，等你扫完地就会打手机给我，诶，我就知道了地扫完了。这个手机就是callback，回调函数。

首先我们需要改一下构造器里的代码，分别添加两个回调函数的数组，分别对应成功回调和失败回调。他们的作用是当成功执行resolve或reject时，执行callback。

//存放成功回调的函数
this.onResolvedCallbacks = []
//存放失败回调的函数
this.onRejectedCallbacks = []
let resolve = (data)=>{
if(this.status==='draft'){
this.status = 'resolve'
this.value = data
//监听回调函数
this.onResolvedCallbacks.forEach(fn=>fn())
}
}
let reject = (data)=>{
if(this.status==='draft'){
this.status = 'reject'
this.reason = data
this.onRejectedCallbacks.forEach(fn=>fn())
}
}

然后是then需要多加一个状态判断，当 Promise 中是异步操作时，需要在我们之前定义的回调函数数组中添加一个回调函数。

if(this.status === 'draft'){
  this.onResolvedCallbacks.push(()=>{
    // to do....
    let x = onFufilled(this.value)
    resolvePromise(promise2,x,resolve,reject)
  })
  this.onRejectedCallbacks.push(()=>{
    let x = onRejected(this.reason)
    resolvePromise(promise2,x,resolve,reject)
  })
}

大功告成，异步已经解决了

3.2 resolvePromise

这也是 Promise 中的重头戏，我来介绍一下，我们在用Promise的时候可能会发现，当then函数中return了一个值，我们可以继续then下去，不过是什么值，都能在下一个then中获取，还有，当我们不在then中放入参数，例： promise.then().then() ，那么其后面的then依旧可以得到之前then返回的值，可能你现在想很迷惑。让我来解开你心中的忧愁，follow me。

then(onFufilled,onRejected){ 
    //解决onFufilled,onRejected没有传值的问题
    onFufilled = typeof onFufilled === 'function'?onFufilled:y=>y
    //因为错误的值要让后面访问到，所以这里也要跑出个错误，不然会在之后then的resolve中捕获
    onRejected = typeof onRejected === 'function'?onRejected:err=>{ throw err ;}
    //声明一个promise对象
    let promise2
    if(this.status === 'resolve'){
      //因为在.then之后又是一个promise对象，所以这里肯定要返回一个promise对象
      promise2 = new Promise((resolve,reject)=>{
        setTimeout(()=>{
          //因为穿透值的缘故，在默认的跑出一个error后，不能再用下一个的reject来接受，只能通过try，catch        
          try{
            //因为有的时候需要判断then中的方法是否返回一个promise对象，所以需要判断
            //如果返回值为promise对象，则需要取出结果当作promise2的resolve结果
            //如果不是，直接作为promise2的resolve结果
            let x = onFufilled(this.value)
            //抽离出一个公共方法来判断他们是否为promise对象
            resolvePromise(promise2,x,resolve,reject)
          }catch(e){
            reject(e)
          }
        },0)
      })
    }
    if(this.status === 'reject'){
      promise2 = new Promise((resolve,reject)=>{
        setTimeout(()=>{
          try{
            let x = onRejected(this.reason)
            resolvePromise(promise2,x,resolve,reject)
          }catch(e){
            reject(e)
          }
        },0)
      })
    }
    if(this.status === 'draft'){
      promise2 = new Promise((resolve,reject)=>{
        this.onResolvedCallbacks.push(()=>{
          // to do....
          setTimeout(()=>{
            try{
              let x = onFufilled(this.value)
              resolvePromise(promise2,x,resolve,reject)
            }catch(e){
              reject(e)
            }
          },0)
        })
        this.onRejectedCallbacks.push(()=>{
          setTimeout(()=>{
            try{
              let x = onRejected(this.reason)
              resolvePromise(promise2,x,resolve,reject)
            }catch(e){
              reject(e)
            }
          })
        })
      })
    }
    return promise2
  }

一下子多了很多方法，不用怕，我会一一解释

返回 Promise ？：首先我们要注意的一点是，then有返回值，then了之后还能在then，那就说明之前的then返回的必然是个 Promise 。
为什么外面要包一层 setTimeout ？：因为Promise本身是一个异步方法，属于微任务一列，必须得在执行栈执行完了在去取他的值，所以所有的返回值都得包一层异步setTimeout。
为什么开头有两个判断？：这就是之前想要解决的如果then函数中的参数不是函数，那么我们需要做处理。如果onFufilled不是函数，就需要自定义个函数用来返回之前resolve的值，如果onRejected不是函数，自定义个函数抛出异常。这里会有个小坑，如果这里不抛出异常，会在下一个then的onFufilled中拿到值。又因为这里抛出了异常所以所有的onFufilled或者onRejected都需要try/catch，这也是Promise/A+的规范。当然本人觉得成功的回调不需要抛出异常也可以，大家可以仔细想想。
resolvePromise 是什么？：这其实是官方Promise/A+的需求。因为你的then可以返回任何职，当然包括 Promise 对象，而如果是 Promise 对象，我们就需要将他拆解，直到它不是一个 Promise 对象，取其中的值。

那就让我们来看看这个 resolvePromise 到底长啥样。

function resolvePromise(promise2,x,resolve,reject){
  //判断x和promise2之间的关系
  //因为promise2是上一个promise.then后的返回结果，所以如果相同，会导致下面的.then会是同一个promise2，一直都是，没有尽头
  if(x === promise2){//相当于promise.then之后return了自己，因为then会等待return后的promise，导致自己等待自己，一直处于等待
    return reject(new TypeError('循环引用'))
  }
  //如果x不是null，是对象或者方法
  if(x !== null && (typeof x === 'object' || typeof x === 'function')){
    //为了判断resolve过的就不用再reject了，（比如有reject和resolve的时候）
    let called
    try{//防止then出现异常，Object.defineProperty
      let then = x.then//取x的then方法可能会取到{then:{}},并没有执行
      if(typeof then === 'function'){
        //我们就认为他是promise,call他,因为then方法中的this来自自己的promise对象
        then.call(x,y=>{//第一个参数是将x这个promise方法作为this指向，后两个参数分别为成功失败回调
          if(called) return;
          called = true
          //因为可能promise中还有promise，所以需要递归
          resolvePromise(promise2,y,resolve,reject)
        },err=>{
          if(called) return;
          called = true
          //一次错误就直接返回
          reject(err)
        })
      }else{
        //如果是个普通对象就直接返回resolve作为结果
        resolve(x)
      }
    }catch(e){
      if(called) return;
      called = true
      reject(e)
    }
  }else{
    //这里返回的是非函数，非对象的值,就直接放在promise2的resolve中作为结果
    resolve(x)
  }
}

它的作用是用来将 onFufilled 的返回值进行判断取值处理，把最后获得的值放入最外面那层的 Promise 的resolve函数中。

参数 promise2 （then函数返回的Promise对象）， x （onFufilled函数的返回值）， resolve、reject （最外层的Promise上的resolve和reject）。
为什么要在一开始判断 promise2 和 x ？：首先在Promise/A+中写了需要判断这两者如果相等，需要抛出异常，我就来解释一下为什么，如果这两者相等，我们可以看下下面的例子，第一次p2是p1.then出来的结果是个 Promise 对象，这个 Promise 对象在被创建的时候调用了resolvePromise(promise2,x,resolve,reject)函数，又因为x等于其本身，是个 Promise ，就需要then方法递归它，直到他不是 Promise 对象，但是x（p2）的结果还在等待，他却想执行自己的then方法，就会导致等待。

let p1 = new Promise((resolve,reject)=>{
  resolve()
})
let p2 = p1.then(d=>{
return p2
})

called 是用来干嘛的？：called 变量主要是用来判断如果 resolvePromise 函数已经resolve或者reject了，那就不需要在执行下面的resolce或者reject。

为什么取then这个属性？：因为我们需要去判断x是否为Promise，then属性如果为普通值，就直接resolve掉，如果是个function，就是Promise对象，之后我们就需要将这个x的then方法进行执行，用call的原因是因为then方法里面this指向的问题。
为什么要递归去调用 resolvePromise 函数？：相信细心的人已经发现了，我这里使用了递归调用法，首先这是Promise/A+中要求的，其次是业务场景的需求，当我们碰到那种Promise的resolve里的Promise的resolve里又包了一个Promise的话，就需要递归取值，直到x不是Promise对象。

4.完善Promise

我们现在已经基本完成了 Promise 的 then 方法，那么现在我们需要看看他的其他方法。

4.1 catch

相信大家都知道catch这个方法是用来捕获Promise中的reject的值，也就是相当于then方法中的onRejected回调函数，那么问题就解决了。我们来看代码。

//catch方法
catch(onRejected){
  return this.then(null,onRejected)
}

该方法是挂在 Promise 原型上的方法。当我们调用catch传 callback 的时候，就相当于是调用了then方法。

4.2 resolve/reject

大家一定都看到过 Promise.resolve()、Promise.reject() 这两种用法，它们的作用其实就是返回一个Promise对象，我们来实现一下。

//resolve方法
Promise.resolve = function(val){
  return new Promise((resolve,reject)=>{
    resolve(val)
  })
}
//reject方法
Promise.reject = function(val){
  return new Promise((resolve,reject)=>{
    reject(val)
  })
}

这两个方法是直接可以通过class调用的，原理就是返回一个内部是 resolve 或 reject 的 Promise 对象。

4.3 all

all方法可以说是Promise中很常用的方法了，它的作用就是将一个数组的Promise对象放在其中，当全部resolve的时候就会执行then方法，当有一个reject的时候就会执行catch，并且他们的结果也是按着数组中的顺序来排放的，那么我们来实现一下。

//all方法(获取所有的promise，都执行then，把结果放到数组，一起返回)
Promise.all = function(promises){
  let arr = []
  let i = 0
  function processData(index,data){
    arr[index] = data
    i++
    if(i == promises.length){
      resolve(arr)
    }
  }
  return new Promise((resolve,reject)=>{
    for(let i=0;i<promises.length;i++){
      promises[i].then(data=>{
        processData(i,data)
      },reject)
    }
  })
}

其原理就是将参数中的数组取出遍历，每当执行成功都会执行 processData 方法， processData 方法就是用来记录每个Promise的值和它对应的下标，当执行的次数等于数组长度时就会执行 resolve，把arr的值给 then。这里会有一个坑，如果你是通过arr数组的长度来判断他是否应该resolve的话就会出错，为什么呢？因为js数组的特性，导致如果先出来的是1位置上的值进arr，那么0位置上也会多一个空的值，所以不合理。

4.4 race

race方法虽然不常用，但是在Promise方法中也是一个能用得上的方法，它的作用是将一个 Promise 数组放入race中，哪个先执行完，race就直接执行完，并从then中取值。我们来实现一下吧。

//race方法
Promise.race = function(promises){
  return new Promise((resolve,reject)=>{
    for(let i=0;i<promises.length;i++){
      promises[i].then(resolve,reject)
    }
  })
}

原理大家应该看懂了，很简单，就是遍历数组执行 Promise，如果有一个 Promise 执行成功就resolve。

Promise语法糖 deferred

语法糖这三个字大家一定很熟悉，作为一个很Swag的前端工程师，对async/await这对兄弟肯定很熟悉，没错他们就是generator的语法糖。而我们这里要讲的语法糖是Promise的。

//promise语法糖 也用来测试
Promise.deferred = Promise.defer = function(){
  let dfd = {}
  dfd.promise = new Promise((resolve,reject)=>{
    dfd.resolve = resolve
    dfd.reject = reject
  })
  return dfd
}

什么作用呢？看下面代码你就知道了

let fs = require('fs')
let Promise = require('./promises')
//Promise上的语法糖，为了防止嵌套，方便调用
//坏处 错误处理不方便
function read(){
  let defer = Promise.defer()
  fs.readFile('./1.txt','utf8',(err,data)=>{
    if(err)defer.reject(err)
    defer.resolve(data)
  })
  return defer.Promise
}

没错，我们可以方便的去调用他语法糖defer中的 Promise 对象。那么它还有没有另外的方法呢？答案是有的。我们需要在全局上安装promises-aplus-tests插件 npm i promises-aplus-tests -g ，再输入promises-aplus-tests [js文件名] 即可验证你的Promise的规范。

分享到QQ

分享到微博