React16源码: React中的setState和forceUpdate源码实现

setState 和 forceUpdate

1 ）概述

通过 class component 内部的 setState，以及 forceUpdate 去更新一个组件的过程
在react的应用当中，我们只有 ReactDOM.render setState，以及 forceUpdate
这几种种方式去更新react的应用是合理的，其他没有什么特别常用的方式去更新了
而且react官方推荐的也是用这些。forceUpdate 其实本身也不是特别推荐的
因为有比较特殊的作用，以及比较特殊的使用场合, 也算是一个比较有用的API
在React16.7之后发布的Hooks, Hooks里面我们通过
useState 返回过来的一个方法，也可以去更新一个组件的状态
- 这块看不到源码，只能看到打包之后的代码
- 它的设置比较特殊，这块先跳过
setState，以及 forceUpdate的核心，首先是他们是给节点 Fiber 去创建更新的
对于 ReactDOM.render 来说，它创建的 update 是放在root上面
它是一个整体的渲染，因为执行 ReactDOM.render 的时候
整个应用都没有任何的产生，任何的节点都没有生成，是一个初始的渲染
所以 setState 和 forceUpdate 是针对某一个 class component 来说的
- 这个 class component 的状态来进行更新
- 把新的状态计算并渲染出来
setState 和 forceUpdate 两者更新的类型不同
- 它们上面有 UpdateState, ForceUpdate 两种不同的 tag
- 这个tag就对应的它们是更新的类型不同

2 ）源码

先跳过组件如何去渲染
- 也就是我们的 class component，什么时候会被实例化
- 这个涉及到后面的更新的流程
我们的component的BaseClass在初始化的时候会拿到一个update的这个对象
这个对象来自于哪个地方
源码: https://github.com/facebook/react/blob/v16.6.0/packages/react-reconciler/src/ReactFiberClassComponent.js

这个文件中，有涉及很多 class 相关的代码

const classComponentUpdater = {
  isMounted,
  enqueueSetState(inst, payload, callback) {
    const fiber = ReactInstanceMap.get(inst); // inst 是调用 this.setState 时的 this，在react渲染的时候会通过 `ReactInstanceMap` 做映射和获取 Fiber对象
    const currentTime = requestCurrentTime();
    const expirationTime = computeExpirationForFiber(currentTime, fiber);

    const update = createUpdate(expirationTime);
    update.payload = payload; // 这里 payload 是 setState时传入的对象
    if (callback !== undefined && callback !== null) {
      if (__DEV__) {
        warnOnInvalidCallback(callback, 'setState');
      }
      update.callback = callback; // 挂载 callback
    }

    enqueueUpdate(fiber, update); // 把fiber对象的updateQueue进行初始化或更新
    scheduleWork(fiber, expirationTime); // 进行调度处理
  },
  enqueueReplaceState(inst, payload, callback) {
    const fiber = ReactInstanceMap.get(inst);
    const currentTime = requestCurrentTime();
    const expirationTime = computeExpirationForFiber(currentTime, fiber);

    const update = createUpdate(expirationTime);
    update.tag = ReplaceState;
    update.payload = payload;

    if (callback !== undefined && callback !== null) {
      if (__DEV__) {
        warnOnInvalidCallback(callback, 'replaceState');
      }
      update.callback = callback;
    }

    enqueueUpdate(fiber, update);
    scheduleWork(fiber, expirationTime);
  },
  enqueueForceUpdate(inst, callback) {
    const fiber = ReactInstanceMap.get(inst);
    const currentTime = requestCurrentTime();
    const expirationTime = computeExpirationForFiber(currentTime, fiber);

    const update = createUpdate(expirationTime);
    update.tag = ForceUpdate;

    if (callback !== undefined && callback !== null) {
      if (__DEV__) {
        warnOnInvalidCallback(callback, 'forceUpdate');
      }
      update.callback = callback;
    }

    enqueueUpdate(fiber, update);
    scheduleWork(fiber, expirationTime);
  },
};

这里面主要关注, enqueueSetState 和 enqueueForceUpdate 这两个enqueue 方法几乎一样，两者区别看 update.tag
看了上面的源码，发现和 react-reconciler/src/ReactFiberReconciler.js 中的 updateContainer 很像
- 内部调用 updateContainerAtExpirationTime 接着调用 scheduleRootUpdate 在这里面都基本相似