1. useEffect 简介

1.1 为什么要有 useEffect

我们在前文中说到 React Hooks 使得 Functional Component 拥有 Class Component 的特性,其主要动机包括:

  1. 在组件之间复用状态逻辑很难
  2. 复杂组件变得难以理解
  3. 难以理解的 class

对于第二点,首先,针对 Class Component 来说,我们写 React 应用时经常要在组件的各种生命周期中编写代码,如在 componentDidMountcomponentDidUpdate 中发送 HTTP 请求、事件绑定、甚至做一些额外的逻辑,使得业务逻辑扎堆在组件的生命周期函数中。在这个时候,我们的编程思路是“在组件装载完毕时我们需要做什么”、“在组件更新时我们需要做什么”,这使得 React 开发成为了面向生命周期编程,而我们在生命周期中写的那些逻辑,则成了组件生命周期函数的副作用

其次,面向生命周期编程会导致业务逻辑散乱在各生命周期函数里。比如,我们在 componentDidMount 进行的事件绑定又需要在 componentDidUnmount 解绑,那事件管理的逻辑就不统一,代码零散 review 起来会比较麻烦:

  1. import React from 'react'
  2. class A extends React.Componment {
  3. componmentDidMount() {
  4. document.getElementById('js_button')
  5. .addEventListener('click', this.log)
  6. }
  7. componentDidUnmount() {
  8. document.getElementById('js_button')
  9. .removeEventListener('click', this.log)
  10. }
  11. log = () => {
  12. console.log('log')
  13. }
  14. render() {
  15. return (
  16. <div id="js_button">button</div>
  17. )
  18. }
  19. }

useEffect 的出现,则让开发者的关注点从生命周期重新抽离出来聚焦在业务逻辑之上,其实 effect 的全称就是 side effect,即副作用,useEffect 就是用来处理原本生命周期函数里的副作用逻辑。

接下来,我们看看 useEffect 的用法。

1.2 useEffect 的用法

上面那段代码用 useEffect 改写之后如下:

  1. import React, { useEffect } from 'react'
  2. function A() {
  3. log() {
  4. console.log('log')
  5. }
  6. useEffect(() => {
  7. document
  8. .getElementById('js_button')
  9. .addEventListener('click', log)
  10. return () => {
  11. document
  12. .getElementById('js_button')
  13. .removeEventListener('click', log)
  14. }
  15. })
  16. return (<div id="js_button">button</div>)
  17. }

useEffect 接受两个参数,第一个参数是一个 function,其实现 bind 操作并将 unbind 作为一个 thunk 函数被返回。第二个参数是一个可选的 dependencies 数组,如果dependencies 不存在,那么 function 每次 render 都会执行;如果 dependencies 存在,只有当它发生了变化,function 才会执行。由此我们也可以推知,如果 dependencies 是一个空数组,那么当且仅当首次 render 的时候才会执行 function。

  1. useEffect(
  2. () => {
  3. const subscription = props.source.subscribe();
  4. return () => {
  5. subscription.unsubscribe();
  6. };
  7. },
  8. [props.source],
  9. );

更多用法请阅读 React 官网的 useEffect API 介绍: https://reactjs.org/docs/hooks-reference.html#useeffect

2. useEffect 的原理与简单实现

根据 useEffect 的用法,我们可以自己实现一个简单的 useEffect:

  1. let _deps;
  2. function useEffect(callback, dependencies) {
  3. const hasChanged = _deps
  4. && !dependencies.every((el, i) => el === _deps[i])
  5. || true;
  6. // 如果 dependencies 不存在,或者 dependencies 有变化,就执行 callback
  7. if (!dependencies || hasChanged) {
  8. callback();
  9. _deps = dependencies;
  10. }
  11. }

3. useEffect 源码解析

3.1 mountEffect & updateEffect

useEffect 的入口和上一节中 useState 的一样,都在 ReactFiberHooks.js 这个文件中,并且同 useState 一样,在首次加载时 useEffect 实际执行的是 mountEffect,之后每次渲染执行的是 updateEffect,此处不再赘述。那我们需要重点看看 mountEffect 和 updateEffect 实际做了什么。

