从零实现一个React(二):组件和生命周期

React是前端最受欢迎的框架之一,解读其源码的文章非常多,但是我想从另一个角度去解读React:从零开始实现一个React,从API层面实现React的大部分功能,在这个过程中去探索为什么有虚拟DOM、diff、为什么setState这样设计等问题。 整个系列大概会有四篇左右,有问题需要探讨也请在github上回复我~

https://github.com/hujiulong/blog


前言

在上一篇文章JSX和虚拟DOM中,我们实现了基础的JSX渲染功能,但是React的意义在于组件化。在这篇文章中,我们就要实现React的组件功能。

组件

React定义组件的方式可以分为两种:函数和类,函数定义可以看做是类定义的一种简单形式。

createElement的变化

回顾一下上一篇文章中我们对React.createElement的实现:

function createElement( tag, attrs, ...children ) {
return {
tag,
attrs,
children
}
}

这种实现我们前面暂时只用来渲染原生DOM元素,而对于组件,createElement得到的参数略有不同:
如果JSX片段中的某个元素是组件,那么createElement的第一个参数tag将会是一个方法,而不是字符串。

区分组件和原生DOM的工作,是babel-plugin-transform-react-jsx帮我们做的

例如在处理<Welcome name="Sara" />时,createElement方法的第一个参数tag,实际上就是我们定义Welcome的方法:

function Welcome( props ) {
return <h1>Hello, {props.name}</h1>;
}

我们不需要对createElement做修改,只需要知道如果渲染的是组件,tag的值将是一个函数

组件基类React.Component

通过类的方式定义组件,我们需要继承React.Component

class Welcome extends React.Component {
render() {
return <h1>Hello, {this.props.name}</h1>;
}
}

所以我们就需要先来实现React.Component这个类:

Component

React.Component包含了一些预先定义好的变量和方法,我们来一步一步地实现它:
先定义一个Component类:

class Component {}

state & props

通过继承React.Component定义的组件有自己的私有状态state,可以通过this.state获取到。同时也能通过this.props来获取传入的数据。
所以在构造函数中,我们需要初始化stateprops

// React.Component
class Component {
constructor( props = {} ) {
this.state = {};
this.props = props;
}
}

setState

组件内部的state和渲染结果相关,当state改变时通常会触发渲染,为了让React知道我们改变了state,我们只能通过setState方法去修改数据。我们可以通过Object.assign来做一个简单的实现。
在每次更新state后,我们需要调用renderComponent方法来重新渲染组件,renderComponent方法的实现后文会讲到。

import { renderComponent } from '../react-dom/render'
class Component {
constructor( props = {} ) {
// ...
}

setState( stateChange ) {
// 将修改合并到state
Object.assign( this.state, stateChange );
renderComponent( this );
}
}

你可能听说过React的setState是异步的,同时它有很多优化手段,这里我们暂时不去管它,在以后会有一篇文章专门来讲setState方法。

render

上一篇文章中实现的render方法只支持渲染原生DOM元素,我们需要修改ReactDOM.render方法,让其支持渲染组件。
修改之前我们先来回顾一下上一篇文章中我们对ReactDOM.render的实现:

function render( vnode, container ) {
return container.appendChild( _render( vnode ) );
}

function _render( vnode ) {

if ( vnode === undefined || vnode === null || typeof vnode === 'boolean' ) vnode = '';

if ( typeof vnode === 'number' ) vnode = String( vnode );

if ( typeof vnode === 'string' ) {
let textNode = document.createTextNode( vnode );
return textNode;
}

const dom = document.createElement( vnode.tag );

if ( vnode.attrs ) {
Object.keys( vnode.attrs ).forEach( key => {
const value = vnode.attrs[ key ];
setAttribute( dom, key, value );
} );
}

vnode.children.forEach( child => render( child, dom ) ); // 递归渲染子节点

return dom;
}

我们需要在其中加一段用来渲染组件的代码:

function _render( vnode ) {

// ...

if ( typeof vnode.tag === 'function' ) {

const component = createComponent( vnode.tag, vnode.attrs );

setComponentProps( component, vnode.attrs );

return component.base;
}

// ...
}

组件渲染和生命周期

