@tramvai/state 1.35.8 → 1.38.0

package/lib/index.es.js

@@ -633,48 +633,45 @@ const useIsomorphicLayoutEffect = typeof window !== 'undefined' ? useLayoutEffec
 
 function useStore(reducer) {
     const context = useConsumerContext();
-    const stateRef = useRef();
     const reducerRef = useRef(reducer);
     const addedReducerRef = useRef(null);
     const unsubscribeRef = useRef(noop);
     const [, forceRender] = useReducer((s) => s + 1, 0);
-    useMemo(() => {
-        // отписываемся от обновлений текущего редьюсера
-        unsubscribeRef.current();
-        // если мы получили новый редьюсер
-        if (reducer.storeName !== reducerRef.current.storeName) {
+    // если текущий редьюсер не зарегистрирован в диспетчере,
+    // регистрируем его вручную, что бы гарантировать работоспособность `context.getState(reducer)`,
+    // и сохраняем в `addedReducerRef`, что бы удалить при unmount
+    if (!context.hasStore(reducer)) {
+        context.registerStore(reducer);
+        addedReducerRef.current = reducer.storeName;
+    }
+    const stateRef = useRef(context.getState(reducer));
+    useIsomorphicLayoutEffect(() => {
+        const subscribe = (updatedState) => {
+            // если состояние текущего редьюсера изменилось,
+            // обновляем локальное состояние и ререндерим компонент
+            if (stateRef.current !== updatedState) {
+                stateRef.current = updatedState;
+                forceRender();
+            }
+        };
+        // сразу обновляем состояние
+        subscribe(context.getState(reducer));
+        // и подписываемся на обновления редьюсера
+        unsubscribeRef.current = context.subscribe(reducer, subscribe);
+        // заменяем текущий редьюсер
+        reducerRef.current = reducer;
+        return () => {
+            // гарантируем отписку от обновлений текущего редьюсера,
+            // при анмаунте компонента
+            unsubscribeRef.current();
             // если текущий редьюсер был зарегистрирован в диспетчере в этом хуке,
             // удаляем его из диспетчера
-            if (addedReducerRef.current === reducerRef.current.storeName) {
+            if (addedReducerRef.current) {
                 context.unregisterStore(reducerRef.current);
                 addedReducerRef.current = null;
             }
-            // заменяем текущий редьюсер
-            reducerRef.current = reducer;
-        }
-        // если текущий редьюсер не зарегистрирован в диспетчере,
-        // регистрируем его вручную
-        if (!context.hasStore(reducerRef.current)) {
-            context.registerStore(reducerRef.current);
-            addedReducerRef.current = reducerRef.current.storeName;
-        }
-        // подписываемся на обновления текущего редьюсера,
-        // если состояние текущего редьюсера изменилось,
-        // ререндерим компонент
-        unsubscribeRef.current = context.subscribe(reducerRef.current, (nextState) => {
-            stateRef.current = nextState;
-            forceRender();
-        });
-        // сохраняем состояние текущего редьюсера
-        stateRef.current = context.getState(reducerRef.current);
-    }, [reducer, context]);
-    // гарантируем отписку от обновлений текущего редьюсера,
-    // при анмаунте компонента
-    useIsomorphicLayoutEffect(() => {
-        return () => {
-            unsubscribeRef.current();
         };
-    }, []);
+    }, [reducer, context]);
     return stateRef.current;
 }
 

package/lib/index.js

@@ -656,48 +656,45 @@ const useIsomorphicLayoutEffect = typeof window !== 'undefined' ? React.useLayou
 
 function useStore(reducer) {
     const context = useConsumerContext();
-    const stateRef = React.useRef();
     const reducerRef = React.useRef(reducer);
     const addedReducerRef = React.useRef(null);
     const unsubscribeRef = React.useRef(noop__default["default"]);
     const [, forceRender] = React.useReducer((s) => s + 1, 0);
-    React.useMemo(() => {
-        // отписываемся от обновлений текущего редьюсера
-        unsubscribeRef.current();
-        // если мы получили новый редьюсер
-        if (reducer.storeName !== reducerRef.current.storeName) {
+    // если текущий редьюсер не зарегистрирован в диспетчере,
+    // регистрируем его вручную, что бы гарантировать работоспособность `context.getState(reducer)`,
+    // и сохраняем в `addedReducerRef`, что бы удалить при unmount
+    if (!context.hasStore(reducer)) {
+        context.registerStore(reducer);
+        addedReducerRef.current = reducer.storeName;
+    }
+    const stateRef = React.useRef(context.getState(reducer));
+    useIsomorphicLayoutEffect(() => {
+        const subscribe = (updatedState) => {
+            // если состояние текущего редьюсера изменилось,
+            // обновляем локальное состояние и ререндерим компонент
+            if (stateRef.current !== updatedState) {
+                stateRef.current = updatedState;
+                forceRender();
+            }
+        };
+        // сразу обновляем состояние
+        subscribe(context.getState(reducer));
+        // и подписываемся на обновления редьюсера
+        unsubscribeRef.current = context.subscribe(reducer, subscribe);
+        // заменяем текущий редьюсер
+        reducerRef.current = reducer;
+        return () => {
+            // гарантируем отписку от обновлений текущего редьюсера,
+            // при анмаунте компонента
+            unsubscribeRef.current();
             // если текущий редьюсер был зарегистрирован в диспетчере в этом хуке,
             // удаляем его из диспетчера
-            if (addedReducerRef.current === reducerRef.current.storeName) {
+            if (addedReducerRef.current) {
                 context.unregisterStore(reducerRef.current);
                 addedReducerRef.current = null;
             }
-            // заменяем текущий редьюсер
-            reducerRef.current = reducer;
-        }
-        // если текущий редьюсер не зарегистрирован в диспетчере,
-        // регистрируем его вручную
-        if (!context.hasStore(reducerRef.current)) {
-            context.registerStore(reducerRef.current);
-            addedReducerRef.current = reducerRef.current.storeName;
-        }
-        // подписываемся на обновления текущего редьюсера,
-        // если состояние текущего редьюсера изменилось,
-        // ререндерим компонент
-        unsubscribeRef.current = context.subscribe(reducerRef.current, (nextState) => {
-            stateRef.current = nextState;
-            forceRender();
-        });
-        // сохраняем состояние текущего редьюсера
-        stateRef.current = context.getState(reducerRef.current);
-    }, [reducer, context]);
-    // гарантируем отписку от обновлений текущего редьюсера,
-    // при анмаунте компонента
-    useIsomorphicLayoutEffect(() => {
-        return () => {
-            unsubscribeRef.current();
         };
-    }, []);
+    }, [reducer, context]);
     return stateRef.current;
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tramvai/state",
-  "version": "1.35.8",
+  "version": "1.38.0",
   "description": "",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
@@ -20,7 +20,7 @@
   "dependencies": {
     "@tinkoff/react-hooks": "0.0.23",
     "@tinkoff/utils": "^2.1.2",
-    "@tramvai/types-actions-state-context": "1.35.8",
+    "@tramvai/types-actions-state-context": "1.38.0",
     "@types/hoist-non-react-statics": "^3.3.1",
     "invariant": "^2.2.4",
     "react-is": "^17.0.1",