引言
<<往期回顾>>
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', () => { let click const Child = { name: 'Child', setup(props, { emit }) { click = () => { emit('click') } return { click } }, render() { return h('div', {}, this.$props.a) } } const app = createApp({ name: 'App', setup() { const a = ref(1); const changeA = () => { a.value++ } return { a, changeA } }, render() { return h('div', { class: 'container' }, [h('p', {}, this.a), h(Child, { a: this.a, onClick: this.changeA })]) } }) const appDoc = document.querySelector('#app') app.mount(appDoc) // 默认开始挂载严重 const containerDom = document.querySelector('.container') expect(containerDom?.innerHTML).toBe('<p>1</p><div>1</div>') // 调用click click() expect(containerDom?.innerHTML).toBe('<p>2</p><div>2</div>')
分析
根据上面的流程图,可以分析出下面的需求
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属性 export function createVNode(type, props?, children?) { const vnode = { ...省略其他属性 // 当前组件的实例 component: null, } } // 在instance中添加 next属性和update方法,方便后续使用 export function createComponentInstance(vnode, parent) { const instance = { // ... 省略其他属性 // 更新后组件的vnode next: null, // 当前组件的更新函数,调用后,自动执行render函数 update: null, } vnode.component = instance } // 绑定insance.update,在setupRenderEffect中调用effect的时候绑定 // 实现updateComponent,并且在processComponent满足n1的时候来进行调用 function updateComponent(n1, n2) { // 更新组件 const instance = (n2.component = n1.component) // 判断是否需要更新 if (需要更新(n1, n2)) { instance.next = n2; instance.update(); } else { // 不需要更新则赋值 n2.el = n1.el; instance.vnode = n2 } } // 在setupRenderEffect中对更新部分进行改造,存在next的时候来调用updateComponentPreRender function updateComponentPreRender(instance, nextVNode) { // 把当前实例赋值给更新的vnode nextVNode.component = instance; // 更新当前实例的vode instance.vnode = nextVNode // 置空newVnode instance.next = null; // 更新属性 instance.props = nextVNode.props; // 更新插槽 instance.slots = nextVNode.slots; }
总结
本期主要实现了vue3的组件更新,在组件更新中,主要的流程是 updateComponent--> updateComponentPreRender --> render, 在这三个函数中交换新老vnode的属性,给把当前的vnode给更新掉即可