在上面的方法中用到了createComponentsetComponentProps两个方法,组件的生命周期方法也会在这里面实现。

生命周期方法是一些在特殊时机执行的函数,例如componentDidMount方法会在组件挂载后执行

createComponent方法用来创建组件实例,并且将函数定义组件扩展为类定义组件进行处理,以免其他地方需要区分不同定义方式。

// 创建组件
function createComponent( component, props ) {

let inst;
// 如果是类定义组件,则直接返回实例
if ( component.prototype && component.prototype.render ) {
inst = new component( props );
// 如果是函数定义组件,则将其扩展为类定义组件
} else {
inst = new Component( props );
inst.constructor = component;
inst.render = function() {
return this.constructor( props );
}
}

return inst;
}

setComponentProps方法用来更新props,在其中可以实现componentWillMountcomponentWillReceiveProps两个生命周期方法

// set props
function setComponentProps( component, props ) {

if ( !component.base ) {
if ( component.componentWillMount ) component.componentWillMount();
} else if ( component.componentWillReceiveProps ) {
component.componentWillReceiveProps( props );
}

component.props = props;

renderComponent( component );

}

renderComponent方法用来渲染组件,setState方法中会直接调用这个方法进行重新渲染,在这个方法里可以实现componentWillUpdatecomponentDidUpdatecomponentDidMount几个生命周期方法。

export function renderComponent( component ) {

let base;

const renderer = component.render();

if ( component.base && component.componentWillUpdate ) {
component.componentWillUpdate();
}

base = _render( renderer );

if ( component.base ) {
if ( component.componentDidUpdate ) component.componentDidUpdate();
} else if ( component.componentDidMount ) {
component.componentDidMount();
}

if ( component.base && component.base.parentNode ) {
component.base.parentNode.replaceChild( base, component.base );
}

component.base = base;
base._component = component;

}

渲染组件

现在大部分工作已经完成,我们可以用它来渲染组件了。

渲染函数定义组件

渲染前文提到的Welcome组件:

const element = <Welcome name="Sara" />;
ReactDOM.render(
element,
document.getElementById( 'root' )
);

在浏览器中可以看到结果:

640?wx_fmt=png

试试更复杂的例子,将多个组件组合起来:

function App() {
return (
<div>
<Welcome name="Sara" />
<Welcome name="Cahal" />
<Welcome name="Edite" />
</div>
);
}
ReactDOM.render(
<App />,
document.getElementById( 'root' )
);

在浏览器中可以看到结果:

640?wx_fmt=png

渲染类定义组件

我们来试一试将刚才函数定义组件改成类定义:

class Welcome extends React.Component {
render() {
return <h1>Hello, {this.props.name}</h1>;
}
}

class App extends React.Component {
render() {
return (
<div>
<Welcome name="Sara" />
<Welcome name="Cahal" />
<Welcome name="Edite" />
</div>
);
}
}
ReactDOM.render(
<App />,
document.getElementById( 'root' )
);

运行起来结果和函数定义组件完全一致:

640?wx_fmt=png

再来尝试一个能体现出类定义组件区别的例子,实现一个计数器Counter,每点击一次就会加1。
并且组件中还增加了两个生命周期函数:

class Counter extends React.Component {
constructor( props ) {
super( props );
this.state = {
num: 0
}
}

componentWillUpdate() {
console.log( 'update' );
}

componentWillMount() {
console.log( 'mount' );
}

onClick() {
this.setState( { num: this.state.num + 1 } );
}

render() {
return (
<div onClick={ () => this.onClick() }>
<h1>number: {this.state.num}</h1>
<button>add</button>
</div>
);
}
}

ReactDOM.render(
<Counter />,
document.getElementById( 'root' )
);

可以看到结果:

640?wx_fmt=gif

mount只在挂载时输出了一次,后面每次更新时会输出update

后话

至此我们已经从API层面实现了React的核心功能。但是我们目前的做法是每次更新都重新渲染整个组件甚至是整个应用,这样的做法在页面复杂时将会暴露出性能上的问题,DOM操作非常昂贵,而为了减少DOM操作,React又做了哪些事?这就是我们下一篇文章的内容了。

这篇文章的代码:https://github.com/hujiulong/simple-react/tree/chapter-2


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值