首页 > 网络编程 > JavaScript > javascript类库 > React > React createElement

React createElement方法使用原理分析介绍

2022-09-01 14:27:45 作者：搞前端的小菜

这篇文章主要为大家介绍了React的createElement方法源码解析，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪

摘要

在上一篇说过，React创建元素有两种方式：

第一种是通过JSX方式创建，第二种是通过React.createElement方法创建。但是通过JSX创建React元素的方式，最终也会经过babel进行转换，再用React.createElement进行元素创建。

而这一篇文章，主要是讲一下React.createElement是如何创建React元素的。

1.创建方法

为了了解React.createElement这个方法，我们自己手动去实现一个简易版的。

OK，首先我们通过React.createElement方法创建一个React元素，并且打印出来：

    const b = React.createElement('div',{data: '1234'},'oneTwo')
    console.log(b);

打印出来的结果为：

所以，如果我们要实现出来React.createElement这个方法，定义的时候我们可以这么写：

function createElement(type, config, children){
  let props = {}
  let myType = ''
  let key = ''
  let ref = ''
  return {
    $$typeOf: 'react.element',
    type: myType,
    props,
    key,
    ref
  }
}

这里我们不管**$$typeOf**这个变量，只考虑其他的变量，最终返回的对象应该是这个数据结构。

2.处理type

好了，方法的定义已经写完了。针对于传进来的这三个参数，我们一个一个看。

第一个参数type，它代表的是创建React元素的类型。

这里面可以是传统的HTML元素，例如div，h1，span等标签。

也可以是一个React组件（注意这里哦）。

而React创建组件又有两种方式，所以type的值，有三种数据类型：

（1）字符串：例如"div",“h1”,“span”,"p"等标签

（2）函数：通过函数的方式创建React组件

（3）类：通过class的方式创建React组件

而这个时候就有一个问题！

class Demo {
}
typeof Demo

上面的值应该为什么呢？答案是function，所以在这里我们只需要考虑type为string和function两种类型即可。

所以我们可以写一个判断类型的方法：

function isValidElementType(type){
  if(typeof type === 'string' || typeof type === 'function'){
    return true;
  }
  return false;
}

在我们的主方法里面引用：

function createElement(type, config, children){
  let props = {}
  let myType = ''
  let key = ''
  let ref = ''
  if(isValidElementType(type)){
    myType = type;
  }
  return {
    $$typeOf: 'react.element',
    type: myType,
    props,
    key,
    ref
  }
}

3.处理config

对于React.createElement方法的第二个参数，接收看一个对象。而对象下所有的属性，都会被放在props里面。

这句话对不对呢？其实也不是很对，是有特例的，如果在这个对象下，有key或者ref这两个属性。它是不会被放在props里面的，而是直接赋值给key和ref

Ok，有了上面的话，我们就可以对config进行处理了：

  if(config != null){
    if(config.ref){
      ref = config.ref
    }
    if(config.key){
      key = config.key
    }
    for(let propName in config){
      if(!(['ref','key'].includes(propName))){
        props[propName] = config[propName]
      }
    }
  }

我们只需要把config的key和ref分别给到我们返回对象里面的key和ref。再便利一下config，拿出来的属性和值把key和ref排除。最终我们的config属性就处理好了。

4.处理children

最后一步就是处理children属性了。而React.createElement方法的第三个参数，也可以是第四个参数（就是后面的所有参数）。都可以为字符串，或者是React元素。

这里的React元素我们不管它是通过JSX创建的，还是通过React.createElement方法创建的都可以

而参数的情况分两种：

第一种是只有三个参数，也就是children为一个值。这个时候props里面的children就是该字符串。

第二种是参数大于三个，这个时候，props里面的children是一个数组，数组里的元素就是后面的所有参数。

OK，有了上面的基础，就可以对children进行处理了：

  let childrenLength = arguments.length - 2
  if(childrenLength === 1){
    props.children = children
  }else if(childrenLength > 1){
    let children = new Array(childrenLength)
    for(let i = 0;i<childrenLength;i++){
      children[i] = arguments[i+2]
    }
    props.children = children
  }

这里通过arguments来判断参数的个数，进而确定分支条件。

然后再根据情况，确定props里的children。

最后再贴上完整的createElement方法（简易版）：

