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);//该是啥就是是啥
}));
}