React源码学习入门（八）React组件挂载Component细节流程

孟健

发布于 2022-12-19 17:08:20

6120

发布于 2022-12-19 17:08:20

文章被收录于专栏：前端工程

React组件挂载细节流程

本文基于React v15.6.2版本介绍，原因请参见新手如何学习React源码

源码分析

在上一篇文章的最后，我们走到了mountComponentIntoNode，它通过调用ReactReconciler.mountComponent来获取Markup，这个也是React执行挂载的核心入口，源码位于src/renderers/shared/stack/reconciler/ReactReconciler.js：

  mountComponent: function(
    internalInstance,
    transaction,
    hostParent,
    hostContainerInfo,
    context,
    parentDebugID, // 0 in production and for roots
  ) {
    var markup = internalInstance.mountComponent(
      transaction,
      hostParent,
      hostContainerInfo,
      context,
      parentDebugID,
    );
    if (
      internalInstance._currentElement &&
      internalInstance._currentElement.ref != null
    ) {
      transaction.getReactMountReady().enqueue(attachRefs, internalInstance);
    }
    return markup;
  },

可以看到这个方法的核心只是调用instance的mountComponent方法，这个instance是我们一开始进来的wrapperInstance，也就是React自己包裹的那一层组件，它是一个CompositeComponent，注意这里的参数顺序换了，实际上其他参数没有变，hostParent在我们初次挂载的时候为null。

一般来说，真正去产生markup的是真实的DOM节点，而不是抽象的React组件，顺序一般是ReactCompositeComponent -> ReactDOMComponent -> ReactDOMTextComponent/ReactEmptyComponent，所以一开始去看一下ReactCompositeComponent是明智的选择。

源码位于src/renderers/shared/stack/reconciler/ReactCompositeComponent.js：

  mountComponent: function(
    transaction,
    hostParent,
    hostContainerInfo,
    context,
  ) {
    this._context = context;
    this._mountOrder = nextMountID++;
    this._hostParent = hostParent;
    this._hostContainerInfo = hostContainerInfo;

    var publicProps = this._currentElement.props;
    var publicContext = this._processContext(context);

    var Component = this._currentElement.type;

    var updateQueue = transaction.getUpdateQueue();

    // 1. 实例化组件
    var doConstruct = shouldConstruct(Component);
    var inst = this._constructComponent(
      doConstruct,
      publicProps,
      publicContext,
      updateQueue,
    );
    var renderedElement;

    // Support functional components
    if (!doConstruct && (inst == null || inst.render == null)) {
      renderedElement = inst;
      warnIfInvalidElement(Component, renderedElement);
      inst = new StatelessComponent(Component);
      this._compositeType = CompositeTypes.StatelessFunctional;
    } else {
      if (isPureComponent(Component)) {
        this._compositeType = CompositeTypes.PureClass;
      } else {
        this._compositeType = CompositeTypes.ImpureClass;
      }
    }

    inst.props = publicProps;
    inst.context = publicContext;
    inst.refs = emptyObject;
    inst.updater = updateQueue;

    this._instance = inst;

    ReactInstanceMap.set(inst, this);

    var initialState = inst.state;
    if (initialState === undefined) {
      inst.state = initialState = null;
    }

    this._pendingStateQueue = null;
    this._pendingReplaceState = false;
    this._pendingForceUpdate = false;

    var markup;
    if (inst.unstable_handleError) {
      markup = this.performInitialMountWithErrorHandling(
        renderedElement,
        hostParent,
        hostContainerInfo,
        transaction,
        context,
      );
    } else {
      // 2. 执行挂载
      markup = this.performInitialMount(
        renderedElement,
        hostParent,
        hostContainerInfo,
        transaction,
        context,
      );
    }

    // 3. 处理`componentDidMount`回调
    if (inst.componentDidMount) {
        transaction.getReactMountReady().enqueue(inst.componentDidMount, inst);
    }

    return markup;
  },

