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

feat(templateRef): support template ref for all vnode types

Browse Source
This commit is contained in:
Evan You 2020-03-31 10:30:12 -04:00
parent 94562daea7
commit 55b364decc
1 changed files with 8 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
packages/runtime-core/src/renderer.ts
View File

@ -363,7 +363,7 @@ function baseCreateRenderer(
n1 = null n1 = null
} }
const { type, shapeFlag } = n2 const { type, ref, shapeFlag } = n2
switch (type) { switch (type) {
case Text: case Text:
processText(n1, n2, container, anchor) processText(n1, n2, container, anchor)
@ -439,6 +439,13 @@ function baseCreateRenderer(
warn('Invalid VNode type:', type, `(${typeof type})`) warn('Invalid VNode type:', type, `(${typeof type})`)
} }
} }
// set ref
if (ref != null && parentComponent) {
const refValue =
shapeFlag & ShapeFlags.STATEFUL_COMPONENT ? n2.component!.proxy : n2.el
setRef(ref, n1 && n1.ref, parentComponent, refValue)
}
} }
const processText: ProcessTextOrCommentFn = (n1, n2, container, anchor) => { const processText: ProcessTextOrCommentFn = (n1, n2, container, anchor) => {
@ -518,9 +525,6 @@ function baseCreateRenderer(
} else { } else {
patchElement(n1, n2, parentComponent, parentSuspense, isSVG, optimized) patchElement(n1, n2, parentComponent, parentSuspense, isSVG, optimized)
} }
if (n2.ref != null && parentComponent) {
setRef(n2.ref, n1 && n1.ref, parentComponent, n2.el)
}
} }
const mountElement = ( const mountElement = (
@ -1005,17 +1009,6 @@ function baseCreateRenderer(
n2.el = n1.el n2.el = n1.el
} }
} }
if (n2.ref != null && parentComponent) {
if (__DEV__ && !(n2.shapeFlag & ShapeFlags.STATEFUL_COMPONENT)) {
pushWarningContext(n2)
warn(
`Functional components do not support "ref" because they do not ` +
`have instances.`
)
popWarningContext()
}
setRef(n2.ref, n1 && n1.ref, parentComponent, n2.component!.proxy)
}
} }
const mountComponent: MountComponentFn = ( const mountComponent: MountComponentFn = (