feat: Store the updater first to avoid data update failure in extreme…

… cases

src/index.ts

@@ -43,9 +43,25 @@ export default function tinyStore<S extends {}, A>(
 
   const useStore = (source: boolean) => {
     const [, setState] = useState({})
+    const updaterRef = useRef<Updater<S>>()
     const stateProxy = useRef<Partial<S>>({})
     const stateCache = useRef<Partial<S>>({})
 
+    if (!updaterRef.current) {
+      updaterRef.current = (data: Partial<S>) => {
+        for (let key in data) {
+          if (key in stateCache.current && !equal(data[key], stateCache.current[key])) {
+            setState({}); break;
+          }
+        }
+      }
+      hooks.push(updaterRef.current)
+    }
+
+    useEffect(() => () => {
+      updaterRef.current && hooks.splice(hooks.indexOf(updaterRef.current), 1)
+    }, [])
+
     useMemo(() => {
       stateProxy.current = new Proxy({}, {
         get: (_, key: string) => {
@@ -60,20 +76,6 @@ export default function tinyStore<S extends {}, A>(
       })
     }, [])
 
-    useEffect(() => {
-      const updater: Updater<S> = (data: Partial<S>) => {
-        for (let key in data) {
-          if (key in stateCache.current && !equal(data[key], stateCache.current[key])) {
-            setState({}); break;
-          }
-        }
-      }
-      hooks.push(updater)
-      return () => {
-        hooks.splice(hooks.indexOf(updater), 1)
-      }
-    }, [])
-
     return stateProxy.current
   }