引言

<<往期回顾>>

1. vue3源码分析——实现组件通信provide,inject
2. vue3源码分析——实现createRenderer，增加runtime-test
3. vue3源码分析——实现element属性更新，child更新
4. vue3源码分析——手写diff算法

前面的两期主要是实现element的更新,vue的更新除了element的更新外,还有component的更新哦,本期就带大家一起来看看,本期所有的源码请查看

正文

vue3在更新element的时候,除了需要分情况讨论更新children外,还需要来看vue3的属性有没有变化;那么同样的道理,对于组件的更新,也是需要来更新属性,插槽等

流程

看到这个流程,是不是感觉比element的更新简单许多(❁´◡`❁)

测试用例

根据上面的流程图,可以写出这样的测试用例

test('test comp update by Child', () => {n let clickn const Child = {n name: 'Child',n setup(props, { emit }) {n click = () => {n emit('click')n }n return {n clickn }n },n render() {n return h('div', {}, this.$props.a)n }n }n const app = createApp({n name: 'App',n setup() {n const a = ref(1);n const changeA = () => {n a.value++n }n return {n a,n changeAn }n },n render() {n return h('div', { class: 'container' }, [h('p', {}, this.a), h(Child, { a: this.a, onClick: this.changeA })])n }n })n const appDoc = document.querySelector('#app')n app.mount(appDoc)n // 默认开始挂载严重n const containerDom = document.querySelector('.container')n expect(containerDom?.innerHTML).toBe('<p>1</p><div>1</div>')n // 调用clickn click()n expect(containerDom?.innerHTML).toBe('<p>2</p><div>2</div>')n

分析

根据上面的流程图,可以分析出下面的需求

1. updateComponent里面需要实现啥;
2. updatePreRender函数里面又是需要做啥;
3. 怎么调用render函数呢?

问题解决:

updateComponent

updateComponent方法的调用肯定是在processComponent中的旧vnode存在的时候来调用,里面需要进行新老节点的对比,判断是否需要进行更新.更新则调用updateComponentPreRender去更新vnode的props,slots等.这里会还需要把新vnode给保存在实例当中方便后续的使用,最后还需要在当前的vnode当中保存当前组件的实例,方便后续交换新老vnode的时候调用.

updatePreRender

这个函数就是只要处理更新的逻辑即可

调用render

render的调用是在 setupRenderEffect中调用的,是不是可以重复利用下这个功能呢? 当然可以,effect函数是默认返回一个runner的,可以手动调用runner来执行effect里面的方法. 那么可以在当前的实例当中保存一个update方法,用于需要调用render的时候来进行调用即可.

编码

// 在创建vnode当中,添加component属性nexport function createVNode(type, props?, children?) {n const vnode = {n ...省略其他属性n // 当前组件的实例n component: null,n } n}n// 在instance中添加 next属性和update方法,方便后续使用nnexport function createComponentInstance(vnode, parent) {n const instance = {n // ... 省略其他属性 n // 更新后组件的vnoden next: null,n // 当前组件的更新函数，调用后，自动执行render函数n update: null,n }n vnode.component = instancen}nn// 绑定insance.update,在setupRenderEffect中调用effect的时候绑定nn// 实现updateComponent,并且在processComponent满足n1的时候来进行调用nfunction updateComponent(n1, n2) {n // 更新组件n const instance = (n2.component = n1.component)n // 判断是否需要更新n if (需要更新(n1, n2)) {n instance.next = n2;n instance.update();n } else {n // 不需要更新则赋值n n2.el = n1.el;n instance.vnode = n2n }n } n n // 在setupRenderEffect中对更新部分进行改造,存在next的时候来调用updateComponentPreRendern n function updateComponentPreRender(instance, nextVNode) {n // 把当前实例赋值给更新的vnoden nextVNode.component = instance;n // 更新当前实例的voden instance.vnode = nextVNoden // 置空newVnoden instance.next = null;n // 更新属性n instance.props = nextVNode.props;n // 更新插槽n instance.slots = nextVNode.slots;n }n n

总结

本期主要实现了vue3的组件更新,在组件更新中,主要的流程是 updateComponent--> updateComponentPreRender --> render, 在这三个函数中交换新老vnode的属性,给把当前的vnode给更新掉即可