fix: make useElmish hook work correctly in React strict mode (#25)

atheck · Sep 7, 2022 · 0e0e855 · 0e0e855
1 parent c3c2b66
commit 0e0e855
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/useElmish.ts b/src/useElmish.ts
@@ -64,8 +64,12 @@ function useElmish<TProps, TModel, TMessage extends MessageBase> ({ name, props,
     const propsRef = useRef(props);
     const isMountedRef = useRef(true);
 
-    useEffect(() => () => {
-        isMountedRef.current = false;
+    useEffect(() => {
+        isMountedRef.current = true;
+
+        return () => {
+            isMountedRef.current = false;
+        };
     }, []);
 
     let initializedModel = model;