手寫Promise - 轉(zhuǎn)載

http://www.itdecent.cn/p/77a4f2c9d18b

手寫 Promise

非常贊,看了簡(jiǎn)易實(shí)現(xiàn)后粗略手寫了一版。
順帶發(fā)現(xiàn)typescript如果是寫庫(kù)定義類型還是很傷腦筋。隨便寫寫還是很方便的。

enum PromiseState {
  Reject = -1,
  Pending,
  Resolved,
}

interface Supplier{
  (param: any) : void
}

class MyPromise {
  private state = PromiseState.Pending;
  private value: any = null;

  private resolveCBList: Function[] = [];
  private rejectCBList: Function[] = [];

  constructor(fn: (resolve: Supplier, reject: Supplier) => void) {
    fn(this.resolve.bind(this), this.reject.bind(this));
  }

  private resolve(value: any) {
    if (this.state === PromiseState.Pending) {
      this.state = PromiseState.Resolved;
      this.value = value;
      this.resolveCBList.map(cb => cb(value))
    }
  }

  private reject(value: any) {
    if (this.state === PromiseState.Pending) {
      this.state = PromiseState.Reject;
      this.value = value;
      this.rejectCBList.map(cb => cb(value));
    }
  }

  then(onFullFilled: Supplier) {
    if (this.state === PromiseState.Pending) {
      this.resolveCBList.push(onFullFilled);
      //this.rejectCBList.push(onRejected);
    }

    if (this.state === PromiseState.Resolved) {
      onFullFilled(this.value);
    }
  }
}

new MyPromise((resovle, reject) => {
  setTimeout(() => {
    console.log('resolve');
    resovle('hello promise');
  }, 1000);
}).then((value) => {
  console.log(value);
});

我們會(huì)通過(guò)手寫一個(gè)符合 Promise/A+ 規(guī)范的 Promise 來(lái)深入理解它,并且手寫 Promise 也是一道大廠??碱},在進(jìn)入正題之前,推薦各位閱讀一下 Promise/A+ 規(guī)范,這樣才能更好地理解這個(gè)章節(jié)的代碼。

實(shí)現(xiàn)一個(gè)簡(jiǎn)易版 Promise

在完成符合 Promise/A+ 規(guī)范的代碼之前,我們可以先來(lái)實(shí)現(xiàn)一個(gè)簡(jiǎn)易版 Promise,因?yàn)樵诿嬖囍?,如果你能?shí)現(xiàn)出一個(gè)簡(jiǎn)易版的 Promise 基本可以過(guò)關(guān)了。

那么我們先來(lái)搭建構(gòu)建函數(shù)的大體框架

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

function MyPromise(fn) {
  const that = this
  that.state = PENDING
  that.value = null
  that.resolvedCallbacks = []
  that.rejectedCallbacks = []
  // 待完善 resolve 和 reject 函數(shù)
  // 待完善執(zhí)行 fn 函數(shù)
}

  • 首先我們創(chuàng)建了三個(gè)常量用于表示狀態(tài),對(duì)于經(jīng)常使用的一些值都應(yīng)該通過(guò)常量來(lái)管理,便于開發(fā)及后期維護(hù)
  • 在函數(shù)體內(nèi)部首先創(chuàng)建了常量 that,因?yàn)榇a可能會(huì)異步執(zhí)行,用于獲取正確的 this 對(duì)象
  • 一開始 Promise 的狀態(tài)應(yīng)該是 pending
  • value 變量用于保存 resolve 或者 reject 中傳入的值
  • resolvedCallbacksrejectedCallbacks 用于保存 then 中的回調(diào),因?yàn)楫?dāng)執(zhí)行完 Promise 時(shí)狀態(tài)可能還是等待中,這時(shí)候應(yīng)該把 then 中的回調(diào)保存起來(lái)用于狀態(tài)改變時(shí)使用

接下來(lái)我們來(lái)完善 resolvereject 函數(shù),添加在 MyPromise 函數(shù)體內(nèi)部

function resolve(value) {
  if (that.state === PENDING) {
    that.state = RESOLVED
    that.value = value
    that.resolvedCallbacks.map(cb => cb(that.value))
  }
}

function reject(value) {
  if (that.state === PENDING) {
    that.state = REJECTED
    that.value = value
    that.rejectedCallbacks.map(cb => cb(that.value))
  }
}

這兩個(gè)函數(shù)代碼類似,就一起解析了

  • 首先兩個(gè)函數(shù)都得判斷當(dāng)前狀態(tài)是否為等待中,因?yàn)橐?guī)范規(guī)定只有等待態(tài)才可以改變狀態(tài)
  • 將當(dāng)前狀態(tài)更改為對(duì)應(yīng)狀態(tài),并且將傳入的值賦值給 value
  • 遍歷回調(diào)數(shù)組并執(zhí)行

完成以上兩個(gè)函數(shù)以后,我們就該實(shí)現(xiàn)如何執(zhí)行 Promise 中傳入的函數(shù)了

