perf(ssr): avoid unnecessary await ticks when unrolling sync buffers

This commit is contained in:
Evan You 2020-06-26 11:10:30 -04:00
parent 6bc0e0a31a
commit 30584bcc61
3 changed files with 54 additions and 10 deletions

View File

@ -20,7 +20,8 @@ import {
isPromise,
isString,
isVoidTag,
ShapeFlags
ShapeFlags,
isArray
} from '@vue/shared'
import { ssrRenderAttrs } from './helpers/ssrRenderAttrs'
import { ssrCompile } from './helpers/ssrCompile'
@ -35,7 +36,7 @@ const {
normalizeSuspenseChildren
} = ssrUtils
export type SSRBuffer = SSRBufferItem[]
export type SSRBuffer = SSRBufferItem[] & { hasAsync?: boolean }
export type SSRBufferItem = string | SSRBuffer | Promise<SSRBuffer>
export type PushFn = (item: SSRBufferItem) => void
export type Props = Record<string, unknown>
@ -68,6 +69,11 @@ export function createBuffer() {
buffer.push(item)
}
appendable = isStringItem
if (isPromise(item) || (isArray(item) && item.hasAsync)) {
// promise, or child buffer with async, mark as async.
// this allows skipping unnecessary await ticks during unroll stage
buffer.hasAsync = true
}
}
}
}

View File

@ -16,6 +16,7 @@ async function unrollBuffer(
buffer: SSRBuffer,
stream: Readable
): Promise<void> {
if (buffer.hasAsync) {
for (let i = 0; i < buffer.length; i++) {
let item = buffer[i]
if (isPromise(item)) {
@ -27,6 +28,23 @@ async function unrollBuffer(
await unrollBuffer(item, stream)
}
}
} else {
// sync buffer can be more efficiently unrolled without unnecessary await
// ticks
unrollBufferSync(buffer, stream)
}
}
function unrollBufferSync(buffer: SSRBuffer, stream: Readable) {
for (let i = 0; i < buffer.length; i++) {
let item = buffer[i]
if (isString(item)) {
stream.push(item)
} else {
// since this is a sync buffer, child buffers are never promises
unrollBufferSync(item as SSRBuffer, stream)
}
}
}
export function renderToStream(

View File

@ -12,6 +12,7 @@ import { SSRContext, renderComponentVNode, SSRBuffer } from './render'
const { isVNode } = ssrUtils
async function unrollBuffer(buffer: SSRBuffer): Promise<string> {
if (buffer.hasAsync) {
let ret = ''
for (let i = 0; i < buffer.length; i++) {
let item = buffer[i]
@ -21,7 +22,26 @@ async function unrollBuffer(buffer: SSRBuffer): Promise<string> {
if (isString(item)) {
ret += item
} else {
ret += await unrollBuffer(item as SSRBuffer)
ret += await unrollBuffer(item)
}
}
return ret
} else {
// sync buffer can be more efficiently unrolled without unnecessary await
// ticks
return unrollBufferSync(buffer)
}
}
function unrollBufferSync(buffer: SSRBuffer): string {
let ret = ''
for (let i = 0; i < buffer.length; i++) {
let item = buffer[i]
if (isString(item)) {
ret += item
} else {
// since this is a sync buffer, child buffers are never promises
ret += unrollBufferSync(item as SSRBuffer)
}
}
return ret