function createElement(type, config, children){
  let props = {}
  let myType = ''
  let key = ''
  let ref = ''
  if(isValidElementType(type)){
    myType = type;
  }
  if(config != null){
    if(config.ref){
      ref = config.ref
    }
    if(config.key){
      key = config.key
    }
    for(let propName in config){
      if(!(['ref','key'].includes(propName))){
        props[propName] = config[propName]
      }
    }
  }
  let childrenLength = arguments.length - 2
  if(childrenLength === 1){
    props.children = children
  }else if(childrenLength > 1){
    let children = new Array(childrenLength)
    for(let i = 0;i<childrenLength;i++){
      children[i] = arguments[i+2]
    }
    props.children = children
  }
  return {
    $$typeOf: 'react.element',
    type: myType,
    props,
    key,
    ref
  }
}

5.对比真正的React.createElement源码

OK，上面只是实现了一个比较简单的React.createElement方法，但是懂了其中的过程，我们就可以看一下真正的React.createElement源码：

isValidElementType方法

function isValidElementType(type) {
  if (typeof type === 'string' || typeof type === 'function') {
    return true;
  } // Note: typeof might be other than 'symbol' or 'number' (e.g. if it's a polyfill).
  if (type === REACT_FRAGMENT_TYPE || type === REACT_PROFILER_TYPE || enableDebugTracing  || type === REACT_STRICT_MODE_TYPE || type === REACT_SUSPENSE_TYPE || type === REACT_SUSPENSE_LIST_TYPE || enableLegacyHidden  || type === REACT_OFFSCREEN_TYPE || enableScopeAPI  || enableCacheElement  || enableTransitionTracing ) {
    return true;
  }
  if (typeof type === 'object' && type !== null) {
    if (type.$$typeof === REACT_LAZY_TYPE || type.$$typeof === REACT_MEMO_TYPE || type.$$typeof === REACT_PROVIDER_TYPE || type.$$typeof === REACT_CONTEXT_TYPE || type.$$typeof === REACT_FORWARD_REF_TYPE || // This needs to include all possible module reference object
    // types supported by any Flight configuration anywhere since
    // we don't know which Flight build this will end up being used
    // with.
    type.$$typeof === REACT_MODULE_REFERENCE || type.getModuleId !== undefined) {
      return true;
    }
  }
  return false;
}

这里面也主要就是判断type的类型，不过判断情况多了几种React自带的元素类型。

createElement方法

function createElement(type, config, children) {
  var propName; // Reserved names are extracted
  var props = {};
  var key = null;
  var ref = null;
  var self = null;
  var source = null;
  if (config != null) {
    if (hasValidRef(config)) {
      ref = config.ref;
      {
        warnIfStringRefCannotBeAutoConverted(config);
      }
    }
    if (hasValidKey(config)) {
      {
        checkKeyStringCoercion(config.key);
      }
      key = '' + config.key;
    }
    self = config.__self === undefined ? null : config.__self;
    source = config.__source === undefined ? null : config.__source; // Remaining properties are added to a new props object
    for (propName in config) {
      if (hasOwnProperty.call(config, propName) && !RESERVED_PROPS.hasOwnProperty(propName)) {
        props[propName] = config[propName];
      }
    }
  } // Children can be more than one argument, and those are transferred onto
  // the newly allocated props object.
  var childrenLength = arguments.length - 2;
  if (childrenLength === 1) {
    props.children = children;
  } else if (childrenLength > 1) {
    var childArray = Array(childrenLength);
    for (var i = 0; i < childrenLength; i++) {
      childArray[i] = arguments[i + 2];
    }

    {
      if (Object.freeze) {
        Object.freeze(childArray);
      }
    }
    props.children = childArray;
  } // Resolve default props
  if (type && type.defaultProps) {
    var defaultProps = type.defaultProps;

    for (propName in defaultProps) {
      if (props[propName] === undefined) {
        props[propName] = defaultProps[propName];
      }
    }
  }
  {
    if (key || ref) {
      var displayName = typeof type === 'function' ? type.displayName || type.name || 'Unknown' : type;
      if (key) {
        defineKeyPropWarningGetter(props, displayName);
      }
      if (ref) {
        defineRefPropWarningGetter(props, displayName);
      }
    }
  }
  return ReactElement(type, key, ref, self, source, ReactCurrentOwner.current, props);
};

这个方法主要是对config和children进行处理。

其余的部分就不粘过来了，对源码感兴趣的可以自己打断点尝试一哈！

到此这篇关于React createElement方法使用原理分析介绍的文章就介绍到这了,更多相关React createElement内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！