Fix for issue developit#169

tumerorkun · Mar 10, 2021 · e531375 · e531375
1 parent f3a4e99
commit e531375
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/src/integrations/preact.js b/src/integrations/preact.js
@@ -24,7 +24,7 @@ export function connect(mapStateToProps, actions) {
 			const store = context.store;
 			let state = mapStateToProps(store ? store.getState() : {}, props);
 			const boundActions = actions ? mapActions(actions, store) : { store };
-			let update = () => {
+			const update = () => {
 				let mapped = mapStateToProps(store ? store.getState() : {}, props);
 				for (let i in mapped) if (mapped[i]!==state[i]) {
 					state = mapped;
@@ -35,13 +35,11 @@ export function connect(mapStateToProps, actions) {
 					return this.setState({});
 				}
 			};
+			store.subscribe(update);
 			this.componentWillReceiveProps = p => {
 				props = p;
 				update();
 			};
-			this.componentDidMount = () => {
-				store.subscribe(update);
-			};
 			this.componentWillUnmount = () => {
 				store.unsubscribe(update);
 			};

diff --git a/src/integrations/react.js b/src/integrations/react.js
@@ -39,13 +39,11 @@ export function connect(mapStateToProps, actions) {
 					return this.forceUpdate();
 				}
 			};
+			store.subscribe(update);
 			this.UNSAFE_componentWillReceiveProps = p => {
 				props = p;
 				update();
 			};
-			this.componentDidMount = () => {
-				store.subscribe(update);
-			};
 			this.componentWillUnmount = () => {
 				store.unsubscribe(update);
 			};