try {
  fn(resolve, reject)
} catch (e) {
  reject(e)
}

  • 實(shí)現(xiàn)很簡(jiǎn)單,執(zhí)行傳入的參數(shù)并且將之前兩個(gè)函數(shù)當(dāng)做參數(shù)傳進(jìn)去
  • 要注意的是,可能執(zhí)行函數(shù)過(guò)程中會(huì)遇到錯(cuò)誤,需要捕獲錯(cuò)誤并且執(zhí)行 reject 函數(shù)

最后我們來(lái)實(shí)現(xiàn)較為復(fù)雜的 then 函數(shù)

MyPromise.prototype.then = function(onFulfilled, onRejected) {
  const that = this
  onFulfilled = typeof onFulfilled === 'function' ? onFulfilled : v => v
  onRejected =
    typeof onRejected === 'function'
      ? onRejected
      : r => {
          throw r
        }
  if (that.state === PENDING) {
    that.resolvedCallbacks.push(onFulfilled)
    that.rejectedCallbacks.push(onRejected)
  }
  if (that.state === RESOLVED) {
    onFulfilled(that.value)
  }
  if (that.state === REJECTED) {
    onRejected(that.value)
  }
}

  • 首先判斷兩個(gè)參數(shù)是否為函數(shù)類型,因?yàn)檫@兩個(gè)參數(shù)是可選參數(shù)

  • 當(dāng)參數(shù)不是函數(shù)類型時(shí),需要?jiǎng)?chuàng)建一個(gè)函數(shù)賦值給對(duì)應(yīng)的參數(shù),同時(shí)也實(shí)現(xiàn)了透?jìng)?,比如如下代碼

    // 該代碼目前在簡(jiǎn)單版中會(huì)報(bào)錯(cuò)
    // 只是作為一個(gè)透?jìng)鞯睦?Promise.resolve(4).then().then((value) => console.log(value))
    
    
  • 接下來(lái)就是一系列判斷狀態(tài)的邏輯,當(dāng)狀態(tài)不是等待態(tài)時(shí),就去執(zhí)行相對(duì)應(yīng)的函數(shù)。如果狀態(tài)是等待態(tài)的話,就往回調(diào)函數(shù)中 push 函數(shù),比如如下代碼就會(huì)進(jìn)入等待態(tài)的邏輯

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

以上就是簡(jiǎn)單版 Promise 實(shí)現(xiàn),接下來(lái)一小節(jié)是實(shí)現(xiàn)完整版 Promise 的解析,相信看完完整版的你,一定會(huì)對(duì)于 Promise 的理解更上一層樓。

實(shí)現(xiàn)一個(gè)符合 Promise/A+ 規(guī)范的 Promise

這小節(jié)代碼需要大家配合規(guī)范閱讀,因?yàn)榇蟛糠执a都是根據(jù)規(guī)范去實(shí)現(xiàn)的。

我們先來(lái)改造一下 resolvereject 函數(shù)

function resolve(value) {
  if (value instanceof MyPromise) {
    return value.then(resolve, reject)
  }
  setTimeout(() => {
    if (that.state === PENDING) {
      that.state = RESOLVED
      that.value = value
      that.resolvedCallbacks.map(cb => cb(that.value))
    }
  }, 0)
}
function reject(value) {
  setTimeout(() => {
    if (that.state === PENDING) {
      that.state = REJECTED
      that.value = value
      that.rejectedCallbacks.map(cb => cb(that.value))
    }
  }, 0)
}

  • 對(duì)于 resolve 函數(shù)來(lái)說(shuō),首先需要判斷傳入的值是否為 Promise 類型
  • 為了保證函數(shù)執(zhí)行順序,需要將兩個(gè)函數(shù)體代碼使用 setTimeout 包裹起來(lái)

接下來(lái)繼續(xù)改造 then 函數(shù)中的代碼,首先我們需要新增一個(gè)變量 promise2,因?yàn)槊總€(gè) then 函數(shù)都需要返回一個(gè)新的 Promise 對(duì)象,該變量用于保存新的返回對(duì)象,然后我們先來(lái)改造判斷等待態(tài)的邏輯

if (that.state === PENDING) {
  return (promise2 = new MyPromise((resolve, reject) => {
    that.resolvedCallbacks.push(() => {
      try {
        const x = onFulfilled(that.value)
        resolutionProcedure(promise2, x, resolve, reject)
      } catch (r) {
        reject(r)
      }
    })

    that.rejectedCallbacks.push(() => {
      try {
        const x = onRejected(that.value)
        resolutionProcedure(promise2, x, resolve, reject)
      } catch (r) {
        reject(r)
      }
    })
  }))
}

  • 首先我們返回了一個(gè)新的 Promise 對(duì)象,并在 Promise 中傳入了一個(gè)函數(shù)
  • 函數(shù)的基本邏輯還是和之前一樣,往回調(diào)數(shù)組中 push 函數(shù)
  • 同樣,在執(zhí)行函數(shù)的過(guò)程中可能會(huì)遇到錯(cuò)誤,所以使用了 try...catch 包裹
  • 規(guī)范規(guī)定,執(zhí)行 onFulfilled 或者 onRejected 函數(shù)時(shí)會(huì)返回一個(gè) x,并且執(zhí)行 Promise 解決過(guò)程,這是為了不同的 Promise 都可以兼容使用,比如 JQuery 的 Promise 能兼容 ES6 的 Promise

