使用store来优化React组件的方法

在使用 React 编写组件的时候，我们常常会碰到两个不同的组件之间需要共享状态情况，而通常的做法就是提升状态到父组件。但是这样做会有一个问题，就是尽管只有两个组件需要这个状态，但是因为把状态提到了父组件，那么在状态变化的时候，父组件以及其下面的所有子组件都会重新 render，如果你的父组件比较复杂，包含了其他很多子组件的话，就有可能引起性能问题。

Redux 通过把状态放在全局的 store 里，然后组件去订阅各自需要的状态，当状态发生变化的时候，只有那些订阅的状态发生变化的组件才重新 render，这样就避免了上面说的提升状态所带来的副作用。但是，当我们在写一个 React 组件库的时候，redux 加 react-redux 的组合可能就有点太重了。所以我们可以自己写一个简单的 store，来实现类似 Redux 的订阅模式。

参考 Redux 的实现来写一个简版的 createStore:

function createStore(initialState) {
 let state = initialState;
 const listeners = [];

 function setState(partial) {
 state = {
 ...state,
 ...partial,
 };
 for (let i = 0; i < listeners.length; i++) {
 listeners[i]();
 }
 }

 function getState() {
 return state;
 }

 function subscribe(listener) {
 listeners.push(listener);

 return function unsubscribe() {
 const index = listeners.indexOf(listener);
 listeners.splice(index, 1);
 };
 }

 return {
 setState,
 getState,
 subscribe,
 };
}

我们的 createStore 非常简单，算上空行也只有 33 行，总共暴露了 3 个方法，没有 Redux 里的 dispatch 和 reducer，直接通过 setState 方法改变状态。下面我们来用它一个计数器的例子。

class Counter extends React.Component {
 constructor(props) {
 super(props);

 // 初始化 store
 this.store = createStore({
 count: 0,
 });
 }

 render() {
 return (
 <div>
 <Buttons store={store} />
 <Result store={store} />
 </div>
 )
 }
}

class Buttons extends React.Component {
 handleClick = (step) => () => {
 const { store } = this.props;
 const { count } = store.getState();
 store.setState({ count: count + step });
 }

 render() {
 return (
 <div>
 <button onClick={this.handleClick(1)}>+</button>
 <button onClick={this.handleClick(1)}>-</button>
 </div>
 );
 }
}

class Result extends React.Component {
 constructor(props) {
 super(props);

 this.state = {
 count: props.store.getState().count,
 };
 }

 componentDidMount() {
 this.props.store.subscribe(() => {
 const { count } = this.props.store.getState();
 if (count !== this.state.count) {
 this.setState({ count });
 }
 });
 }

 render() {
 return (
 <div>{this.state.count}</div>
 );
 };
}

例子中 Buttons 里通过 store.setState 来改变 store 中的状态，并不会引起整个 Counter 的重新 render，但是因为 Result 中订阅了 store 的变化，所以当 count 有变化的时候就可以通过改变自己组件内的状态来重新 render，这样就巧妙地避免了不必须要的 render。

最后，上面的 createStore 虽然只有几十行代码，我还是把它写成了一个叫 mini-store 库放在 GitHub 上，并且提供了类似 Redux 的 Provider 和 connect 方法，总共加起来也就 100 多行代码。如果你也在写 React 组件库，需要管理一个复杂组件的状态，不妨试试这个优化方式。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。