fix: Update useForceUpdate.ts

easy-form · Mar 14, 2024 · b1b3de2 · b1b3de2
1 parent d29e7c6
commit b1b3de2
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/use/useForceUpdate.ts b/src/use/useForceUpdate.ts
@@ -2,8 +2,14 @@ import { useState } from 'react';
 
 export default function useForceUpdate() {
   const [, setState] = useState({});
-  const forceUpdate = () => {
-    setState({});
+  const forceUpdate = (delay: boolean = true) => {
+    if (delay) {
+      setTimeout(() => {
+        setState({});
+      });
+    } else {
+      setState({});
+    }
   };
   return forceUpdate;
 }