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

摘要

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

javascript 复制代码
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。

javascript 复制代码
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。

javascript 复制代码
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,而是也要判断是否是可复用的。

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

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

4.实现updateCompleteHostComponent方法

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

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

javascript 复制代码
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)
}

addPropsToDOM方法是我们在实现事件机制的时候,需要调用的方法。

OK,这样我们就实现了单节点的Diff算法。

相关推荐
mit6.8241 小时前
前后缀分解
算法
demi_meng1 小时前
reactNative 遇到的问题记录
javascript·react native·react.js
你好,我叫C小白1 小时前
C语言 循环结构(1)
c语言·开发语言·算法·while·do...while
千码君20162 小时前
React Native:从react的解构看编程众多语言中的解构
java·javascript·python·react native·react.js·解包·解构
寂静山林4 小时前
UVa 10228 A Star not a Tree?
算法
Neverfadeaway4 小时前
【C语言】深入理解函数指针数组应用(4)
c语言·开发语言·算法·回调函数·转移表·c语言实现计算器
EndingCoder4 小时前
WebSocket实时通信:Socket.io
服务器·javascript·网络·websocket·网络协议·node.js
我胡为喜呀4 小时前
Vue3 中的 watch 和 watchEffect:如何优雅地监听数据变化
前端·javascript·vue.js
Madison-No75 小时前
【C++】探秘vector的底层实现
java·c++·算法
Swift社区5 小时前
LeetCode 401 - 二进制手表
算法·leetcode·ssh