对于React组件的mountComponent过程，主要做了3件事：

实例化React组件本身
执行performInitialMount，获得Markup
处理componentDidMount回调

其中第二步处理是最复杂的，我们先来看看1和3：

实例化React组件

首先我们先拿到React组件本身：

var Component = this._currentElement.type;

这里的_currentElement，其实就是我们JSX所调用的createElement的返回值，这一点之前已经详细介绍过，而对于React组件来说，它的整个class都是挂载type这个字段下的，所以这里通过type就可以拿到React组件的类。

接着有一个shouldConstruct的判断：

var doConstruct = shouldConstruct(Component);

function shouldConstruct(Component) {
  return !!(Component.prototype && Component.prototype.isReactComponent);
}

我们知道，React组件有两种模式：类组件和函数组件。

对于类组件的写法之前已经分析过，它会在prototype上挂一个isReactComponent属性，所以这里的判断是用来区分类组件和函数组件的。

紧接着开始实例化，实例化过程最终会走到这个函数：

_constructComponentWithoutOwner: function(
  doConstruct,
  publicProps,
  publicContext,
  updateQueue,
) {
  var Component = this._currentElement.type;

  if (doConstruct) {
      return new Component(publicProps, publicContext, updateQueue);
  }

  return Component(publicProps, publicContext, updateQueue);
},

可以看到这里函数组件就直接执行函数了，所以这里拿到的实例有两种情况：

类组件，这里拿到的是instance
函数组件，这里拿到的其实是element

所以有下面这段代码来修正一下函数组件的instance：

if (!doConstruct && (inst == null || inst.render == null)) {
  renderedElement = inst;
  warnIfInvalidElement(Component, renderedElement);
  inst = new StatelessComponent(Component);
  this._compositeType = CompositeTypes.StatelessFunctional;
} else {
  if (isPureComponent(Component)) {
    this._compositeType = CompositeTypes.PureClass;
  } else {
    this._compositeType = CompositeTypes.ImpureClass;
  }
}

这里对于函数组件，会实例化一个StatelessComponent的实例出来：

function StatelessComponent(Component) {}
StatelessComponent.prototype.render = function() {
  var Component = ReactInstanceMap.get(this)._currentElement.type;
  var element = Component(this.props, this.context, this.updater);
  warnIfInvalidElement(Component, element);
  return element;
};

这个实例只挂载了一个render方法，用来执行函数组件的渲染。这个render方法会在后续更新流程中被统一调用。

处理`componentDidMount`回调

transaction.getReactMountReady().enqueue(inst.componentDidMount, inst);

这里其实之前看过了，队列会在transaction周期的末尾被触发，而上个函数本身是个递归的过程，根据队列的特性，这里叶子节点的componentDidMount就会被先触发。

执行`performInitialMount`拿到Markup

performInitialMount: function(
  renderedElement,
  hostParent,
  hostContainerInfo,
  transaction,
  context,
) {
  var inst = this._instance;

  // 1. 执行componentWillMount钩子
  if (inst.componentWillMount) {
      inst.componentWillMount();
      inst.state = this._processPendingState(inst.props, inst.context);
  }

  // If not a stateless component, we now render
  if (renderedElement === undefined) {
    renderedElement = this._renderValidatedComponent();
  }

  // 2. 初始化child实例
  var nodeType = ReactNodeTypes.getType(renderedElement);
  this._renderedNodeType = nodeType;
  var child = this._instantiateReactComponent(
    renderedElement,
    nodeType !== ReactNodeTypes.EMPTY /* shouldHaveDebugID */,
  );
  this._renderedComponent = child;
  
  // 3. 挂载child
  var markup = ReactReconciler.mountComponent(
    child,
    transaction,
    hostParent,
    hostContainerInfo,
    this._processChildContext(context),
    debugID,
  );

  return markup;
},

在这个performInitialMount核心实现中，主要做了3件事：