theluyuan/vue3-yuanma
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: adjust custom element hmr, fix tests

Browse Source
This commit is contained in:
Evan You 2021-07-22 23:18:15 -04:00
parent bff4ea74c5
commit 231dafd55a
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
packages/runtime-core/src/hmr.ts
View File

@ -7,7 +7,7 @@ import {
ClassComponent, ClassComponent,
isClassComponent isClassComponent
} from './component' } from './component'
import { nextTick, queueJob } from './scheduler' import { queueJob, queuePostFlushCb } from './scheduler'
import { extend } from '@vue/shared' import { extend } from '@vue/shared'
import { warn } from './warning' import { warn } from './warning'
@ -124,7 +124,7 @@ function reload(id: string, newComp: ComponentOptions | ClassComponent) {
// on patch. // on patch.
hmrDirtyComponents.add(component) hmrDirtyComponents.add(component)
// 3. Make sure to unmark the component after the reload. // 3. Make sure to unmark the component after the reload.
nextTick(() => { queuePostFlushCb(() => {
hmrDirtyComponents.delete(component) hmrDirtyComponents.delete(component)
}) })
} }
@ -135,7 +135,9 @@ function reload(id: string, newComp: ComponentOptions | ClassComponent) {
if (instance.ceReload) { if (instance.ceReload) {
// custom element // custom element
hmrDirtyComponents.add(component)
instance.ceReload() instance.ceReload()
hmrDirtyComponents.delete(component)
} else if (instance.parent) { } else if (instance.parent) {
// 4. Force the parent instance to re-render. This will cause all updated // 4. Force the parent instance to re-render. This will cause all updated
// components to be unmounted and re-mounted. Queue the update so that we // components to be unmounted and re-mounted. Queue the update so that we