对于 mountEffect:

  1. function mountEffect(
  2. create: () => (() => void) | void,
  3. deps: Array<mixed> | void | null,
  4. ): void {
  5. return mountEffectImpl(
  6. UpdateEffect | PassiveEffect,
  7. UnmountPassive | MountPassive,
  8. create,
  9. deps,
  10. );
  11. }

对于 updateEffect:

  1. function updateEffect(
  2. create: () => (() => void) | void,
  3. deps: Array<mixed> | void | null,
  4. ): void {
  5. return updateEffectImpl(
  6. UpdateEffect | PassiveEffect,
  7. UnmountPassive | MountPassive,
  8. create,
  9. deps,
  10. );
  11. }

mountEffect 和 updateEffect 的入参是一个 function 和一个 array,对应的就是我们前文 useEffect 传的 callback 和 deps。同时,我们可以发现 mountEffect 和 updateEffect 实际调用的是 mountEffectImpl 和 updateEffectImpl,它们接受的四个参数一模一样的,后面两个参数直接透传的不用说,主要是前面的 UpdateEffect | PassiveEffectUnmountPassive | MountPassive 究竟是什么?

阅读代码可知他们是从 ReactSideEffectTagsReactHookEffectTags 中引入的。

  1. import {
  2. Update as UpdateEffect,
  3. Passive as PassiveEffect,
  4. } from 'shared/ReactSideEffectTags';
  5. import {
  6. NoEffect as NoHookEffect,
  7. UnmountPassive,
  8. MountPassive,
  9. } from './ReactHookEffectTags';

看一下 ReactSideEffectTags.js 与 ReactHookEffectTags.js 中的定义:

  1. // Don't change these two values. They're used by React Dev Tools.
  2. export const NoEffect = /* */ 0b0000000000000;
  3. export const PerformedWork = /* */ 0b0000000000001;
  4. // You can change the rest (and add more).
  5. export const Placement = /* */ 0b0000000000010;
  6. export const Update = /* */ 0b0000000000100;
  7. export const PlacementAndUpdate = /* */ 0b0000000000110;
  8. export const Deletion = /* */ 0b0000000001000;
  9. export const ContentReset = /* */ 0b0000000010000;
  10. export const Callback = /* */ 0b0000000100000;
  11. export const DidCapture = /* */ 0b0000001000000;
  12. export const Ref = /* */ 0b0000010000000;
  13. export const Snapshot = /* */ 0b0000100000000;
  14. export const Passive = /* */ 0b0001000000000;
  15. export const Hydrating = /* */ 0b0010000000000;
  16. export const HydratingAndUpdate = /* */ 0b0010000000100;
  17. export const NoEffect = /* */ 0b00000000;
  18. export const UnmountSnapshot = /* */ 0b00000010;
  19. export const UnmountMutation = /* */ 0b00000100;
  20. export const MountMutation = /* */ 0b00001000;
  21. export const UnmountLayout = /* */ 0b00010000;
  22. export const MountLayout = /* */ 0b00100000;
  23. export const MountPassive = /* */ 0b01000000;
  24. export const UnmountPassive = /* */ 0b10000000;

这么设计是为了简化类型比较与类型复合,如果项目开发的过程中有过一些复合权限系统的设计经验,那么可能第一眼就能反应过来,所以 UnmountPassive | MountPassive 就是 0b11000000。如果对应的位为非零,则表示 tag 实现了指定的行为。这个在未来会用到,我们这里先不涉及,所以就先放在这里了解即可。

3.2 mountEffectImpl & updateEffectImpl

接着我们来看看 mountEffectImplupdateEffectImpl 的具体实现。

3.2.1 mountEffectImpl

首先是 mountEffectImpl

  1. function mountEffectImpl(fiberEffectTag, hookEffectTag, create, deps): void {
  2. const hook = mountWorkInProgressHook(); // 创建一个新的 Hook 并返回当前 workInProgressHook
  3. const nextDeps = deps === undefined ? null : deps;
  4. sideEffectTag |= fiberEffectTag;
  5. hook.memoizedState = pushEffect(hookEffectTag, create, undefined, nextDeps);
  6. }

mountWorkInProgressHook 我们在第 3 篇 4.3.3: mountWorkInProgressHook 中解析过,其就是创建一个新的 Hook 并返回当前 workInProgressHook,具体原理不再赘述。

