fix: useUnmountPromise stops on immediate update

streamich · Jan 29, 2021 · 9ff5c09 · 9ff5c09
1 parent 5425ba9
commit 9ff5c09
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/useUnmountPromise.ts b/src/useUnmountPromise.ts
@@ -1,10 +1,11 @@
-import { useMemo, useRef, useEffect } from 'react';
+import { useMemo, useRef } from 'react';
+import useEffectOnce from './useEffectOnce';
 
 export type Race = <P extends Promise<any>, E = any>(promise: P, onError?: (error: E) => void) => P;
 
 const useUnmountPromise = (): Race => {
   const refUnmounted = useRef(false);
-  useEffect(() => () => {
+  useEffectOnce(() => () => {
     refUnmounted.current = true;
   });