[Fizz] Support SuspenseList revealOrder="together" (#33311)

Stacked on #33308.

For "together" mode, we can be a self-blocking row that adds all its
boundaries to the blocked set, but there's no parent row that unblocks
it.

A particular quirk of this mode is that it's not enough to just unblock
them all on the server together. Because if one boundary downloads all
its html and then issues a complete instruction it'll appear before the
others while streaming in. What we actually want is to reveal them all
in a single batch.

This implementation takes a short cut by unblocking the rows in
`flushPartialBoundary`. That ensures that all the segments of every
boundary has a chance to flush before we start emitting any of the
complete boundary instructions. Once the last one unblocks, all the
complete boundary instructions are queued. Ideally this would be a
single `<script>` tag so that they can't be split up even if we get a
chunk containing some of them.

~A downside of this approach is that we always outline these boundaries.
We could inline them if they all complete before the parent flushes.
E.g. by checking if the row is blocked only by its own boundaries and if
all the boundaries would fit without getting outlined, then we can
inline them all at once.~ I went ahead and did this because it solves an
issue with `renderToString` where it doesn't support the script runtime
so it can only handle this if inlined.
This commit is contained in:
Sebastian Markbåge
2025-05-20 14:42:05 -04:00
committed by GitHub
parent d38c7e10d3
commit 99aa685cef
3 changed files with 404 additions and 3 deletions
@@ -172,7 +172,6 @@ export {
completeResumableState,
emitEarlyPreloads,
supportsClientAPIs,
canHavePreamble,
hoistPreambleState,
isPreambleReady,
isPreambleContext,
@@ -194,6 +193,10 @@ export function getViewTransitionFormatContext(
return parentContext;
}
export function canHavePreamble(formatContext: FormatContext): boolean {
return false;
}
export function pushTextInstance(
target: Array<Chunk | PrecomputedChunk>,
text: string,
@@ -183,6 +183,323 @@ describe('ReactDOMFizSuspenseList', () => {
);
});
// @gate enableSuspenseList
it('displays all "together"', async () => {
const A = createAsyncText('A');
const B = createAsyncText('B');
const C = createAsyncText('C');
function Foo() {
return (
<div>
<SuspenseList revealOrder="together">
<Suspense fallback={<Text text="Loading A" />}>
<A />
</Suspense>
<Suspense fallback={<Text text="Loading B" />}>
<B />
</Suspense>
<Suspense fallback={<Text text="Loading C" />}>
<C />
</Suspense>
</SuspenseList>
</div>
);
}
await A.resolve();
await serverAct(async () => {
const {pipe} = ReactDOMFizzServer.renderToPipeableStream(<Foo />);
pipe(writable);
});
assertLog([
'A',
'Suspend! [B]',
'Suspend! [C]',
'Loading A',
'Loading B',
'Loading C',
]);
expect(getVisibleChildren(container)).toEqual(
<div>
<span>Loading A</span>
<span>Loading B</span>
<span>Loading C</span>
</div>,
);
await serverAct(() => B.resolve());
assertLog(['B']);
expect(getVisibleChildren(container)).toEqual(
<div>
<span>Loading A</span>
<span>Loading B</span>
<span>Loading C</span>
</div>,
);
await serverAct(() => C.resolve());
assertLog(['C']);
expect(getVisibleChildren(container)).toEqual(
<div>
<span>A</span>
<span>B</span>
<span>C</span>
</div>,
);
});
// @gate enableSuspenseList
it('displays all "together" in a single pass', async () => {
function Foo() {
return (
<div>
<SuspenseList revealOrder="together">
<Suspense fallback={<Text text="Loading A" />}>
<Text text="A" />
</Suspense>
<Suspense fallback={<Text text="Loading B" />}>
<Text text="B" />
</Suspense>
<Suspense fallback={<Text text="Loading C" />}>
<Text text="C" />
</Suspense>
</SuspenseList>
</div>
);
}
const {pipe} = ReactDOMFizzServer.renderToPipeableStream(<Foo />);
pipe(writable);
await 0;
const bufferedContent = buffer;
buffer = '';
assertLog(['A', 'B', 'C', 'Loading A', 'Loading B', 'Loading C']);
expect(bufferedContent).toMatchInlineSnapshot(
`"<div><!--$--><span>A</span><!--/$--><!--$--><span>B</span><!--/$--><!--$--><span>C</span><!--/$--></div>"`,
);
});
// @gate enableSuspenseList
it('displays all "together" even when nested as siblings', async () => {
const A = createAsyncText('A');
const B = createAsyncText('B');
const C = createAsyncText('C');
function Foo() {
return (
<div>
<SuspenseList revealOrder="together">
<div>
<Suspense fallback={<Text text="Loading A" />}>
<A />
</Suspense>
<Suspense fallback={<Text text="Loading B" />}>
<B />
</Suspense>
</div>
<div>
<Suspense fallback={<Text text="Loading C" />}>
<C />
</Suspense>
</div>
</SuspenseList>
</div>
);
}
await A.resolve();
await serverAct(async () => {
const {pipe} = ReactDOMFizzServer.renderToPipeableStream(<Foo />);
pipe(writable);
});
assertLog([
'A',
'Suspend! [B]',
'Suspend! [C]',
'Loading A',
'Loading B',
'Loading C',
]);
expect(getVisibleChildren(container)).toEqual(
<div>
<div>
<span>Loading A</span>
<span>Loading B</span>
</div>
<div>
<span>Loading C</span>
</div>
</div>,
);
await serverAct(() => B.resolve());
assertLog(['B']);
expect(getVisibleChildren(container)).toEqual(
<div>
<div>
<span>Loading A</span>
<span>Loading B</span>
</div>
<div>
<span>Loading C</span>
</div>
</div>,
);
await serverAct(() => C.resolve());
assertLog(['C']);
expect(getVisibleChildren(container)).toEqual(
<div>
<div>
<span>A</span>
<span>B</span>
</div>
<div>
<span>C</span>
</div>
</div>,
);
});
// @gate enableSuspenseList
it('displays all "together" in nested SuspenseLists', async () => {
const A = createAsyncText('A');
const B = createAsyncText('B');
const C = createAsyncText('C');
function Foo() {
return (
<div>
<SuspenseList revealOrder="together">
<Suspense fallback={<Text text="Loading A" />}>
<A />
</Suspense>
<SuspenseList revealOrder="together">
<Suspense fallback={<Text text="Loading B" />}>
<B />
</Suspense>
<Suspense fallback={<Text text="Loading C" />}>
<C />
</Suspense>
</SuspenseList>
</SuspenseList>
</div>
);
}
await A.resolve();
await B.resolve();
await serverAct(async () => {
const {pipe} = ReactDOMFizzServer.renderToPipeableStream(<Foo />);
pipe(writable);
});
assertLog([
'A',
'B',
'Suspend! [C]',
'Loading A',
'Loading B',
'Loading C',
]);
expect(getVisibleChildren(container)).toEqual(
<div>
<span>Loading A</span>
<span>Loading B</span>
<span>Loading C</span>
</div>,
);
await serverAct(() => C.resolve());
assertLog(['C']);
expect(getVisibleChildren(container)).toEqual(
<div>
<span>A</span>
<span>B</span>
<span>C</span>
</div>,
);
});
// @gate enableSuspenseList
it('displays all "together" in nested SuspenseLists where the inner is default', async () => {
const A = createAsyncText('A');
const B = createAsyncText('B');
const C = createAsyncText('C');
function Foo() {
return (
<div>
<SuspenseList revealOrder="together">
<Suspense fallback={<Text text="Loading A" />}>
<A />
</Suspense>
<SuspenseList>
<Suspense fallback={<Text text="Loading B" />}>
<B />
</Suspense>
<Suspense fallback={<Text text="Loading C" />}>
<C />
</Suspense>
</SuspenseList>
</SuspenseList>
</div>
);
}
await A.resolve();
await B.resolve();
await serverAct(async () => {
const {pipe} = ReactDOMFizzServer.renderToPipeableStream(<Foo />);
pipe(writable);
});
assertLog([
'A',
'B',
'Suspend! [C]',
'Loading A',
'Loading B',
'Loading C',
]);
expect(getVisibleChildren(container)).toEqual(
<div>
<span>Loading A</span>
<span>Loading B</span>
<span>Loading C</span>
</div>,
);
await serverAct(() => C.resolve());
assertLog(['C']);
expect(getVisibleChildren(container)).toEqual(
<div>
<span>A</span>
<span>B</span>
<span>C</span>
</div>,
);
});
// @gate enableSuspenseList
it('displays each items in "forwards" order', async () => {
const A = createAsyncText('A');
+83 -2
View File
@@ -236,6 +236,7 @@ type LegacyContext = {
type SuspenseListRow = {
pendingTasks: number, // The number of tasks, previous rows and inner suspense boundaries blocking this row.
boundaries: null | Array<SuspenseBoundary>, // The boundaries in this row waiting to be unblocked by the previous row. (null means this row is not blocked)
together: boolean, // All the boundaries within this row must be revealed together.
next: null | SuspenseListRow, // The next row blocked by this one.
};
@@ -1404,6 +1405,11 @@ function renderSuspenseBoundary(
}
return;
}
} else {
const boundaryRow = prevRow;
if (boundaryRow !== null && boundaryRow.together) {
tryToResolveTogetherRow(request, boundaryRow);
}
}
} catch (thrownValue: mixed) {
newBoundary.status = CLIENT_RENDERED;
@@ -1670,8 +1676,14 @@ function replaySuspenseBoundary(
function finishSuspenseListRow(request: Request, row: SuspenseListRow): void {
// This row finished. Now we have to unblock all the next rows that were blocked on this.
unblockSuspenseListRow(request, row.next);
}
function unblockSuspenseListRow(
request: Request,
unblockedRow: null | SuspenseListRow,
): void {
// We do this in a loop to avoid stack overflow for very long lists that get unblocked.
let unblockedRow = row.next;
while (unblockedRow !== null) {
// Unblocking the boundaries will decrement the count of this row but we keep it above
// zero so they never finish this row recursively.
@@ -1692,12 +1704,41 @@ function finishSuspenseListRow(request: Request, row: SuspenseListRow): void {
}
}
function tryToResolveTogetherRow(
request: Request,
togetherRow: SuspenseListRow,
): void {
// If we have a "together" row and all the pendingTasks are really the boundaries themselves,
// and we won't outline any of them then we can unblock this row early so that we can inline
// all the boundaries at once.
const boundaries = togetherRow.boundaries;
if (boundaries === null || togetherRow.pendingTasks !== boundaries.length) {
return;
}
let allCompleteAndInlinable = true;
for (let i = 0; i < boundaries.length; i++) {
const rowBoundary = boundaries[i];
if (
rowBoundary.pendingTasks !== 1 ||
rowBoundary.parentFlushed ||
isEligibleForOutlining(request, rowBoundary)
) {
allCompleteAndInlinable = false;
break;
}
}
if (allCompleteAndInlinable) {
unblockSuspenseListRow(request, togetherRow);
}
}
function createSuspenseListRow(
previousRow: null | SuspenseListRow,
): SuspenseListRow {
const newRow: SuspenseListRow = {
pendingTasks: 1, // At first the row is blocked on attempting rendering itself.
boundaries: null,
together: false,
next: null,
};
if (previousRow !== null && previousRow.pendingTasks > 0) {
@@ -1978,7 +2019,27 @@ function renderSuspenseList(
}
if (revealOrder === 'together') {
// TODO
const prevKeyPath = task.keyPath;
const prevRow = task.row;
const newRow = (task.row = createSuspenseListRow(null));
// This will cause boundaries to block on this row, but there's nothing to
// unblock them. We'll use the partial flushing pass to unblock them.
newRow.boundaries = [];
newRow.together = true;
task.keyPath = keyPath;
renderNodeDestructive(request, task, children, -1);
if (--newRow.pendingTasks === 0) {
finishSuspenseListRow(request, newRow);
}
task.keyPath = prevKeyPath;
task.row = prevRow;
if (prevRow !== null && newRow.pendingTasks > 0) {
// If we are part of an outer SuspenseList and our row is still pending, then that blocks
// the parent row from completing. We can continue the chain.
prevRow.pendingTasks++;
newRow.next = prevRow;
}
return;
}
// For other reveal order modes, we just render it as a fragment.
const prevKeyPath = task.keyPath;
@@ -4759,6 +4820,8 @@ function finishedTask(
if (row !== null) {
if (--row.pendingTasks === 0) {
finishSuspenseListRow(request, row);
} else if (row.together) {
tryToResolveTogetherRow(request, row);
}
}
request.allPendingTasks--;
@@ -4848,6 +4911,10 @@ function finishedTask(
}
}
}
const boundaryRow = boundary.row;
if (boundaryRow !== null && boundaryRow.together) {
tryToResolveTogetherRow(request, boundaryRow);
}
}
}
@@ -5604,6 +5671,20 @@ function flushPartialBoundary(
}
completedSegments.splice(0, i);
const row = boundary.row;
if (row !== null && row.together && boundary.pendingTasks === 1) {
// "together" rows are blocked on their own boundaries.
// We have now flushed all the boundary's segments as partials.
// We can now unblock it from blocking the row that will eventually
// unblock the boundary itself which can issue its complete instruction.
// TODO: Ideally the complete instruction would be in a single <script> tag.
if (row.pendingTasks === 1) {
unblockSuspenseListRow(request, row);
} else {
row.pendingTasks--;
}
}
return writeHoistablesForBoundary(
destination,
boundary.contentState,