由实际问题探究setState的执行机制

一.几个开发中经常会遇到的问题

以下几个问题是我们在实际开发中经常会遇到的场景，下面用几个简单的示例代码来还原一下。

1.setState是同步还是异步的，为什么有的时候不能立即拿到更新结果而有的时候可以?

1.1 钩子函数和React合成事件中的 setState

现在有两个组件

1.  componentDidMount() {

2.    console.log('parent componentDidMount');

3.  }

4.  render() {

5.    return (

6.      <div>

7.        <SetState2></SetState2>

8.        <SetState></SetState>

9.      </div>

10.    );

11.  }

组件内部放入同样的代码，并在 Setstate1中的 componentDidMount中放入一段同步延时代码，打印延时时间：

1.  componentWillUpdate() {

2.    console.log('componentWillUpdate');

3.  }

4.  componentDidUpdate() {

5.    console.log('componentDidUpdate');

6.  }

7.  componentDidMount() {

8.    console.log('SetState调用setState');

9.    this.setState({

10.      index: this.state.index + 1

11.    })

12.    console.log('state', this.state.index);

13.    console.log('SetState调用setState');

14.    this.setState({

15.      index: this.state.index + 1

16.    })

17.    console.log('state', this.state.index);

18.  }

下面是执行结果：

说明：

• 1.调用 setState不会立即更新

• 2.所有组件使用的是同一套更新机制，当所有组件 didmount后，父组件 didmount，然后执行更新

• 3.更新时会把每个组件的更新合并，每个组件只会触发一次更新的生命周期。

1.2 异步函数和原生事件中的 setstate？

在 setTimeout中调用 setState（例子和在浏览器原生事件以及接口回调中执行效果相同）

1.  componentDidMount() {

2.    setTimeout(() => {

3.      console.log('调用setState');

4.      this.setState({

5.        index: this.state.index + 1

6.      })

7.      console.log('state', this.state.index);

8.      console.log('调用setState');

9.      this.setState({

10.        index: this.state.index + 1

11.      })

12.      console.log('state', this.state.index);

13.    }, 0);

14.  }

执行结果：

说明：

• 1.在父组件 didmount后执行

• 2.调用 setState同步更新

2.为什么有时连续两次 setState只有一次生效？

分别执行以下代码：

1.  componentDidMount() {

2.    this.setState({ index: this.state.index + 1 }, () => {

3.      console.log(this.state.index);

4.    })

5.    this.setState({ index: this.state.index + 1 }, () => {

6.      console.log(this.state.index);

7.    })

8.  }

1.  componentDidMount() {

2.    this.setState((preState) => ({ index: preState.index + 1 }), () => {

3.      console.log(this.state.index);

4.    })

5.    this.setState(preState => ({ index: preState.index + 1 }), () => {

6.      console.log(this.state.index);

7.    })

8.  }

执行结果：

1. 1

2. 1

1. 2

2. 2

说明：

• 1.直接传递对象的 setstate会被合并成一次

• 2.使用函数传递 state不会被合并

二.setState执行过程

由于源码比较复杂，就不贴在这里了，有兴趣的可以去 github上 clone一份然后按照下面的流程图去走一遍。

1.流程图

• partialState： setState传入的第一个参数，对象或函数

• _pendingStateQueue：当前组件等待执行更新的 state队列

• isBatchingUpdates：react用于标识当前是否处于批量更新状态，所有组件公用

• dirtyComponent：当前所有处于待更新状态的组件队列

• transcation：react的事务机制，在被事务调用的方法外包装n个 waper对象，并一次执行： waper.init、被调用方法、 waper.close

• FLUSH_BATCHED_UPDATES：用于执行更新的 waper，只有一个 close方法

2.执行过程

对照上面流程图的文字说明，大概可分为以下几步：

• 1.将setState传入的 partialState参数存储在当前组件实例的state暂存队列中。

• 2.判断当前React是否处于批量更新状态，如果是，将当前组件加入待更新的组件队列中。

• 3.如果未处于批量更新状态，将批量更新状态标识设置为true，用事务再次调用前一步方法，保证当前组件加入到了待更新组件队列中。

