Promise源码实现

/* eslint-disable no-param-reassign */
// Promise 三种状态
const PENDING = 'pending';
const FULFILLED = 'fulfiled';
const REJECTED = 'rejected';

// 默认的成功回调和失败回调
const DEFAULT_SUCCESS_CALLBACK = (value) => value;
const DEFAULT_FAIL_CALLBACK = (reason) => reason;

class MyPromise {
  constructor(executor) {
    this._status = PENDING;
    this._successCallbackArray = [];
    this._failCallbackArray = [];
    // 此处需传入this,只传入this._resolve会在Promise真正执行时找不到this指向
    executor(this._resolve.bind(this), this._reject.bind(this));
  }

  // 私有resolve方法,仅在MyPromise内部使用
  _resolve(value) {
    // 状态一经改变,则不会向下执行
    if (this._status !== PENDING) {
      return;
    }
    this._status = FULFILLED;
    this._value = value;
    // 如果事先被保存,则依次执行
    while (this._successCallbackArray.length) {
      this._successCallbackArray.shift()(this._value);
    }
  }

  // 私有reject方法,仅在MyPromise内部使用
  _reject(reason) {
    if (this._status !== PENDING) {
      return;
    }
    this._status = REJECTED;
    this._reason = reason;
    while (this._failCallbackArray.length) {
      this._failCallbackArray.shift()(this._reason);
    }
  }

  // 处理then方法返回的Promise的结果
  static _parseResult(promsie, result, resolve, reject) {
    if (promsie === result) {
      reject(new TypeError('Chaining cycle detected for promise #<Promise>'));
    } else if (result instanceof MyPromise) {
      result.then(resolve, reject);
    } else {
      resolve(result);
    }
  }

  then(successCallback, failCallback) {
    // 如果传入的参数不是函数,则使用默认的callback
    if (typeof successCallback !== 'function') {
      successCallback = DEFAULT_SUCCESS_CALLBACK;
    }
    if (typeof failCallback !== 'function') {
      failCallback = DEFAULT_FAIL_CALLBACK;
    }
    const thenPromise = new MyPromise((resolve, reject) => {
      if (this._status === FULFILLED) {
        // 如果Promise已成功
        // 使用setTimeout模拟异步效果,下同
        setTimeout(() => {
          try {
            // 获得callback的执行结果,并把传递给后边的then方法,下同
            const result = successCallback(this._value);
            MyPromise._parseResult(thenPromise, result, resolve, reject);
          } catch (e) {
            reject(e);
          }
        }, 0);
      } else if (this._status === REJECTED) {
        // 如果Promise已成功
        setTimeout(() => {
          try {
            const result = failCallback(this._reason);
            MyPromise._parseResult(thenPromise, result, resolve, reject);
          } catch (e) {
            reject(e);
          }
        }, 0);
      } else {
        // 如果Promise仍未等待状态,保存then的成功和失败回调到数组中
        this._successCallbackArray.push(() => {
          setTimeout(() => {
            try {
              const result = successCallback(this._value);
              MyPromise._parseResult(thenPromise, result, resolve, reject);
            } catch (e) {
              reject(e);
            }
          }, 0);
        });
        this._failCallbackArray.push(() => {
          setTimeout(() => {
            try {
              const result = failCallback(this._reason);
              MyPromise._parseResult(thenPromise, result, resolve, reject);
            } catch (e) {
              reject(e);
            }
          }, 0);
        });
      }
    });

    return thenPromise;
  }

  // 捕获错误,就是增加一个reject的回调来接收错误,并且继续返回Promise
  catch(failCallback) {
    return this.then(undefined, failCallback);
  }

  // 无论如何都会执行,在成功和失败的回调中调用,并且把值向下传递
  finally(callback) {
    const final = (resolve, reject) => {
      const result = callback();
      if (result instanceof MyPromise) {
        result.then(resolve, reject);
      } else {
        resolve(result);
      }
    };
    return new MyPromise((resolve, reject) => {
      this.then(() => {
        final(resolve, reject);
      }, () => {
        final(resolve, reject);
      });
    });
    // return this.then(
    //   (value) => MyPromise.resolve(callback()).then(() => value),
    //   (reason) => MyPromise.resolve(callback()).then(() => { throw reason; }),
    // );
  }

  static resolve(value) {
    if (value instanceof MyPromise) {
      return value;
    }
    return new MyPromise((res) => {
      res(value);
    });
  }

  static reject(value) {
    if (value instanceof MyPromise) {
      return value;
    }
    return new MyPromise((_, rej) => {
      rej(value);
    });
  }

  // 数组内的元素类型可以为Promise或普通值
  static all(array) {
    const result = [];
    let index = 0;
    return new MyPromise((resolve, reject) => {
      // 记录结果的方法
      function addData(key, value) {
        result[key] = value;
        index += 1;
        if (index === array.length) {
          // 结果全部记录完毕后,将结果数组传递出去
          resolve(result);
        }
      }
      for (let i = 0; i < array.length; i += 1) {
        const current = array[i];
        if (current instanceof MyPromise) {
          // 如果值为Promise,则将结果记录下来。一旦失败,整个返回结果失败
          current.then((value) => addData(i, value), (reason) => reject(reason));
        } else {
          // 如果为普通值,则将这个值作为结果记录下来
          addData(i, current);
        }
      }
    });
  }

  // 任何一个resolve或reject,则直接返回结果。普通值则直接resolve
  static race(array) {
    return new MyPromise((resolve, reject) => {
      for (let i = 0; i < array.length; i += 1) {
        const current = array[i];
        if (current instanceof MyPromise) {
          current.then((value) => resolve(value), (value) => reject(value));
        } else {
          resolve(current);
        }
      }
    });
  }
}

export default MyPromise;

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 211,743评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,296评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,285评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,485评论 1 283
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,581评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,821评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,960评论 3 408
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,719评论 0 266
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,186评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,516评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,650评论 1 340
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,329评论 4 330
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,936评论 3 313
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,757评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,991评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,370评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,527评论 2 349