feat: support preRender

alibaba · Nov 29, 2023 · 3f89726 · 3f89726
1 parent f926df2
commit 3f89726
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 1 deletion.
diff --git a/packages/ice/src/middlewares/renderMiddleware.ts b/packages/ice/src/middlewares/renderMiddleware.ts
@@ -48,9 +48,22 @@ export default function createRenderMiddleware(options: Options): Middleware {
           req,
           res,
         };
+
+        // Simulating window object for wormhole.
+        // @ts-ignore
+        global.window = {};
+
         serverModule.renderToResponse(requestContext, {
           renderMode,
           documentOnly,
+          preRender: true,
+          streamOptions: {
+            onAllReady: (params) => {
+              console.log('middleware onAllReady');
+              console.log('params', params);
+              res.write(params.renderAssets.join(','));
+            },
+          },
         });
       }
     } else {

diff --git a/packages/runtime/src/runServerApp.tsx b/packages/runtime/src/runServerApp.tsx
@@ -42,6 +42,7 @@ export interface RenderOptions {
   documentDataLoader?: DocumentDataLoaderConfig;
   Document?: DocumentComponent;
   documentOnly?: boolean;
+  preRender?: boolean;
   renderMode?: RenderMode;
   // basename is used both for server and client, once set, it will be sync to client.
   basename?: string;

diff --git a/packages/runtime/src/server/streamRender.tsx b/packages/runtime/src/server/streamRender.tsx
@@ -32,9 +32,18 @@ export function renderToNodeStream(
   renderToNodeStreamOptions: RenderToNodeStreamOptions,
 ): NodeWritablePiper {
   return (res, options) => {
+    const {
+      renderOptions,
+    } = renderToNodeStreamOptions;
+    const {
+      preRender = false,
+    } = renderOptions;
+    console.log('preRender', preRender);
     const { pipe } = ReactDOMServer.renderToPipeableStream(element, {
       onShellReady() {
-        pipe(res);
+        if (!preRender) {
+          pipe(res);
+        }
         options?.onShellReady && options.onShellReady();
       },
       onShellError(error) {
@@ -66,6 +75,10 @@ export function renderToNodeStream(
         options?.onAllReady && options?.onAllReady({
           renderAssets,
         });
+
+        if (preRender) {
+          pipe(res);
+        }
       },
     });
   };