fix(reactive-react): fix browser crash in strict-mode async linkages …

…scence
alibaba · Apr 28, 2021 · feb6487 · feb6487
1 parent 6bb3bed
commit feb6487
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/packages/reactive-react/src/hooks/useForceUpdate.ts b/packages/reactive-react/src/hooks/useForceUpdate.ts
@@ -10,11 +10,15 @@ export function useForceUpdate() {
   const unmountRef = useRef(false)
 
   const update = useCallback(() => {
+    if (unmountRef.current) return
+    setTick((tick) => {
+      return tick + 1
+    })
+  }, EMPTY_ARRAY)
+
+  const scheduler = useCallback(() => {
     if (RENDER_COUNT.value === 0) {
-      if (unmountRef.current) return
-      setTick((tick) => {
-        return tick + 1
-      })
+      update()
     } else {
       if (!RENDER_QUEUE.has(update)) {
         RENDER_QUEUE.add(update)
@@ -42,5 +46,5 @@ export function useForceUpdate() {
     }
   }, [])
 
-  return update
+  return scheduler
 }