接下來(lái)我們改造判斷執(zhí)行態(tài)的邏輯

if (that.state === RESOLVED) {
  return (promise2 = new MyPromise((resolve, reject) => {
    setTimeout(() => {
      try {
        const x = onFulfilled(that.value)
        resolutionProcedure(promise2, x, resolve, reject)
      } catch (reason) {
        reject(reason)
      }
    })
  }))
}

  • 其實(shí)大家可以發(fā)現(xiàn)這段代碼和判斷等待態(tài)的邏輯基本一致,無(wú)非是傳入的函數(shù)的函數(shù)體需要異步執(zhí)行,這也是規(guī)范規(guī)定的
  • 對(duì)于判斷拒絕態(tài)的邏輯這里就不一一贅述了,留給大家自己完成這個(gè)作業(yè)

最后,當(dāng)然也是最難的一部分,也就是實(shí)現(xiàn)兼容多種 PromiseresolutionProcedure 函數(shù)

function resolutionProcedure(promise2, x, resolve, reject) {
  if (promise2 === x) {
    return reject(new TypeError('Error'))
  }
}

首先規(guī)范規(guī)定了 x 不能與 promise2 相等,這樣會(huì)發(fā)生循環(huán)引用的問(wèn)題,比如如下代碼

let p = new MyPromise((resolve, reject) => {
  resolve(1)
})
let p1 = p.then(value => {
  return p1
})

然后需要判斷 x 的類型

if (x instanceof MyPromise) {
    x.then(function(value) {
        resolutionProcedure(promise2, value, resolve, reject)
    }, reject)
}

這里的代碼是完全按照規(guī)范實(shí)現(xiàn)的。如果 xPromise 的話,需要判斷以下幾個(gè)情況:

  1. 如果 x 處于等待態(tài),Promise 需保持為等待態(tài)直至 x 被執(zhí)行或拒絕
  2. 如果 x 處于其他狀態(tài),則用相同的值處理 Promise

當(dāng)然以上這些是規(guī)范需要我們判斷的情況,實(shí)際上我們不判斷狀態(tài)也是可行的。

接下來(lái)我們繼續(xù)按照規(guī)范來(lái)實(shí)現(xiàn)剩余的代碼

let called = false
if (x !== null && (typeof x === 'object' || typeof x === 'function')) {
  try {
    let then = x.then
    if (typeof then === 'function') {
      then.call(
        x,
        y => {
          if (called) return
          called = true
          resolutionProcedure(promise2, y, resolve, reject)
        },
        e => {
          if (called) return
          called = true
          reject(e)
        }
      )
    } else {
      resolve(x)
    }
  } catch (e) {
    if (called) return
    called = true
    reject(e)
  }
} else {
  resolve(x)
}

  • 首先創(chuàng)建一個(gè)變量 called 用于判斷是否已經(jīng)調(diào)用過(guò)函數(shù)
  • 然后判斷 x 是否為對(duì)象或者函數(shù),如果都不是的話,將 x 傳入 resolve
  • 如果 x 是對(duì)象或者函數(shù)的話,先把 x.then 賦值給 then,然后判斷 then 的類型,如果不是函數(shù)類型的話,就將 x 傳入 resolve
  • 如果 then 是函數(shù)類型的話,就將 x 作為函數(shù)的作用域 this 調(diào)用之,并且傳遞兩個(gè)回調(diào)函數(shù)作為參數(shù),第一個(gè)參數(shù)叫做 resolvePromise ,第二個(gè)參數(shù)叫做 rejectPromise,兩個(gè)回調(diào)函數(shù)都需要判斷是否已經(jīng)執(zhí)行過(guò)函數(shù),然后進(jìn)行相應(yīng)的邏輯
  • 以上代碼在執(zhí)行的過(guò)程中如果拋錯(cuò)了,將錯(cuò)誤傳入 reject 函數(shù)中

以上就是符合 Promise/A+ 規(guī)范的實(shí)現(xiàn)了,如果你對(duì)于這部分代碼尚有疑問(wèn),歡迎在評(píng)論中與我互動(dòng)。

作者:Adoins
鏈接:http://www.itdecent.cn/p/77a4f2c9d18b
來(lái)源:簡(jiǎn)書
簡(jiǎn)書著作權(quán)歸作者所有,任何形式的轉(zhuǎn)載都請(qǐng)聯(lián)系作者獲得授權(quán)并注明出處。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容