本文由作者陈旭锋(任职网易考拉)授权网易云社区发布。

Promise源码详解
学习知识要善于思考,思考,再思考。 —— 爱因斯坦

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 = 'pendding'
this.value = undefined
this.reason = undefined

//定义resolve函数
let resolve = (data)=>{      //这里pendding,主要是为了防止executor中调用了两次resovle或reject方法,而我们只调用一次
  if(this.status==='pendding'){        this.status = 'resolve'
    this.value = data
  } 
}    //定义reject函数
let reject = (data)=>{      if(this.status==='pendding'){        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的状态,一开始是默认的pendding状态,每当调用道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==='pendding'){ this.status = 'resolve'

this.value = data    //监听回调函数
this.onResolvedCallbacks.forEach(fn=>fn())

}
}let reject = (data)=>{ if(this.status==='pendding'){ this.status = 'reject'

this.reason = data    this.onRejectedCallbacks.forEach(fn=>fn())

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

if(this.status === 'pendding'){ 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)

})
}
ok!大功告成,异步已经解决了

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 === 'pendding'){
  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的规范。

5.结尾
今天我们就做了一个属于自己的Promise项目,理解里面的源码,方法原理,希望大家都有收获。当然有什么意见大家都可以在评论区评论,peace and love。

git地址:https://github.com/Shinemax1/...

更多网易技术、产品、运营经验分享请访问网易云社区。

文章来源: 网易云社区


网易数帆
391 声望550 粉丝

网易数智旗下全链路大数据生产力平台,聚焦全链路数据开发、治理及分析,为企业量身打造稳定、可控、创新的数据生产力平台,服务“看数”、“管数”、“用数”等业务场景,盘活数据资产,释放数据价值。