React中setState的怪异行为 ——setState没有即时生效

setState可以说是React中使用频率最高的一个函数了，我们都知道，React是通过管理状态来实现对组件的管理的，当this.setState()被调用的时候，React会重新调用render方法来重新渲染UI

但实际使用的时候，我们会发现，有时候我们setState之后，并没有立刻生效，例如我们看一下以下的示例代码

class Test extends Component {

constructor(props) {

super(props);

this.state = {

count: 0

};

}

componentDidMount() {

this.setState({count: this.state.count + 1});

console.log(this.state.count); // 输出0

this.setState({count: this.state.count + 1});

console.log(this.state.count); // 输出0

setTimeout(() => {

this.setState({count: this.state.count + 1});

console.log(this.state.count); // 输出2

this.setState({count: this.state.count + 1});

console.log(this.state.count); // 输出3

}, 0);

}

render() {

return <div> test </div>;

}

}

开发过程中我们会发现，在componentDidMount方法中，我们调用setState之后state的值并没有立即改变，但如果我们在setTimeOut里面调用，我们却能立刻就能获得更新，原因就在于react中的使用了基于事务（传送门，关于事务原理的解析）的异步更新机制，但对于这个异步的理解，又跟ajax的异步有所不同，因为毕竟react是一个js框架，所有的操作都是单线程的，所有的操作，都得按顺序来，那么它具体是怎么实现的呢？

我们都知道，对于dom的操作对性能的损耗是非常严重的，所以react为了提高整体的渲染性能，会将一次渲染周期中的state进行合并，在这个渲染周期中你对所有setState的所有调用都会被合并起来之后，再一次性的渲染，这样可以避免频繁的调用setState导致频繁的操作dom，提高渲染性能。具体的实现方面，可以简单的理解为react中存在一个状态变量isBatchingUpdates，当处于渲染周期开始时，这个变量会被设置成true，渲染周期结束时，会被设置成false，react会根据这个状态变量，当出在渲染周期中时，仅仅只是将当前的改变缓存起来，等到渲染周期结束时，再一次性的全部render,，具体的流程可以参照下面的流程图

现在，我们回到最开始的问题，为什么一开始在componentDidMount中直接执行setState会无法立刻得到更新呢，原因就在于，我们在componentDidMount中其实处于首次渲染的事务当中，这次事务的渲染尚未完成，而首次渲染的时候会将isBatchingUpdates设置为true，这是我们在componentDidMount中调用setState，react会发现当前事务尚未完成，只会直接将修改后的state放入到dirtyComponents中，等待最终渲染周期完成时，将所有的state进行合并，一次性render。而当我们放在setTimeOut里面的时候，setTimeOut会将操作放到执行队列的最后方，也就是说会等待渲染周期结束之后再进行setState，这个时候状态变量已经被重置回来了，所以此时我们的每一次setState都会立刻生效

接下来，我们从源码的角度来看看setState是怎么操作的

function enqueueUpdate(component) {

ensureInjected();

//不在渲染周期中

if (!batchingStrategy.isBatchingUpdates) {

batchingStrategy.batchedUpdates(enqueueUpdate, component);

return;

}

//渲染周期中，直接缓存state等待下一步更新

dirtyComponents.push(component);

}

这个逻辑跟上图的逻辑是一样的，当我们调用setState函数的时候，实际上最终会调用到enqueueUpdate函数，整体逻辑上面已经分析过了，就不再赘述，接下来看看setState是如何通过事务来进行渲染的，也就是batchingStrategy.batchedUpdates到底做了些什么，往下走之前，如果对事务不了解建议先看看这篇文章（传送门，关于事务原理的解析）

var ReactUpdates = require('ReactUpdates');

var Transaction = require('Transaction');

var emptyFunction = require('emptyFunction');

//第二个wrapper

var RESET_BATCHED_UPDATES = {

initialize: emptyFunction,

close: function() {

ReactDefaultBatchingStrategy.isBatchingUpdates = false;

},

};

//第一个wrapper

var FLUSH_BATCHED_UPDATES = {

initialize: emptyFunction,

close: ReactUpdates.flushBatchedUpdates.bind(ReactUpdates),

};

//wrapper列表

var TRANSACTION_WRAPPERS = [FLUSH_BATCHED_UPDATES, RESET_BATCHED_UPDATES];

//事务构造函数

function ReactDefaultBatchingStrategyTransaction() {

//原型中定义的初始化方法

this.reinitializeTransaction();

}

//继承原型

Object.assign(

ReactDefaultBatchingStrategyTransaction.prototype,

Transaction.Mixin,

{

getTransactionWrappers: function() {

return TRANSACTION_WRAPPERS;

},

}

);

//新建一个事务

var transaction = new ReactDefaultBatchingStrategyTransaction();

var ReactDefaultBatchingStrategy = {

isBatchingUpdates: false,

batchedUpdates: function(callback, a, b, c, d, e) {

var alreadyBatchingUpdates = ReactDefaultBatchingStrategy.isBatchingUpdates;

ReactDefaultBatchingStrategy.isBatchingUpdates = true;

// The code is written this way to avoid extra allocations

if (alreadyBatchingUpdates) {

callback(a, b, c, d, e);

} else {

//在这个地方调用事务，callback是从外部传入的方法

transaction.perform(callback, null, a, b, c, d, e);

}

},

};

上面这个就是react渲染所使用的事务，react就是用这个事务来处理setState引起的渲染，根据我们刚刚的解释，我们可以看到，事务开始时就把isBatchingUpdates设置成了true，防止在一次渲染周期中重复渲染，我们还可以看到这个事务定义了两个wrapper，其出口方法close分别用于执行渲染和设置状态变量，而执行渲染的FLUSH_BATCHED_UPDATES 要先于执行设置状态变量的RESET_BATCHED_UPDATES ，也就是说，执行渲染之后，才会通过RESET_BATCHED_UPDATES的close方法执行这句代码

ReactDefaultBatchingStrategy.isBatchingUpdates = false;

之后整个渲染周期结束。这时候当我们执行setState的时候，重新进入一个新的渲染周期

那么，问题来了，当我们在渲染周期中执行了setState之后，我们要如何获取到最新的state的值呢，setTimeOut是一个方案，但是不太优雅，有没有其他方法呢，我们注意到，setState提供了一个回调函数，我们只需要在回调里面获取更新后的state即可，像这样

componentDidMount() {

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

console.log(this.state.count);//该是啥就是是啥

}));

}