pass children to StreamedHydration

vikejs · Dec 2, 2023 · f46e324 · f46e324
1 parent 2f8beb4
commit f46e324
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/packages/vike-react-query/renderer/StreamedHydration.tsx b/packages/vike-react-query/renderer/StreamedHydration.tsx
@@ -3,6 +3,7 @@ export { StreamedHydration }
 import type { QueryClient } from '@tanstack/react-query'
 import { dehydrate, hydrate, DehydratedState } from '@tanstack/react-query'
 import { uneval } from 'devalue'
+import type { ReactNode } from 'react'
 import { useStream } from 'react-streaming'
 
 declare global {
@@ -18,7 +19,7 @@ declare global {
  * - hydrating the query client on the client
  * - if react-streaming is not used, it doesn't do anything
  */
-function StreamedHydration({ client }: { client: QueryClient }) {
+function StreamedHydration({ client, children }: { client: QueryClient; children: ReactNode }) {
   const stream = useStream()
 
   if (import.meta.env.SSR && stream) {
@@ -48,5 +49,5 @@ function StreamedHydration({ client }: { client: QueryClient }) {
     }
     window._rqd_ = { push: onEntry }
   }
-  return null
+  return children
 }
diff --git a/packages/vike-react-query/renderer/VikeReactQueryWrapper.tsx b/packages/vike-react-query/renderer/VikeReactQueryWrapper.tsx
@@ -20,12 +20,11 @@ export default function VikeReactQueryWrapper({ pageContext, children }: VikeRea
   const queryClient = __queryClientGlobal ?? new QueryClient(queryClientConfig)
 
   return (
-    <>
-      <StreamedHydration client={queryClient} />
-      <QueryClientProvider client={queryClient}>
-        <FallbackErrorBoundary>{children}</FallbackErrorBoundary>
-      </QueryClientProvider>
-    </>
+    <QueryClientProvider client={queryClient}>
+      <FallbackErrorBoundary>
+        <StreamedHydration client={queryClient}>{children}</StreamedHydration>
+      </FallbackErrorBoundary>
+    </QueryClientProvider>
   )
 }