简单手写实现React的createElement和render

今天开始正式，学习写源码系列，本系列是学习系列。

本文的目标是，手写实现createElement和render

TL;DR

• React.createElement实现的本质就是整合参数变成对象，这个对象就是react元素
• ReactDOM.render实现的本质就是根据react元素（对象）创建真实元素及其属性和子元素

科普概念

• JSX 语法 - 就是类似 html 的写法<h1>颜酱<span>最酷</span></h1>
• JSX 语法本质 - JSX 本质是语法糖，实际会被 babel 编译成React.createElement
• react 元素 - 并不是真实 DOM，实际是React.createElement函数返回的对象，结构大约如下，核心属性就是type、props、props.children

reactElement = {
  type: 'div',
  props: {
    className: 'title',
    style: { color: '#f69' },
    children: 'world',
  },
};

准备工作

1. 先创建一个 react 项目：

create-react-app react-source

1. 安装cross-env和修改命令，兼容 React 旧写法

yarn add cross-env

修改package.json的命令：

"scripts": {
  "start": "cross-env DISABLE_NEW_JSX_TRANSFORM=true react-scripts start",
  "build": "cross-env DISABLE_NEW_JSX_TRANSFORM=true react-scripts build"
},

1. 删除 src 里面所有内容，然后新建index.js

import React from 'react';
import ReactDOM from 'react-dom';

const reactElement_text = 'only text';
const reactElement_oneNode = (
  <div className="title" style={{ color: '#f69' }}>
    one node
  </div>
);
const reactElement_oneChildNode = (
  <div>
    <span>one child node</span>
  </div>
);
const reactElement_multipleNode = (
  <div>
    <span>first child</span>
    text child
  </div>
);
// only text
console.log(reactElement_text);
// { type: 'div', props: { className: 'title', style: { color: '#f69' }, children: 'one node' }, };
console.log(reactElement_oneNode);
// { type: 'div', props: { children: { type: 'span', props: { children: 'one child node' }, }, }, };
console.log(reactElement_oneChildNode);
// { type: 'div', props: { children: [ { type: 'span', props: { children: 'first child node' } }, 'text', ], }, };
console.log(reactElement_multipleNode);

ReactDOM.render(reactElement_multipleNode, document.getElementById('root'));

留心看，props.children的结构，有三种类型：字符串、对象、数组。

1. 启动项目

npm start

(^▽^)，就可以看到http://localhost:3000/了！

实现 createElement

JSX 语法相当于React.createElement，那么先实现createElement。

先看看怎么相当，在线访问：

1. 分析参数 - 输入

• 第一个参数，始终是最外层元素名
• 第二个参数，始终是最外层的元素属性集合 或 null
• 第三个参数及以后的参数，始终是子元素，没有子元素的话，就没有第三个参数
• 几个子元素，就多几个参数，每个子元素的类型是字符串或React.createElement()

1. 分析返回值 - 输出

• 对象，也叫react元素
• 属性 type 是元素名
• 属性 props 是对象，里面是元素的属性集合，props.children 可能是字符串、react元素、数组（单项还是字符串 或 react元素）

1. 新建source文件夹，建react.js。输入和输出已经明了了，接下来就是将输入变成输出。

function createElement(type, config, ...children) {
  console.log(type, config, ...children)
  // children分 0个 单个 多个
  const isNoneChildren = children.length === 0;
  const isOneChildren = children.length === 1;
  children = isNoneChildren
    ? null
    : isOneChildren
    ? children[0]
    : children;

  const res = {
    type,
    props: {
      ...config,
      children,
    },
  };
  return res;
}

const React = {
  createElement,
};
export default React;

实现 render

render说白了就是将React元素转化成真实DOM，插入到root容器中，从而自动渲染。

1. 分析参数 - 输入

• 第一个参数，可能是react元素、普通字符串、null
• 第二个参数，挂载的元素

1. 分析返回值 - 无

2. 分析执行的结果 - 显示第一个参数

render 执行之后，将第一个参数变成真实 DOM，插入到第二个参数挂载元素上，因为挂载元素本身存在于文档之中，所以挂载操作会触发渲染，显示出来。因此，render本质就是挂载

第一个参数变成真实 DOM，本质就是创建元素，增加属性，增加 children

1. 建source/react-dom.js

function render(vdom, container) {
  // 本质就是挂载，因container本身存在于文档之中，所以挂载操作会触发渲染
  mount(vdom, container);
}

function mount(vdom, container) {
  // 将第一个参数变成真实DOM，插入到第二个参数挂载元素上
  const DOM = createDOM(vdom);
  container.append(DOM);
}

function createDOM(vdom) {
  const isTextNode = typeof vdom === 'string' || vdom == null;
  if (isTextNode) return document.createTextNode(vdom || '');

  const isElementNode = typeof vdom === 'object';
  if (isElementNode) return createElementDOM(vdom);

  function createElementDOM(vdom) {
    const { type, props } = vdom;
    let DOM = document.createElement(type);
    if (props) {
      updateProps(DOM, props);
      const { children } = props;
      children && updateChildren(DOM, children);
    }
    return DOM;
  }
}

function updateProps(DOM, props) {
  console.log(DOM, props);
  // 正常遍历就好，特殊的特殊处理
  for (const key in props) {
    if (key === 'children') continue;
    if (key === 'style') {
      updateStyle(DOM, props[key]);
      continue;
    }
    DOM[key] = props[key];
  }
  function updateStyle(DOM, styleObj) {
    for (const key in styleObj) {
      DOM.style[key] = styleObj[key];
    }
  }
}

function updateChildren(DOM, children) {
  // 单个节点，直接插入（挂载）到DOM上; 多个节点，遍历插入
  const isOneChildren = !Array.isArray(children);
  isOneChildren
    ? mount(children, DOM)
    : children.forEach((child) => mount(child, DOM));
}

const ReactDOM = {
  render,
};
export default ReactDOM;

测试

将index.js里面的react和react-dom换成我们写的。

成功展示，表示✌️~