sideEffectTag 是按位或上 fiberEffectTag 然后赋值,在 renderWithHooks 中挂载在 renderedWork.effectTag 上,并在每次渲染后重置为 0。

  1. renderedWork.effectTag |= sideEffectTag;
  2. sideEffectTag = 0;

具体 renderedWork.effectTag 有什么用,我们后续会说到。

renderWithHooks 在 第 3 篇 4.3.1: renderWithHooks 中解析过,此处不再赘述。

hook.memoizedState 记录 pushEffect 的返回结果,这个同记录 useState 中的 newState 的原理是一致的。那么现在的重点转移到了 pushEffect 究竟做了什么。

3.3.2 updateEffectImpl

接下来我们看看 updateEffectImpl 又做了些什么工作呢?

  1. function updateEffectImpl(fiberEffectTag, hookEffectTag, create, deps): void {
  2. const hook = updateWorkInProgressHook(); // 获取当前正在工作中的 Hook
  3. const nextDeps = deps === undefined ? null : deps;
  4. let destroy = undefined;
  5. if (currentHook !== null) {
  6. const prevEffect = currentHook.memoizedState;
  7. destroy = prevEffect.destroy;
  8. if (nextDeps !== null) {
  9. const prevDeps = prevEffect.deps;
  10. if (areHookInputsEqual(nextDeps, prevDeps)) {
  11. pushEffect(NoHookEffect, create, destroy, nextDeps);
  12. return;
  13. }
  14. }
  15. }
  16. sideEffectTag |= fiberEffectTag;
  17. hook.memoizedState = pushEffect(hookEffectTag, create, destroy, nextDeps);
  18. }

updateWorkInProgressHook 我们在第 3 篇 4.4.3: updateWorkInProgressHook 中解析过,其就是获取当前正在工作中的 Hook,具体原理不再赘述。

可以发现在 currentHook 为空的时候,updateEffectImpl 的逻辑与 mountEffectImpl 的逻辑是一模一样的;当 currentHook 不为空的时候,pushEffect 的第三个参数不是 undefined,而是 destroy。并且,在这个分支存在 areHookInputsEqual(nextDeps, prevDeps),即如果当前 useEffect 的 deps 和上一阶段的 useEffect 的 deps 相等(areHookInputsEqual 所做的事情就是遍历比较两个 deps 是否相等,这里就不展开解读了),那就执行 pushEffect(NoHookEffect, create, destroy, nextDeps);,大胆猜测 NoHookEffect 的意思就是不执行这次的 useEffect。如此,这段代码的逻辑就和我们之前自己实现的 useEffect 是一致的。

根据 第 3 篇 4.4.3: updateWorkInProgressHook,我们得知 currentHook 就是当前阶段正在处理的 Hook,其正常逻辑下不会为空。那我们接下来需要重点关注的应该是 pushEffect 做了什么,其第三个参数有什么含义?

3.3 pushEffect

  1. function pushEffect(tag, create, destroy, deps) {
  2. // 声明一个新的 effect
  3. const effect: Effect = {
  4. tag,
  5. create,
  6. destroy,
  7. deps,
  8. // Circular
  9. next: (null: any), // 函数组件中定义的下一个 effect 的引用
  10. };
  11. if (componentUpdateQueue === null) {
  12. componentUpdateQueue = createFunctionComponentUpdateQueue(); // 初始化 componentUpdateQueue
  13. componentUpdateQueue.lastEffect = effect.next = effect;
  14. } else {
  15. const lastEffect = componentUpdateQueue.lastEffect;
  16. if (lastEffect === null) {
  17. componentUpdateQueue.lastEffect = effect.next = effect;
  18. } else {
  19. const firstEffect = lastEffect.next;
  20. lastEffect.next = effect;
  21. effect.next = firstEffect;
  22. componentUpdateQueue.lastEffect = effect;
  23. }
  24. }
  25. return effect;
  26. }
  1. type Effect = {
  2. tag: HookEffectTag, // 一个二进制数,它将决定 effect 的行为
  3. create: () => (() => void) | void, // 绘制后应该运行的回调
  4. destroy: (() => void) | void, // 用于确定是否应销毁和重新创建 effect
  5. deps: Array<mixed> | null, // 决定重绘制后是否执行的 deps
  6. next: Effect, // 函数组件中定义的下一个 effect 的引用
  7. };

