diff options
Diffstat (limited to 'src/client/main.tsx')
| -rw-r--r-- | src/client/main.tsx | 22 |
1 files changed, 18 insertions, 4 deletions
diff --git a/src/client/main.tsx b/src/client/main.tsx index a1d296a..b9296f4 100644 --- a/src/client/main.tsx +++ b/src/client/main.tsx @@ -1,9 +1,19 @@ -import { StrictMode } from "react"; +import { Provider, useStore } from "jotai/react"; +import { useHydrateAtoms } from "jotai/react/utils"; +import { queryClientAtom } from "jotai-tanstack-query"; +import { type ReactNode, StrictMode } from "react"; import { createRoot } from "react-dom/client"; import { App } from "./App"; import { StoreInitializer } from "./components/StoreInitializer"; +import { queryClient } from "./queryClient"; import "./styles.css"; +function HydrateQueryClient({ children }: { children: ReactNode }) { + const store = useStore(); + useHydrateAtoms([[queryClientAtom, queryClient]], { store }); + return <>{children}</>; +} + const rootElement = document.getElementById("root"); if (!rootElement) { throw new Error("Root element not found"); @@ -11,8 +21,12 @@ if (!rootElement) { createRoot(rootElement).render( <StrictMode> - <StoreInitializer> - <App /> - </StoreInitializer> + <Provider> + <HydrateQueryClient> + <StoreInitializer> + <App /> + </StoreInitializer> + </HydrateQueryClient> + </Provider> </StrictMode>, ); |