React setState 源码实现理解

Q1

setState改变状态之后,不会立即更新state值。所以,如果改变state值,react是什么时候进行组件的更新呢?setState()到底做了一些什么呢?

A1

1. react生命周期

react生命周期

2. react更新state具体做了什么

引入一段源码

react中定义的setState方法,定义了两个参数(partialState,callback)。

partialState: 新的state值;
callback: 回调函数。

setState

getInternalInstanceReadyForUpdate方法的目的是获取当前组件对象,将其赋值给internalInstance变量。接下来判断当前组件对象的state更新队列是否存在,如果存在则将partialState也就是新的state值加入队列;如果不存在,则创建该对象的更新队列。然后进入enqueueUpdate方法。

enqueueSetState

enqueueCallback也是先获取当前组件对象,如果已经存在其他回调,就加入等待回调队列,如果当前没有回调,就创建等待回调队列。然后进入enqueueUpdate方法。

可以发现,enqueueSetState&enqueueCallback最终都是进入enqueueUpdate方法。下面我们来看看enqueueUpdate方法。

enqueueCallback

官方注解是:给组件做个标记:需要重新渲染,并且将可选的回调函数添加到函数列表中,这些函数将在重新渲染的时候执行。

我们看一下函数具体做了哪些事。发现这个函数只是做了一个判断:如果batchingStrategy.isBatchingUpdates为false,就执行batchingStrategy.batchedUpdates(enqueueUpdate,component),否则就加入dirtyComponents。

这里提到batchingStrategy,批量更新策略。

enqueueUpdate

批量更新策略是什么呢?看代码发现batchingStrategy批量更新策略只是一个简单的对象,定义了一个 isBatchingUpdates 的布尔值和一个 batchedUpdates 方法。默认isBatchingUpdates(下面称为更新标志)为false,然后会进入batchedUpdates方法,先把更新标志isBatchingUpdates设为true,然后执行transaction.perform(callback),即transaction.perform(enqueueUpdate)。

React内部采用了"状态机"的概念,组件处于不同的状态时,所执行的逻辑也并不相同。以组件更新流程为例,React以事务+状态的形式对组件进行更新。

batching

通过上面的一部分代码,我们发现setState()方法主要是enqueueUpdate()进行状态更新,怎样进行状态更新呢?定义了一个批量更新策略:判断更新标志isBatchingUpdates的值,如果为false,调用batchedUpdates()-->(先把更新标志isBatchingUpdates改为true,然后调用transaction.perform(enqueueUpdate))。如果为true,就把组件加入dirtyComponents数组中。

###2

React内部采用了"状态机"的概念,组件处于不同的状态时,所执行的逻辑也并不相同。以组件更新流程为例,React以事务+状态的形式对组件进行更新,因此接下来我们看看事务的机制。

3. transaction 事务

                    wrappers (injected at creation time)
                                   +        +
                                   |        |
                 +-----------------|--------|--------------+
                 |                 v        |              |
                 |      +---------------+   |              |
                 |   +--|    wrapper1   |---|----+         |
                 |   |  +---------------+   v    |         |
                 |   |          +-------------+  |         |
                 |   |     +----|   wrapper2  |--------+   |
                 |   |     |    +-------------+  |     |   |
                 |   |     |                     |     |   |
                 |   v     v                     v     v   | wrapper
                 | +---+ +---+   +---------+   +---+ +---+ | invariants
perform(anyMethod) | |   | |   |   |         |   |   | |   | | maintained
+----------------->|-|---|-|---|-->|anyMethod|---|---|-|---|-|-------->
                 | |   | |   |   |         |   |   | |   | |
                 | |   | |   |   |         |   |   | |   | |
                 | |   | |   |   |         |   |   | |   | |
                 | +---+ +---+   +---------+   +---+ +---+ |
                 |  initialize                    close    |
                 +-----------------------------------------+

这是官方代码的解析图。

可以看出调用函数是perform(anyMethod),然后方法anyMethod被wrapper包裹了,wrapper依次执行了initialize->anyMethod->close

function anyMethod(){
    console.log('xx')
};
transaction.perform(anyMethod);

代码的执行顺序是

initialize()
输出xx
close()

所以这里wrapper是怎样定义的呢?

wrapper

第二个wrapper比较简单,先来看一下第二个wrapper。

