React的createElement和render手写实现示例

2024-04-02 19:04:59 382人浏览薄情痞子

摘要

目录TL;DR科普概念准备工作实现 createElement实现 render测试TL;DR 本文的目标是，手写实现createElement和render React.creat

TL;DR

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

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',
  },
};

准备工作

先创建一个 react 项目：

create-react-app react-source

安装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"
},

删除 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的结构，有三种类型：字符串、对象、数组。

启动项目

npm start

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

实现 createElement

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

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

分析参数 - 输入

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

分析返回值 - 输出

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

新建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容器中，从而自动渲染。

分析参数 - 输入

第一个参数，可能是react元素、普通字符串、null
第二个参数，挂载的元素
分析返回值 - 无
分析执行的结果 - 显示第一个参数

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

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

建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;