当前位置：首页 > news >正文

React源码解析18（8）------ 实现单节点的Diff算法

news 2025/9/12 1:21:01

摘要

经过之前的几篇文章，我们已经实现了一个可以进行更新渲染的假React。但是如果我们把我们的jsx修改成这样：

function App() {const [age, setAge] = useState(20)const click = function() {setAge(age + 1)}return age % 2 === 0 ? jsx("div", {key: 'div1',children: jsx("span", {key: 'span',children: 'div1',onClick: click})}) : jsx("div", {key: 'div1',children: jsx("span", {key: 'span',children: 'div2',onClick: click})})
}ReactDOM.createRoot(root).render(<App />)

虽然页面的效果是正确的。但对于两个jsx，二者的区别仅仅是span标签里面的内容不同。
但是在程序里面，我们是相当于每次都重新beginWork，重新的创建Filber树，重新的创建真实DOM。
而对于这里div和span标签，它没有任何的改变，我们是否可以用一种优化策略，从而对旧资源进行利用呢？所以Diff由此而来。

这一篇先只说单节点的Diff，因为目前还没实现带有sibling的情况。

1.修改beginWork

我们回顾一下在beginWork里面干了什么。在将jsx转换为ReactElement后，我们会通过beginWork来构建一颗Filber树。
那如果，对于可复用的FilberNode，我们是否可以不去创建，直接复用呢？
可以的，那对于React来说，什么节点是可复用的呢：

如果旧的FilberNode和新的ReactElement
key相同，type相同。
那么就是可以复用的。

所以在beginWork中的reconcileChildren方法里，如果我们发现上面的情况，我们就不会创建新的FilberNode。

function reconcileChildren(parent,element) {const newChild = diffReconcileChildren(parent, element);if(newChild) {return newChild}//其他代码。。。return filberNode
}

2.实现diffReconcileChildren方法

该方法接受两个参数，第一个是父节点，第二个是新的ReactElement。

（1）我们要先拿到父节点的child，比较child和element的key和type。
（2）将element保存在child的memoizedState里面。
（3）然后其他逻辑和reconcileChildren里的一样即可。
（4）直接返回child。

function diffReconcileChildren(filberNode, element) {const child = filberNode.child;if(child && child.key && child.type) {if(child.key === element.key && child.type === child.type) {child.memoizedProps = element;child.pendingProps = element.props;if(child.tag === HostText) {child.pendingProps = element}return child;}}
}

这里为什么要将element保存在memoizedState里面，是因为虽然节点没有改变，但是子节点可能有改变，或者属性会有改变。所以要在后面的completeWork里进行处理。

3.修改completeWork

经过上面的处理后，FilberNode不会无脑的重复创建，而是复用了。而completeWork的工作，主要是创建真实DOM，挂载在FilberNode的stateNode上。

所以在completeWork中，也不能无脑的创建真实DOM，而是也要判断是否是可复用的。

export const completeWork = (filberNode) => {const tag = filberNode.tagswitch (tag) {case HostComponent: {if(filberNode.stateNode !== null){//更新updateCompleteHostComponent(filberNode)}else{completeHostComponent(filberNode)}break;}//其他代码。。。。}
}

所以在对HostComponent的处理上，如果发现不是mount阶段，就要判断是否需要复用旧的。

4.实现updateCompleteHostComponent方法

在该方法中，我们接受filberNode。同时我们可以拿到它的memoizedState（在beginWork中传过来的）。

再判断一下key和type，如果依旧相同，那么说明是可复用的。我们直接不创建新的DOM即可。

function updateCompleteHostComponent(filberNode) {const element = filberNode.memoizedProps;if(element.key === filberNode.key && element.type === filberNode.type) {addPropsToDOM(filberNode.stateNode, filberNode.pendingProps)}else{const type = filberNode.type;const element = document.createElement(type);addPropsToDOM(element, filberNode.pendingProps)filberNode.stateNode = element;const parent = filberNode.return;if(parent && parent.stateNode && parent.tag === HostComponent) {parent.stateNode.appendChild(element)}}completeWork(filberNode.child)
}