这个函数首先根据入参声明了一个新的 effect,数据结构也给出来了,它同样也是一个循环链表。tag 是

接下来根据 componentUpdateQueue 是否为空走两套逻辑,而 componentUpdateQueue 的结构其实很简单:

  1. export type FunctionComponentUpdateQueue = {
  2. lastEffect: Effect | null,
  3. };

可见,componentUpdateQueue 其实就是一个存储 Effect 的全局变量。

  1. componentUpdateQueue 为空:这种情况就是 mountEffect 时候的逻辑,它会创建一个空的 componentUpdateQueue,它其实只是 {lastEffect: null},之后将 componentUpdateQueue.lastEffect 指向 effect.next,其实就是存了一下 effect。
  2. componentUpdateQueue 不为空:这种情况就是 updateEffect 时候会走到的逻辑
  3. lastEffect 为空:这种情况是新的渲染阶段的第一个 useEffect,逻辑处理和 componentUpdateQueue 为空时一致。
  4. lastEffect 不为空:这种情况意味着这个组件有多个 useEffect,是第二个及其之后的 useEffect 会走到的分支,将 lastEffect 指向下一个 effect。

最后 return 一个 effect。

3.4 React Fiber 流程分析

看似源码到这里就结束了,但我们还存留几个问题没有解决:

  1. effect.tag 的那些二进制数是什么意思?
  2. pushEffect 之后还有什么逻辑?
  3. componentUpdateQueue 存储 Effect 之后会在哪里被用到?

renderWithHooks 中,componentUpdateQueue 会被赋值到 renderedWork.updateQueue 上,包括我们 3.2 中的 sideEffectTag 也会赋值到 renderedWork.effectTag 上。

  1. renderedWork.updateQueue = (componentUpdateQueue: any);
  2. renderedWork.effectTag |= sideEffectTag;

第 3 篇 4.3.1: renderWithHooks中,我们分析出 renderWithHooks 是在函数组件更新阶段(updateFunctionComponent)执行的函数,这里我们要想知道上面三个问题的答案,必须要把整个 Reconciler 的流程走一遍才能解析清楚。我个人认为 Fiber 是 React 16 中最复杂的一块逻辑了,所以在前面几篇中我只是略微提及,并没有展开篇幅解析。Fiber 里面的内容很多,如果展开的话足够写几篇文章了,因此这里也尽量简单快捷的走一遍流程,忽略本文不相关的细节,只梳理部分逻辑的实现,重点关注我们调用 useEffect 之后的逻辑。

注:如果对这部分不感兴趣的同学可以直接跳到 3.5 继续阅读。

React Fiber 优秀的文章有很多,这里再推荐阅读几篇文章和视频来帮助有兴趣的同学来了解

  1. A Cartoon Intro to Fiber - React Conf 2017
  2. React Fiber初探
  3. 这可能是最通俗的 React Fiber 打开方式

那我们开始吧!

3.4.1 ReactDOM.js

页面渲染的唯一入口便是 ReactDOM.render,

  1. ReactRoot.prototype.render = ReactSyncRoot.prototype.render = function(
  2. children: ReactNodeList,
  3. callback: ?() => mixed,
  4. ): Work {
  5. // ... 忽略无关代码
  6. updateContainer(children, root, null, work._onCommit);
  7. return work;
  8. };

render 的核心是调用 updateContainer,这个函数来自于 react-reconciler 中的 ReactFiberReconciler.js。

3.4.2 ReactFiberReconciler.js

这个文件其实也是 react-reconciler 的入口,我们先看看 updateContainer 究竟是什么:

  1. export function updateContainer(
  2. element: ReactNodeList,
  3. container: OpaqueRoot,
  4. parentComponent: ?React$Component<any, any>,
  5. callback: ?Function,
  6. ): ExpirationTime {
  7. // ... 忽略无关代码
  8. return updateContainerAtExpirationTime(
  9. element,
  10. container,
  11. parentComponent,
  12. expirationTime,
  13. suspenseConfig,
  14. callback,
  15. );
  16. }