第二个wrapper(RESET_BATCHED_UPDATES)的作用是将更新标志isBatchingUpdates重置为false;我的理解这里是收集完所有要更新的state值,都加入_pendingStateQueue待更新状态队列了,然后组件更新完了之后,将更新标志重置为false,等待下次更新。然后下面来看一下第一个wrapper。

wwrapper2

5&f=png&s=54081)

第一个wrapper主要的作用是更新组件,执行了ReactUpdates.flushBatchedUpdates.bind(ReactUpdates)。

wrapper2
runBatchedUpdates
updateIfNecessary

可以看到flushBatchedUpdates方法循环遍历所有的dirtyComponents,又通过事务的形式调用runBatchedUpdates方法。

一共做了两件事:

  • 一是通过执行updateComponent方法来更新组件
  • 二是若setState方法传入了回调函数则将回调函数存入callbackQueue队列。

然后看一下updateComponent方法,官方注释是:更新组件,会调用shouldComponentUpdate,然后调用剩余的生命周期函数,更新DOM结构

annotation
updateComponent
pendingstate

这里终于更新了组件。看代码会发现在shouldComponentUpdate之前,执行了_processPendingState方法,该方法主要对state进行处理:

  • 1.如果更新队列为null,那么返回原来的state;
  • 2.如果更新队列有一个更新,那么返回更新值;
  • 3.如果更新队列有多个更新,那么通过for循环将它们合并;

综上说明了,在一个生命周期内,在componentShouldUpdate执行之前,所有的state变化都会被合并,最后统一处理。

wrapper2s

4. 回顾上述问题

综上,

  • setState()为啥没有立即更新this.state值呢
  • 如果在componentDidMount()中连续多次setState,无法进行state累加呢
  • 批量更新策略isBatchingStrategy干了什么,怎么做到更新的呢

那按照上述说的批量更新,第一次setState-->进入enqueueUpdate()-->此时isBatchingUpdates默认为false-->batchedUpdates(enqueueUpdate,...)-->设置isBatchingUpdates为true;transaction.perform(enqueueUpdates);-->(第一个wrapper:FLUSH_BATCHED_UPDATES)组件更新-->(第二个wrapper:RESET_BATCHED_UPDATES的close方法)设置isBatchingUpdates为false-->第二次setState-->isBatchingUpdates为false-->..-->组件更新-->isBatchingUpdates恢复为false。

这样和结果不对呀?按上述逻辑的话,岂不是每次setState都会更新this.state的值?

调试代码会发现,原来整个将 React 组件渲染到 DOM 中的过程就处于一个大的 Transaction 中。

result

在进入生命周期之前,就会调用batchedUpdates(),所以此时isBatchingUpdates已经修改为true了。后面第一次进入setState()时,就会进入加入dirtyComponent中。所以这也就是为什么两次打印 this.state.foods 都是 '' 的原因,新的 state 还没有被应用到组件中。

5. 总结

  • setState(partialState, callback),不会立即更新state值。setState()会触发页面的重新渲染,渲染之前,要合并所有的state变化后,state值才会更新。
  • setState(partialState, callback): callback会在所有状态更新之后再调用(demo中state的foods&drinks全部更新之后才会调用)
  • 事务这么有用,那我们可以调用事务吗?答案是不可以。
  • 另外在componentWillMount里面setState()不会触发重新渲染
willMount

Q2

在render函数里,无法setState

A2

在render函数中不能setState()。

从react生命周期可以看出:state更新会重新触发render(),所以会导致setState()-->re-render()-->setState()--re-render()-->...-->setState()-->re-render(),一直循环往复。

react生命周期

所以,同理在state更新的生命周期的函数中(componentWillUpdate/componentDidUpdate),都不能setState()

参考资料

https://juejin.im/post/59cc4c4bf265da0648446ce0#heading-0

https://zh-hans.reactjs.org/docs/react-component.html#setstate

https://www.imooc.com/article/22426

https://segmentfault.com/a/1190000003969996

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,539评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,911评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,337评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,723评论 1 290
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,795评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,762评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,742评论 3 416
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,508评论 0 271
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,954评论 1 308
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,247评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,404评论 1 345
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,104评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,736评论 3 324
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,352评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,557评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,371评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,292评论 2 352

推荐阅读更多精彩内容