手寫簡易前端框架:function 和 class 元件

語言: CN / TW / HK

上篇文章我們實現了 vdom 的渲染,這是前端框架的基礎。但手寫 vdom 太麻煩,我們又支援了 jsx,用它來寫頁面更簡潔。

jsx 不是直接編譯成 vdom 的,而是生成 render function,執行之後產生 vdom。

中間多加了一層 render function,可以執行一些動態邏輯。別小看這一層 render function,它恰恰是實現元件的原理。

實現元件渲染

支援了 jsx 後,可以執行一些動態邏輯,比如迴圈、比如從上下文中取值:

const list = ['aaa', 'bbb'];
const jsx = <ul className="list">
{
list.map(item => <li className="item">{item}</li>)
}
</ul>


render(jsx, document.getElementById('root'));

這個封裝成函式,然後傳入引數不就是元件麼?

我們在 render 函式裡處理下函式元件的渲染:

if (isComponentVdom(vdom)) {
const props = Object.assign({}, vdom.props, {
children: vdom.children
});

const componentVdom = vdom.type(props);
return render(componentVdom, parent);
}

如果是 vdom 是一個元件,那麼就建立 props 作為引數傳入(props 要加上 children),執行該函式元件,拿到返回的 vdom 再渲染。

判斷元件就是根據 type 是否為 function:

function isComponentVdom(vdom) {
return typeof vdom.type == 'function';
}

就這幾行程式碼,我們就實現了函式元件。

測試下效果,宣告兩個函式元件,傳入 props:

function Item(props) {
return <li className="item" style={props.style} onClick={props.onClick}>{props.children}</li>;
}

function List(props) {
return <ul className="list">
{props.list.map((item, index) => {
return <Item style={{ background: item.color }} onClick={() => alert(item.text)}>{item.text}</Item>
})}
</ul>
;
}

const list = [
{
text: 'aaa',
color: 'blue'
},
{
text: 'ccc',
color: 'orange'
},
{
text: 'ddd',
color: 'red'
}
]

render(<List list={list}/>, document.getElementById('root'));

在瀏覽器跑一下:

我們實現了函式元件!

是不是非常簡單!它其實就是在 jsx 的基礎上封裝成了函式,然後傳入引數而已。

然後再實現下 class 元件:

class 元件需要宣告一個類,有 state 的屬性:

class Component {
constructor(props) {
this.props = props || {};
this.state = null;
}

setState(nextState) {
this.state = nextState;
}
}

然後渲染 vdom 的時候,如果是類元件,單獨處理下:

if (isComponentVdom(vdom)) {
const props = Object.assign({}, vdom.props, {
children: vdom.children
});

if (Component.isPrototypeOf(vdom.type)) {
const instance = new vdom.type(props);

const componentVdom = instance.render();
instance.dom = render(componentVdom, parent);

return instance.dom;
} else {
const componentVdom = vdom.type(props);
return render(componentVdom, parent);
}
}

判斷如果 vdom 是 Component,就傳入 props 建立例項,然後呼叫 render 拿到 vdom 再渲染。

還可以加上渲染前後的生命週期函式:

const instance = new vdom.type(props);

instance.componentWillMount();

const componentVdom = instance.render();
instance.dom = render(componentVdom, parent);

instance.componentDidMount();

return instance.dom;

這樣就實現了 class 元件。

我們測試下,宣告一個 class 元件,傳入 props,設定 state:

function Item(props) {
return <li className="item" style={props.style} onClick={props.onClick}>{props.children}</li>;
}

class List extends Component {
constructor(props) {
super();
this.state = {
list: [
{
text: 'aaa',
color: 'blue'
},
{
text: 'bbb',
color: 'orange'
},
{
text: 'ccc',
color: 'red'
}
],
textColor: props.textColor
}
}

render() {
return <ul className="list">
{this.state.list.map((item, index) => {
return <Item style={{ background: item.color, color: this.state.textColor}} onClick={() => alert(item.text)}>{item.text}</Item>
})}
</ul>
;
}
}

render(<List textColor={'pink'}/>, document.getElementById('root'));

瀏覽器跑一下:

class 元件渲染成功!

就這樣,我們實現了 class 元件,支援了 props 和 state。

程式碼上傳到了 github:https://github.com/QuarkGluonPlasma/frontend-framework-exercize

總結

上篇文章我們支援了 jsx,它編譯產生 render function,執行之後可以拿到 vdom,然後再渲染。

多了這層 render function 之後,它可以執行很多動態邏輯,比如條件判斷、迴圈,從上下文取值等。

對這些邏輯封裝一下就是元件了:

  • 封裝成函式,傳入 props,就是函式元件

  • 封裝成 class,傳入 props,設定 state 屬性,就是 class 元件

「元件本質上是對 vdom 的動態渲染邏輯的封裝,class 和 function 是兩種封裝形式」。

實現了 vdom 的渲染之後,支援元件的兩種封裝形式是非常簡單的事情。

至此,我們支援了 vdom 渲染、jsx 編譯、class 和 function 元件,渲染部分基本差不多了,下篇文章我們來實現渲染之後的更新,也就是 patch 的功能。