0%

【React 原理(一)】实现 createElement 和 render 方法

在 React 中,我们都知道可以写 jsx 代码会被编译成真正的 DOM 插入到要显示的页面上。这具体是怎么实现的,今天我们就自己动手做一下。

实现 createElement 方法

这个方法平时开发我们并不会用到,因为它是经 babel 编译后的代码,我们新建一个 React 项目,index.js 最简单的代码结构如下:

1
2
3
import React from 'react'
import ReactDOM from 'react-dom'
ReactDOM.render(<h1 className='title'>Hello React</h1>, document.getElementById('root'))

这里就 jsx 会变编译成真正的 DOM ,把 html 代码拿到 babel 官网编译

于是我们就看到了 React.createElement() 方法,但这只是调用这个方法,它具体做了什么返回什么我们还不知道,我们可以打印这个函数运行的结果:

1
2
3
4
5
6
7
8
9
console.log(
React.createElement(
'h1',
{
className: 'title',
},
'Hello React'
)
)


返回的这个对象就是虚拟 DOM 了。

我们来分析它返回的对象参数,首先第一个是

  • $$typeof: REACT_ELEMENT_TYPE

这个是 React 元素对象的标识属性

REACT_ELEMENT_TYPE 的值是一个 Symbol 类型,代表了一个独一无二的值。如果浏览器不支持 Symbol 类型,值就是一个二进制值。

为什么是 Symbol?主要防止 XSS 攻击伪造一个假的 React 组件。因为 JSON 中是不会存在 Symbol 类型的。

  • key:这个比如循环中会用到这个 key 值
  • props:传入的属性值,比如 id, className, style, children 等
  • ref: DOM 的引用
  • 剩下的是私有属性(本篇不展开讨论)

在本篇我们会用自己简单的方式实现这两个方法,而不是根据源码,所以实现上的方法只要能实现它的基本功能即可;有个基本概念在,以后再循序渐进学习源码。

而 createElement 中有三个参数,更确切说是 n 个参数:

  • type:表示要渲染的元素类型。这里可以传入一个元素 Tag 名称,也可以传入一个组件(如 div span 等,也可以是是函数组件和类组件)
  • props:创建 React 元素所需要的 props。
  • childrens(可选参数):要渲染元素的子元素,这里可以向后传入 n 个参数。可以为文本字符串,也可以为数组

初步 createElement 方法:

1
2
3
4
5
6
7
8
9
// 创建 JSX 对象
function createElement(type, props, ...childrens) {
return {
type,
props: {
...props,
children: childrens.length <= 1 ? childrens[0] || '' : childrens,
},
}

参数中 props 和 childrens 是并列关系,然后返回的 props 对象,里面包含了 children,所以我们需要再 props 里面添加 children 参数,然后根据 children 参数为一个或多个的可能在进行取值处理。

调用该方法:

1
2
3
4
5
6
7
8
9
console.log(
createElement(
'h1',
{
className: 'title',
},
'Hello React'
)
)


除去其它本篇我们不讨论的属性,目前算是实现了一半;我们观察原来 React 自身方法输出的结果有 key, ref, 同输出的 props 也是并列关系,于是我们进一步作出处理

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
function createElement(type, props, ...childrens) {
let ref, key
if ('ref' in props) {
ref = props['ref']
props['ref'] = undefined
}
if ('key' in props) {
key = props['key']
props['key'] = undefined
}
return {
type,
props: {
...props,
children: childrens.length <= 1 ? childrens[0] || '' : childrens,
},
ref,
key,
}
}

同样的方式调用结果如下:


如果添加多一些属性,我们来看看结果

1
2
3
4
5
6
7
8
9
console.log(
createElement(
'div',
{ id: 'box', className: 'box', style: { color: 'red' }, key: '20' },
'this is text',
createElement('h2', { className: 'title' }, 'hello'),
createElement('div', { className: 'content' }, 'Hi')
)
)


用了这种比较粗鲁的方式添加,设置为 undefined 在实现 render 方法的时候我们会根据这个忽略 props 内部的 key 和 props 属性,这里就实现了最基本的 createElement 方法了。

实现 render 方法

render 方法的第一个参数接收的是 createElement 返回的对象,也就是虚拟 DOM;
第二个参数则是挂载的目标 DOM。同样的做法,我们用 babel 编译来看:


执行后,就被挂在到页面了

实现代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
/*
* 功能:把创建的对象生成对应的DOM元素,最后插入到页面中
* objJSX: createElement 返回的 JSX 对象
* container:挂载的容器,如 document.getElementById('root')
*/
function render(objJSX, container) {
let { type, props } = objJSX
let newElement = document.createElement(type)
for (let attr in props) {
// 遍历传入的 props 属性
if (!props.hasOwnProperty(attr)) break // 不是私有的直接结束遍历
let value = props[attr] // >如果当前属性没有值,直接不处理即可
if (value == undefined) continue // NULL OR UNDEFINED

// 对几个特殊属性单独设置
switch (attr.toUpperCase()) {
case 'ID':
newElement.setAttribute('id', value)
break
case 'CLASSNAME':
newElement.setAttribute('class', value)
break
case 'STYLE': // 传入的行内样式 style 是个对象,故需遍历赋值
for (let styleAttr in value) {
if (value.hasOwnProperty(styleAttr)) {
newElement['style'][styleAttr] = value[styleAttr]
}
}
break
case 'CHILDREN':
/*
* 可能是一个值:可能是字符串也可能是一个JSX对象
* 可能是一个数组:数组中的每一项可能是字符串也可能是JSX对象
*/
// 首先把一个值也变为数组,这样后期统一操作数组即可
!(value instanceof Array) ? (value = [value]) : null
value.forEach((item, index) => {
// 验证ITEM是什么类型的:如果是字符串就是创建文本节点,如果是对象,我们需要再次执行RENDER方法,把创建的元素放到最开始创建的大盒子中
if (typeof item === 'string') {
let text = document.createTextNode(item)
newElement.appendChild(text)
} else {
render(item, newElement)
}
})
break
default:
newElement.setAttribute(attr, value)
}
}
container.appendChild(newElement)
}

-------------本文结束感谢您的阅读-------------