feat: capture initialState for getServerState in react

this avoids hydration mismatches when updates happen to the store state between ssr and csr
pmndrs · Dec 29, 2023 · 2d868c6 · 2d868c6
1 parent 6d046fe
commit 2d868c6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/react.ts b/src/react.ts
@@ -104,6 +104,11 @@ const createImpl = <T>(createState: StateCreator<T, [], []>) => {
   const api =
     typeof createState === 'function' ? createStore(createState) : createState
 
+  if (!api.getServerState) {
+    const initialState = api.getState()
+    api.getServerState = () => initialState
+  }
+
   const useBoundStore: any = (selector?: any, equalityFn?: any) =>
     useStore(api, selector, equalityFn)