add noStreaming option

codehz · Jun 10, 2024 · bee95e7 · bee95e7
1 parent 23d02d4
commit bee95e7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/index.tsx b/index.tsx
@@ -45,12 +45,14 @@ export class StaticRouters {
         if (error instanceof ClientOnlyError) return;
         console.error(error, errorInfo);
       },
+      noStreaming,
     }: {
       Shell: React.ComponentType<{ children: React.ReactElement }>;
       preloadScript?: string;
       bootstrapModules?: string[];
       context?: T;
       onError?(error: unknown, errorInfo: React.ErrorInfo): string | void;
+      noStreaming?: boolean;
     }
   ): Promise<Response | null> {
     const { pathname, search } = new URL(request.url);
@@ -108,6 +110,9 @@ export class StaticRouters {
         onError,
       }
     );
+    if (noStreaming) {
+      await stream.allReady;
+    }
     return new Response(stream, {
       headers: {
         "Content-Type": "text/html; charset=utf-8",