Vue3生命周期的实现原理
Vue3的生命周期Hooks函数的实现原理还是比较简单的,就是把各个生命周期的函数挂载或者叫注册到组件的实例上,然后等到组件运行到某个时刻,再去组件实例上把相应的生命周期的函数取出来执行
生命周期类型
// packages/runtime-core/src/component.ts
export const enum LifecycleHooks {
BEFORE_CREATE = 'bc', // 创建之前
CREATED = 'c', // 创建
BEFORE_MOUNT = 'bm', // 挂载之前
MOUNTED = 'm', // 挂载之后
BEFORE_UPDATE = 'bu', // 更新之前
UPDATED = 'u', // 更新之后
BEFORE_UNMOUNT = 'bum', // 卸载之前
UNMOUNTED = 'um', // 卸载之后
// ...
}
各个生命周期Hooks函数的创建
// packages/runtime-core/src/apiLifecycle.ts
export const onBeforeMount = createHook(LifecycleHooks.BEFORE_MOUNT)
export const onMounted = createHook(LifecycleHooks.MOUNTED)
export const onBeforeUpdate = createHook(LifecycleHooks.BEFORE_UPDATE)
export const onUpdated = createHook(LifecycleHooks.UPDATED)
export const onBeforeUnmount = createHook(LifecycleHooks.BEFORE_UNMOUNT)
export const onUnmounted = createHook(LifecycleHooks.UNMOUNTED)
可以看到各个生命周期的Hooks函数是通过createHook这个函数创建的
创建生命周期函数createHook
// packages/runtime-core/src/apiLifecycle.ts
export const createHook = (lifecycle) => (hook, target = currentInstance) => injectHook(lifecycle, hook, target)
createHook是一个闭包函数,通过闭包缓存当前是属于哪个生命周期的Hooks,target表示该生命周期Hooks函数被绑定到哪个组件实例上,默认是当前工作的组件实例。createHook底层又调用了一个injectHook的函数,那么下面我们继续来看看这个injectHook函数。
injectHook函数: injectHook是一个闭包函数,通过闭包缓存绑定对应生命周期Hooks到对应的组件实例上。
// packages/runtime-core/src/apiLifecycle.ts
export function injectHook(type, hook, target) {
if(target) {
// 把各个生命周期的Hooks函数挂载到组件实例上,并且是一个数组,因为可能你会多次调用同一个组件的同一个生命周期函数
const hooks = target[type] || (target[type] = [])
// 把生命周期函数进行包装并且把包装函数缓存在__weh上
const wrappedHook =
hook.__weh ||
(hook.__weh = (...args: unknown[]) => {
if (target.isUnmounted) {
return
}
// 当生命周期调用时 保证currentInstance是正确的
setCurrentInstance(target)
// 执行生命周期Hooks函数
const res = args ? hook(...args) : hook()
unsetCurrentInstance()
return res
})
// 把生命周期的包装函数绑定到组件实例对应的hooks上
hooks.push(wrappedHook)
// 返回包装函数
return wrappedHook
}
}
生命周期Hooks的调用
instance.update = effect(() => {
if (!instance.isMounted) {
const { bm, m } = instance
// 生命周期:beforeMount hook
if (bm) {
invokeArrayFns(bm)
}
// 组件初始化的时候会执行这里
// 为什么要在这里调用 render 函数呢
// 是因为在 effect 内调用 render 才能触发依赖收集
// 等到后面响应式的值变更后会再次触发这个函数
const subTree = (instance.subTree = renderComponentRoot(instance))
patch(null, subTree, container, instance, anchor)
instance.vnode.el = subTree.el
instance.isMounted = true
// 生命周期:mounted
if(m) {
// mounted需要通过Scheduler的函数来调用
queuePostFlushCb(m)
}
} else {
// 响应式的值变更后会从这里执行逻辑
// 主要就是拿到新的 vnode ,然后和之前的 vnode 进行对比
// 拿到最新的 subTree
const { bu, u, next, vnode } = instance
// 如果有 next 的话, 说明需要更新组件的数据(props,slots 等)
// 先更新组件的数据,然后更新完成后,在继续对比当前组件的子元素
if(next) {
next.el = vnode.el
updateComponentPreRender(instance, next)
}
// 生命周期:beforeUpdate hook
if (bu) {
invokeArrayFns(bu)
}
const subTree = renderComponentRoot(instance)
// 替换之前的 subTree
const prevSubTree = instance.subTree
instance.subTree = subTree
// 用旧的 vnode 和新的 vnode 交给 patch 来处理
patch(prevSubTree, subTree, container, instance, anchor)
// 生命周期:updated hook
if (u) {
// updated 需要通过Scheduler的函数来调用
queuePostFlushCb(u)
}
}
}, {
scheduler() {
queueJobs(instance.update)
}
})
上面这个是Vue3组件实例化之后,通过effect包装一个更新的副作用函数来和响应式数据进行依赖收集。在这个副作用函数里面有两个分支,第一个是组件挂载之前执行的,也就是生命周期函数beforeMount和mount调用的地方,第二个分支是组件挂载之后更新的时候执行的,在这里就是生命周期函数beforeUpdate和updated调用的地方。 具体就是在挂载之前,还没生成虚拟DOM之前就执行beforeMount函数,之后则去生成虚拟DOM经过patch之后,组件已经被挂载到页面上了,也就是页面上显示视图了,这个时候就去执行mount函数;在更新的时候,还没获取更新之后的虚拟DOM之前执行beforeUpdate,然后去获取更新之后的虚拟DOM,然后再去patch,更新视图,之后就执行updated。 需要注意的是beforeMount和beforeUpdate是同步执行的,都是通过invokeArrayFns来调用的。 invokeArrayFns函数
export const invokeArrayFns = (fns: Function[], arg?: any) => {
for (let i = 0; i < fns.length; i++) {
fns[i](arg)
}
}