忽略无关代码发现它其实只是 updateContainerAtExpirationTime 的一层封装,那我们看看这个是什么:

  1. export function updateContainerAtExpirationTime(
  2. element: ReactNodeList,
  3. container: OpaqueRoot,
  4. parentComponent: ?React$Component<any, any>,
  5. expirationTime: ExpirationTime,
  6. suspenseConfig: null | SuspenseConfig,
  7. callback: ?Function,
  8. ) {
  9. // ... 忽略无关代码
  10. return scheduleRootUpdate(
  11. current,
  12. element,
  13. expirationTime,
  14. suspenseConfig,
  15. callback,
  16. );
  17. }

再次忽略一些无关代码,发现它又是 scheduleRootUpdate 的一层封装……那我们再看看 scheduleRootUpdate 是什么:

  1. function scheduleRootUpdate(
  2. current: Fiber,
  3. element: ReactNodeList,
  4. expirationTime: ExpirationTime,
  5. suspenseConfig: null | SuspenseConfig,
  6. callback: ?Function,
  7. ) {
  8. // ... 忽略无关代码
  9. enqueueUpdate(current, update);
  10. scheduleWork(current, expirationTime);
  11. return expirationTime;
  12. }

忽略一小段无关代码,发现它的核心是做两件事,enqueueUpdate 我们这里暂时先不管,重点看看任务调度 scheduleWork,它相当于是 Fiber 逻辑的入口了,在 ReactFiberWorkLoop.js 中定义。

3.4.3 ReactFiberWorkLoop.js - render

ReactFiberWorkLoop.js 的内容非常长,有 2900 行代码,是包含任务循环主逻辑,不过我们刚才弄清楚要从 scheduleWork 开始着手那就慢慢梳理:

  1. export function scheduleUpdateOnFiber(
  2. fiber: Fiber,
  3. expirationTime: ExpirationTime,
  4. ) {
  5. // ... 忽略无关代码
  6. const priorityLevel = getCurrentPriorityLevel();
  7. if (expirationTime === Sync) {
  8. if (
  9. (executionContext & LegacyUnbatchedContext) !== NoContext &&
  10. (executionContext & (RenderContext | CommitContext)) === NoContext
  11. ) {
  12. schedulePendingInteractions(root, expirationTime);
  13. let callback = renderRoot(root, Sync, true);
  14. while (callback !== null) {
  15. callback = callback(true);
  16. }
  17. } else {
  18. scheduleCallbackForRoot(root, ImmediatePriority, Sync);
  19. if (executionContext === NoContext) {
  20. flushSyncCallbackQueue();
  21. }
  22. }
  23. } else {
  24. scheduleCallbackForRoot(root, priorityLevel, expirationTime);
  25. }
  26. // ... 忽略特殊情况的处理
  27. }
  28. export const scheduleWork = scheduleUpdateOnFiber;

其实这段代码大部分分支都会收回到 renderRoot 上,再对 renderRoot 的回调做 while 循环处理。所以我们与其说 scheduleWork 是 Fiber 逻辑的入口,不如说 renderRoot 是入口。renderRoot 就是大名鼎鼎的 Fiber 两个阶段中的 render 阶段。

React Hooks 源码解析(4):useEffect - 图1

图源 A Cartoon Intro to Fiber - React Conf 2017

其实 debug 一下也容易看出这两个阶段:

React Hooks 源码解析(4):useEffect - 图2

