使用 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 组件库，须要管理一个复杂组件的状态，不妨试试这个优化方式。