fix: Next.js Error document is not defined

StructureBuilder · Aug 12, 2019 · 12ee736 · 12ee736
1 parent 2fda562
commit 12ee736
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "react-keep-alive",
-  "version": "2.3.0",
+  "version": "2.4.0",
   "description": "Package will allow components to maintain their status, to avoid repeated re-rendering.",
   "author": "Shen Chang",
   "homepage": "https://github.com/StructureBuilder/react-keep-alive",

diff --git a/src/components/Provider.tsx b/src/components/Provider.tsx
@@ -49,7 +49,7 @@ export interface IKeepAliveProviderProps {
 export default class KeepAliveProvider extends React.PureComponent<IKeepAliveProviderProps> implements IKeepAliveProviderImpl {
   public static displayName = keepAliveProviderTypeName;
 
-  public storeElement = createStoreElement();
+  public storeElement: HTMLElement;
 
   // Sometimes data that changes with setState cannot be synchronized, so force refresh
   public cache: ICache = Object.create(null);
@@ -64,6 +64,11 @@ export default class KeepAliveProvider extends React.PureComponent<IKeepAlivePro
 
   public providerIdentification: string = createUniqueIdentification();
 
+  public componentDidMount() {
+    this.storeElement = createStoreElement();
+    this.forceUpdate();
+  }
+
   public componentDidUpdate() {
     if (this.needRerender) {
       this.needRerender = false;
@@ -124,6 +129,9 @@ export default class KeepAliveProvider extends React.PureComponent<IKeepAlivePro
       include,
       exclude,
     } = this.props;
+    if (!storeElement) {
+      return null;
+    }
     return (
       <KeepAliveContext.Provider
         value={{