renderRoot 中的代码也非常复杂,我们重点关注和本文有关的逻辑:

  1. function renderRoot(
  2. root: FiberRoot,
  3. expirationTime: ExpirationTime,
  4. isSync: boolean,
  5. ): SchedulerCallback | null {
  6. if (isSync && root.finishedExpirationTime === expirationTime) {
  7. // There's already a pending commit at this expiration time.
  8. return commitRoot.bind(null, root); // 进入 commit 阶段
  9. }
  10. // ...
  11. do {
  12. try {
  13. if (isSync) {
  14. workLoopSync();
  15. } else {
  16. workLoop(); // 核心逻辑
  17. }
  18. break;
  19. } catch (thrownValue) {
  20. // ...
  21. } while (true);
  22. // ...
  23. }

把一些多余的代码略去之后,我们关注到两个重要的点:

  1. workLoop 是代码的核心部分,配合循环来实现任务循环。
  2. 在超时的情况下,会进入 commit 阶段。

我们先看看 workLoop 的逻辑:

  1. function workLoop() {
  2. while (workInProgress !== null && !shouldYield()) {
  3. workInProgress = performUnitOfWork(workInProgress);
  4. }
  5. }

看来我们重点是需要看看 performUnitOfWork:

  1. function performUnitOfWork(unitOfWork: Fiber): Fiber | null {
  2. const current = unitOfWork.alternate;
  3. // ... 忽略计时逻辑
  4. let next;
  5. if (enableProfilerTimer && (unitOfWork.mode & ProfileMode) !== NoMode) {
  6. next = beginWork(current, unitOfWork, renderExpirationTime);
  7. } else {
  8. next = beginWork(current, unitOfWork, renderExpirationTime);
  9. }
  10. // ... 忽略特殊逻辑
  11. ReactCurrentOwner.current = null;
  12. return next;
  13. }

我们忽略计时逻辑,发现这段代码的内容其实就是两个 beginWork(这里解答了我们在第3篇中 4.3.1 中留下的问题)。这个 beginWork 引自 ReactFiberBeginWork.js。

3.4.4 ReactFiberBeginWork.js

本节代码分析同 第 3 篇 4.3.1: renderWithHooks,不再赘述。

也就是现在我们 renderedWork 上的 updateQueue(还记得它吗?它的内容是 Effect 链表) 和 effectTag 挂到了 Fiber 上,跳过这部分逻辑,我们看看 Fiber 最后怎么处理它们。

3.4.5 ReactFiberWorkLoop.js - commit

在刚才分析 renderRoot 的过程中,我们关注到任务超时之后会直接进入 commit 阶段。我们先看看 commitRoot 的代码:

  1. function commitRoot(root) {
  2. const renderPriorityLevel = getCurrentPriorityLevel();
  3. runWithPriority(
  4. ImmediatePriority,
  5. commitRootImpl.bind(null, root, renderPriorityLevel),
  6. );
  7. return null;
  8. }

好的,这里发现我们应该关注 commitRootImpl,来看看:

  1. function commitRootImpl(root, renderPriorityLevel) {
  2. // ...
  3. startCommitTimer();
  4. // Get the list of effects.
  5. let firstEffect;
  6. if (finishedWork.effectTag > PerformedWork) {
  7. if (finishedWork.lastEffect !== null) {
  8. finishedWork.lastEffect.nextEffect = finishedWork;
  9. firstEffect = finishedWork.firstEffect;
  10. } else {
  11. firstEffect = finishedWork;
  12. }
  13. } else {
  14. firstEffect = finishedWork.firstEffect;
  15. }
  16. if (firstEffect !== null) {
  17. do {
  18. try {
  19. commitBeforeMutationEffects();
  20. } catch (error) {
  21. invariant(nextEffect !== null, 'Should be working on an effect.');
  22. captureCommitPhaseError(nextEffect, error);
  23. nextEffect = nextEffect.nextEffect;
  24. }
  25. } while (nextEffect !== null);
  26. stopCommitSnapshotEffectsTimer();
  27. if (enableProfilerTimer) {
  28. // Mark the current commit time to be shared by all Profilers in this
  29. // batch. This enables them to be grouped later.
  30. recordCommitTime();
  31. }
  32. // The next phase is the mutation phase, where we mutate the host tree.
  33. startCommitHostEffectsTimer();
  34. nextEffect = firstEffect;
  35. do {
  36. try {
  37. commitMutationEffects(root, renderPriorityLevel);
  38. } catch (error) {
  39. invariant(nextEffect !== null, 'Should be working on an effect.');
  40. captureCommitPhaseError(nextEffect, error);
  41. nextEffect = nextEffect.nextEffect;
  42. }
  43. } while (nextEffect !== null);
  44. stopCommitHostEffectsTimer();
  45. resetAfterCommit(root.containerInfo);
  46. // The work-in-progress tree is now the current tree. This must come after
  47. // the mutation phase, so that the previous tree is still current during
  48. // componentWillUnmount, but before the layout phase, so that the finished
  49. // work is current during componentDidMount/Update.
  50. root.current = finishedWork;
  51. // The next phase is the layout phase, where we call effects that read
  52. // the host tree after it's been mutated. The idiomatic use case for this is
  53. // layout, but class component lifecycles also fire here for legacy reasons.
  54. startCommitLifeCyclesTimer();
  55. nextEffect = firstEffect;
  56. do {
  57. try {
  58. commitLayoutEffects(root, expirationTime);
  59. } catch (error) {
  60. invariant(nextEffect !== null, 'Should be working on an effect.');
  61. captureCommitPhaseError(nextEffect, error);
  62. nextEffect = nextEffect.nextEffect;
  63. }
  64. } while (nextEffect !== null);
  65. stopCommitLifeCyclesTimer();
  66. nextEffect = null;
  67. // Tell Scheduler to yield at the end of the frame, so the browser has an
  68. // opportunity to paint.
  69. requestPaint();
  70. if (enableSchedulerTracing) {
  71. __interactionsRef.current = ((prevInteractions: any): Set<Interaction>);
  72. }
  73. executionContext = prevExecutionContext;
  74. } else {
  75. // No effects.
  76. // ...
  77. }
  78. stopCommitTimer();
  79. nextEffect = firstEffect;
  80. while (nextEffect !== null) {
  81. const nextNextEffect = nextEffect.nextEffect;
  82. nextEffect.nextEffect = null;
  83. nextEffect = nextNextEffect;
  84. }
  85. // ...
  86. return null;
  87. }

commitRootImpl 的代码是真的很长,我这里忽略了一些和 effect 处理无关的代码,剩下我们阅读一下,发现当 effect 存在的时候,有三段逻辑要处理,它们的逻辑基本相同,循环 effect 链表传给三个不同的函数,分别是:

  • commitBeforeMutationEffects
  • commitMutationEffects
  • commitLayoutEffects

最后将循环 effect,将 nextEffect 赋值成 nextNextEffect。

限于篇幅问题,且第三个函数关于 useLayoutEffect,所以左右这里这三个函数我们这里都不一一展开解释了,留给下篇文章中分析 useLayoutEffect 再来详解。所以 3.4 中我们留下的问题—— effect.tag 的那些二进制数是什么意思?这个问题也需要等到下一篇文章中来解释了。

我们这里只需要知道这三个函数的核心代码分别引用了 ReactFiberCommitWork.js 中的 commitWorkcommitBeforeMutationLifeCyclescommitLifeCycles,而这三个函数的核心代码在处理 FunctionCompoment 的逻辑时都走到了 commitHookEffectList 中即可。

3.5 commitHookEffectList

分析了一大圈,最后我们看看 ReactFiberCommitWork.js 中 commitHookEffectList 的逻辑,这里便是 useEffect 终点了:

  1. function commitHookEffectList(
  2. unmountTag: number,
  3. mountTag: number,
  4. finishedWork: Fiber,
  5. ) {
  6. const updateQueue: FunctionComponentUpdateQueue | null = (finishedWork.updateQueue: any);
  7. let lastEffect = updateQueue !== null ? updateQueue.lastEffect : null;
  8. if (lastEffect !== null) {
  9. const firstEffect = lastEffect.next;
  10. let effect = firstEffect;
  11. do {
  12. if ((effect.tag & unmountTag) !== NoHookEffect) {
  13. // Unmount
  14. const destroy = effect.destroy;
  15. effect.destroy = undefined;
  16. if (destroy !== undefined) {
  17. destroy();
  18. }
  19. }
  20. if ((effect.tag & mountTag) !== NoHookEffect) {
  21. // Mount
  22. const create = effect.create;
  23. effect.destroy = create();
  24. }
  25. effect = effect.next;
  26. } while (effect !== firstEffect);
  27. }
  28. }

可以发现,这里的代码很清楚,这里把 renderedWork.updateQueue 上的 effect 取了下来,在 unmount 的时候执行 effect.destory(也就是 useEffect 第一个参数的返回值),在 mount 的时候执行 effect.create(也就是 useEffect 传入的第一个参数)。并且,循环所有的 effect 直到结束。

同时这里也印证了我们之前的猜想:当 tag 是 NoHookEffect 的时候什么也不做。

这里我们把 useEffect 的源码解释清楚了,但是遗留了一个问题:effect.tag 这个参数究竟有什么用?目前我们仅仅知道当它是 NoHookEffect 时的作用是不执行 useEffect 的内容,但是其他的值我们还没有分析到,它们分析逻辑主要在我们 3.4.5 略过的那三个函数里。在下篇文章中,我们分析 useLayoutEffect 中会拿出来详细分析。

大家再见。

最后附上 3.4 节分析的流程图:

React Hooks 源码解析(4):useEffect - 图3