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

时间:2023-05-06 10:27:26

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

}));

}