摘要：如果狀態(tài)是等待態(tài)的話，就往回調(diào)函數(shù)中函數(shù)，比如如下代碼就會(huì)進(jìn)入等待態(tài)的邏輯以上就是簡(jiǎn)單版實(shí)現(xiàn)實(shí)現(xiàn)一個(gè)符合規(guī)范的接下來(lái)大部分代碼都是根據(jù)規(guī)范去實(shí)現(xiàn)的。

為更好的理解， 推薦閱讀Promise/A+ 規(guī)范

在完成符合 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)管理，便于開(kāi)發(fā)及后期維護(hù)

在函數(shù)體內(nèi)部首先創(chuàng)建了常量 that，因?yàn)榇a可能會(huì)異步執(zhí)行，用于獲取正確的 this 對(duì)象

一開(kāi)始 Promise 的狀態(tài)應(yīng)該是 pending

value 變量用于保存 resolve 或者 reject 中傳入的值

resolvedCallbacks 和 rejectedCallbacks 用于保存 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)完善 resolve 和 reject 函數(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)大部分代碼都是根據(jù)規(guī)范去實(shí)現(xiàn)的。

我們先來(lái)改造一下 resolve 和 reject 函數(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)兼容多種 Promise 的 resolutionProcedure 函數(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)的。如果 x 為 Promise 的話，需要判斷以下幾個(gè)情況：

如果 x 處于等待態(tài)，Promise 需保持為等待態(tài)直至 x 被執(zhí)行或拒絕

如果 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)