Skip to content

Commit

Permalink
fix(runtime-core): disable block tracking when calling compiled slot …
Browse files Browse the repository at this point in the history
…function in tempalte expressions

fix #1745, fix #1918
  • Loading branch information
yyx990803 committed Aug 21, 2020
1 parent c0411b3 commit f02e2f9
Show file tree
Hide file tree
Showing 3 changed files with 22 additions and 14 deletions.
6 changes: 3 additions & 3 deletions packages/runtime-core/src/helpers/renderSlot.ts
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ import {
import { PatchFlags, SlotFlags } from '@vue/shared'
import { warn } from '../warning'

export let shouldTrackInSlotRendering = 0
export let isRenderingCompiledSlot = 0

/**
* Compiler runtime helper for rendering `<slot/>`
Expand Down Expand Up @@ -39,7 +39,7 @@ export function renderSlot(
// invocation interfering with template-based block tracking, but in
// `renderSlot` we can be sure that it's template-based so we can force
// enable it.
shouldTrackInSlotRendering++
isRenderingCompiledSlot++
const rendered = (openBlock(),
createBlock(
Fragment,
Expand All @@ -49,6 +49,6 @@ export function renderSlot(
? PatchFlags.STABLE_FRAGMENT
: PatchFlags.BAIL
))
shouldTrackInSlotRendering--
isRenderingCompiledSlot--
return rendered
}
17 changes: 11 additions & 6 deletions packages/runtime-core/src/helpers/withRenderContext.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,8 @@ import {
currentRenderingInstance
} from '../componentRenderUtils'
import { ComponentInternalInstance } from '../component'
import { setBlockTracking } from '../vnode'
import { isRenderingCompiledSlot } from './renderSlot'
import { closeBlock, openBlock } from '../vnode'

/**
* Wrap a slot function to memoize current rendering instance
Expand All @@ -16,15 +17,19 @@ export function withCtx(
) {
if (!ctx) return fn
return function renderFnWithContext() {
// By default, compiled slots disables block tracking since the user may
// call it inside a template expression (#1745). It should only track when
// it's called by a template `<slot>`.
setBlockTracking(-1)
// If a user calls a compiled slot inside a template expression (#1745), it
// can mess up block tracking, so by default we need to push a null block to
// avoid that. This isn't necessary if rendering a compiled `<slot>`.
if (!isRenderingCompiledSlot) {
openBlock(true /* null block that disables tracking */)
}
const owner = currentRenderingInstance
setCurrentRenderingInstance(ctx)
const res = fn.apply(null, arguments as any)
setCurrentRenderingInstance(owner)
setBlockTracking(1)
if (!isRenderingCompiledSlot) {
closeBlock()
}
return res
}
}
13 changes: 8 additions & 5 deletions packages/runtime-core/src/vnode.ts
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,6 @@ import { currentRenderingInstance } from './componentRenderUtils'
import { RendererNode, RendererElement } from './renderer'
import { NULL_DYNAMIC_COMPONENT } from './helpers/resolveAssets'
import { hmrDirtyComponents } from './hmr'
import { shouldTrackInSlotRendering } from './helpers/renderSlot'

export const Fragment = (Symbol(__DEV__ ? 'Fragment' : undefined) as any) as {
__isFragment: true
Expand Down Expand Up @@ -153,7 +152,7 @@ export interface VNode<
// can divide a template into nested blocks, and within each block the node
// structure would be stable. This allows us to skip most children diffing
// and only worry about the dynamic nodes (indicated by patch flags).
const blockStack: (VNode[] | null)[] = []
export const blockStack: (VNode[] | null)[] = []
let currentBlock: VNode[] | null = null

/**
Expand All @@ -176,6 +175,11 @@ export function openBlock(disableTracking = false) {
blockStack.push((currentBlock = disableTracking ? null : []))
}

export function closeBlock() {
blockStack.pop()
currentBlock = blockStack[blockStack.length - 1] || null
}

// Whether we should be tracking dynamic child nodes inside a block.
// Only tracks when this value is > 0
// We are not using a simple boolean because this value may need to be
Expand Down Expand Up @@ -227,8 +231,7 @@ export function createBlock(
// save current block children on the block vnode
vnode.dynamicChildren = currentBlock || EMPTY_ARR
// close block
blockStack.pop()
currentBlock = blockStack[blockStack.length - 1] || null
closeBlock()
// a block is always going to be patched, so track it as a child of its
// parent block
if (currentBlock) {
Expand Down Expand Up @@ -403,7 +406,7 @@ function _createVNode(
normalizeChildren(vnode, children)

if (
(shouldTrack > 0 || shouldTrackInSlotRendering > 0) &&
shouldTrack > 0 &&
// avoid a block node from tracking itself
!isBlockNode &&
// has current parent block
Expand Down

0 comments on commit f02e2f9

Please sign in to comment.