diff --git a/.changeset/new-ants-speak.md b/.changeset/new-ants-speak.md new file mode 100644 index 000000000000..71ba9b807ebd --- /dev/null +++ b/.changeset/new-ants-speak.md @@ -0,0 +1,5 @@ +--- +'astro': patch +--- + +Fix streaming Astro components diff --git a/packages/astro/src/runtime/server/render/component.ts b/packages/astro/src/runtime/server/render/component.ts index 33de90a26575..f78f11d517f7 100644 --- a/packages/astro/src/runtime/server/render/component.ts +++ b/packages/astro/src/runtime/server/render/component.ts @@ -413,28 +413,35 @@ async function renderHTMLComponent( }; } -async function renderAstroComponent( +function renderAstroComponent( result: SSRResult, displayName: string, Component: AstroComponentFactory, props: Record, slots: any = {} -): Promise { +): RenderInstance { const instance = createAstroComponentInstance(result, displayName, Component, props, slots); - // Eagerly render the component so they are rendered in parallel - const chunks: RenderDestinationChunk[] = []; - const temporaryDestination: RenderDestination = { - write: (chunk) => chunks.push(chunk), + // Eagerly render the component so they are rendered in parallel. + // Render to buffer for now until our returned render function is called. + const bufferChunks: RenderDestinationChunk[] = []; + const bufferDestination: RenderDestination = { + write: (chunk) => bufferChunks.push(chunk), }; - await instance.render(temporaryDestination); + // Don't await for the render to finish to not block streaming + const renderPromise = instance.render(bufferDestination); return { - render(destination) { - // The real render function will simply pass on the results from the temporary destination - for (const chunk of chunks) { + async render(destination) { + // Write the buffered chunks to the real destination + for (const chunk of bufferChunks) { destination.write(chunk); } + // Free memory + bufferChunks.length = 0; + // Re-assign the real destination so `instance.render` will continue and write to the new destination + bufferDestination.write = (chunk) => destination.write(chunk); + await renderPromise; }, }; } @@ -460,7 +467,7 @@ export async function renderComponent( } if (isAstroComponentFactory(Component)) { - return await renderAstroComponent(result, displayName, Component, props, slots); + return renderAstroComponent(result, displayName, Component, props, slots); } return await renderFrameworkComponent(result, displayName, Component, props, slots); diff --git a/packages/astro/test/streaming.test.js b/packages/astro/test/streaming.test.js index e3627d7ba0e2..c7b835de1b7c 100644 --- a/packages/astro/test/streaming.test.js +++ b/packages/astro/test/streaming.test.js @@ -48,7 +48,7 @@ describe('Streaming', () => { let chunk = decoder.decode(bytes); chunks.push(chunk); } - expect(chunks.length).to.equal(2); + expect(chunks.length).to.equal(3); }); });