• 4.调用事务的 waper方法，遍历待更新组件队列依次执行更新。

• 5.执行生命周期 componentWillReceiveProps。

• 6.将组件的state暂存队列中的 state进行合并，获得最终要更新的state对象，并将队列置为空。

• 7.执行生命周期 componentShouldUpdate，根据返回值判断是否要继续更新。

• 8.执行生命周期 componentWillUpdate。

• 9.执行真正的更新， render。

• 10.执行生命周期 componentDidUpdate。

三.总结

1.钩子函数和合成事件中：

在 react的生命周期和合成事件中， react仍然处于他的更新机制中，这时 isBranchUpdate为true。

按照上述过程，这时无论调用多少次 setState，都会不会执行更新，而是将要更新的 state存入 _pendingStateQueue，将要更新的组件存入 dirtyComponent。

当上一次更新机制执行完毕，以生命周期为例，所有组件，即最顶层组件 didmount后会将 isBranchUpdate设置为false。这时将执行之前累积的 setState。

2.异步函数和原生事件中

由执行机制看， setState本身并不是异步的，而是如果在调用 setState时，如果 react正处于更新过程，当前更新会被暂存，等上一次更新执行后在执行，这个过程给人一种异步的假象。

在生命周期，根据JS的异步机制，会将异步函数先暂存，等所有同步代码执行完毕后在执行，这时上一次更新过程已经执行完毕， isBranchUpdate被设置为false，根据上面的流程，这时再调用 setState即可立即执行更新，拿到更新结果。

3. partialState合并机制

我们看下流程中 _processPendingState的代码，这个函数是用来合并 state暂存队列的，最后返回一个合并后的 state。

1.  _processPendingState: function (props, context) {

2.    var inst = this._instance;

3.    var queue = this._pendingStateQueue;

4.    var replace = this._pendingReplaceState;

5.    this._pendingReplaceState = false;

6.    this._pendingStateQueue = null;

7.    if (!queue) {

8.      return inst.state;

9.    }

10.    if (replace && queue.length === 1) {

11.      return queue[0];

12.    }

13.    var nextState = _assign({}, replace ? queue[0] : inst.state);

14.    for (var i = replace ? 1 : 0; i < queue.length; i++) {

15.      var partial = queue[i];

16.      _assign(nextState, typeof partial === 'function' ? partial.call(inst, nextState, props, context) : partial);

17.    }

18.    return nextState;

19.  },

我们只需要关注下面这段代码：

1. _assign(nextState, typeof partial === 'function' ? partial.call(inst, nextState, props, context) : partial);

如果传入的是对象，很明显会被合并成一次：

1. Object.assign(

2.  nextState,

3.  {index: state.index+ 1},

4.  {index: state.index+ 1}

5. )

如果传入的是函数，函数的参数preState是前一次合并后的结果，所以计算结果是准确的。

4. componentDidMount调用 setstate

在componentDidMount()中，你 可以立即调用setState()。它将会触发一次额外的渲染，但是它将在浏览器刷新屏幕之前发生。这保证了在此情况下即使render()将会调用两次，用户也不会看到中间状态。谨慎使用这一模式，因为它常导致性能问题。在大多数情况下，你可以 在constructor()中使用赋值初始状态来代替。然而，有些情况下必须这样，比如像模态框和工具提示框。这时，你需要先测量这些DOM节点，才能渲染依赖尺寸或者位置的某些东西。

以上是官方文档的说明，不推荐直接在 componentDidMount直接调用 setState，由上面的分析： componentDidMount本身处于一次更新中，我们又调用了一次 setState，就会在未来再进行一次 render，造成不必要的性能浪费，大多数情况可以设置初始值来搞定。

当然在 componentDidMount我们可以调用接口，再回调中去修改 state，这是正确的做法。

当state初始值依赖dom属性时，在 componentDidMount中 setState是无法避免的。

5. componentWillUpdate componentDidUpdate

这两个生命周期中不能调用 setState。

由上面的流程图很容易发现，在它们里面调用 setState会造成死循环，导致程序崩溃。

6.推荐使用方式

在调用 setState时使用函数传递 state值，在回调函数中获取最新更新后的 state。