Hooks - implementation初探

上一篇文章简单介绍了一下两种比较常用的Hooks - useStateuseEffect
最近稍微拜读了一下Fiber的代码,尝试解释一下这两个Hook的实现。

Fiber reconciler

首先我们知道的是React维护了一个自己的Virtual DOM tree,当我们要创建一个新的元素Element时,是在VDOM tree上挂载mount, 更新update, 渲染render,最终在真实的DOM tree上绘制paint的。

在React里,实际上对一个元素,自顶向下的来说有三层认知层次:
a. DOM 
真实的DOM结点,是我们最终看到的HTML上所写的结点。
b. Instance
也就是React所维护的实例,即Virtual DOM结点
c. Element 
也就是我们所编写的代码,用来描述一个元素的样式和要展示的数据。我们调用的render方法实际上是告诉React要更新对应的Instance了。

同时JS是在浏览器的主线程上运行的,和样式计算、布局等绘制工作一起运行。

这样就导致一个问题,如果一个Element的更新时间太长,导致JS占用了很多浏览器资源,使得这次render以及paint的时间超过了1/24秒,那么就会出现肉眼可见的页面卡顿。

于是React 16发布了React Fiber reconciler来解决这个卡顿问题。
主要解决方法是把一次render任务拆分成一些更小的任务,每做完一段就把时间控制权交回给浏览器,不让JS一次占用太多时间。
为此我们又种了两棵树fiber tree(取代了原来的VDOM tree) 和 workInProgress tree,其中fiber tree是由fiber nodes组成的,记录了增量更新时需要的上下文信息,而workInProgress tree则是一个进度快照,用来进行断点恢复。
一个fiber node长得像这样:

{
  return, //当前结点处理完后,向谁提交结果,即父结点
  child,
  sibling,
  ...
}

workInProgress tree则是由fiber tree构造出来的,其维护了一个effect list,当fiber结点需要更新时,则给当前这个结点打一个tag,同时当前结点的effect(需要实施的更新)会返回给自己的return,这样当workInProgress tree构造出来时,其根节点的effect list就是要做的所有side effect。此过程中的任何一步都可以中断。
之后执行commit操作,即渲染DOM Node,此过程是不可中断的。
值得注意的是,fiber nodeworkInProgress node使用的是同样的数据结构。
实际上当commit操作完成以后,reactworkInProgress treefiber node的指针互换,因为此时workInProgress tree的状态和真实的DOM tree相同。

Hooks

现在我们对上文所述的fiber node扩充一下

{
   return, //当前结点处理完后,向谁提交结果,即父结点
  child,
  sibling,
  memoizedState,
}

React在每次结点render之前会计算出当前的state并赋值给fiber实例的memoizedState,再调用render方法。所以React可以根据这条属性拿到当前的state
对于一个class形式的component来说,我们可以很轻松的将statememoizedState对应起来。
而在一个function形式的component里,我们一般是这样使用state

const Example = () => {
  const [name, setName] = useState("");
  const [email, setEmail] = useState("");
}

我们知道fiber node上只有一个memoizedState,但在Hooks中,React并不知道我们调用了几次useState,我们要怎么把每个Hookstate合并到fiber node上的memoizedState上呢?
React定义了一个Hook对象:

{
  memoizedState: any,

  baseState: any,
  baseUpdate: Update<any, any> | null,
  queue: UpdateQueue<any, any> | null,

  next: Hook | null,
}

这样当我们每调用一次useStateReact就创建了一个新的Hook对象,然后连接到当前Hooks链表的尾部。
也因此,我们在看Hooks文档的时候会发现有一条规则,不要在条件判断语句里使用Hooks
每次useXXX在执行的时候,第一个运行的函数是下面这个:

function mountWorkInProgressHook(): Hook {
  const hook: Hook = {
    memoizedState: null,

    baseState: null,
    queue: null,
    baseUpdate: null,

    next: null,
  };

  if (workInProgressHook === null) {
    // This is the first hook in the list
    firstWorkInProgressHook = workInProgressHook = hook;
  } else {
    // Append to the end of the list
    workInProgressHook = workInProgressHook.next = hook;
  }
  return workInProgressHook;
}

如上例中,如果Name这个Hook因为某些原因被跳过的话,那么我们的Email会成为这次函数执行里第一次被调用的Hook,也就是会拿到当前Hooks list里的第一个值。

那么setState是怎样实现的呢?上源码

function mountState<S>(
  initialState: (() => S) | S,
): [S, Dispatch<BasicStateAction<S>>] {
  const hook = mountWorkInProgressHook();
  if (typeof initialState === 'function') {
    initialState = initialState();
  }
  hook.memoizedState = hook.baseState = initialState;
  const queue = (hook.queue = {
    last: null,
    dispatch: null,
    lastRenderedReducer: basicStateReducer,
    lastRenderedState: (initialState: any),
  });
  const dispatch: Dispatch<
    BasicStateAction<S>,
  > = (queue.dispatch = (dispatchAction.bind(
    null,
    // Flow doesn't know this is non-null, but we do.
    ((currentlyRenderingFiber: any): Fiber),
    queue,
  ): any));
  return [hook.memoizedState, dispatch];
}

mountState是我们实际调用的useState实现,根据代码我们可以看出来,我们拿到的setState方法实际上是一个Dispatch,而当我们调用得到的setState时,会创建一个Update对象:

type Update<S, A> = {
  expirationTime: ExpirationTime,
  suspenseConfig: null | SuspenseConfig,
  action: A,
  eagerReducer: ((S, A) => S) | null,
  eagerState: S | null,
  next: Update<S, A> | null,
};

action是我们传给dispatchaction也就是setState传入的值。
当我们收集到所有的update之后,就会调用React的更新,当其执行到我们的这个Functional Component时,就会执行对应的useState, 其Hook对象上的queue保存了我们要执行的update,执行完所有update后拿到最终的state保存到memoizedState上,起到setState的效果。你可能要问为什么queue是个UpdateQueue,因为我们可能会调用多次setState
当所有的Hook执行完以后,拿到全部memoizedState,更新到fiber node上。

同样的,React也为Effect提供了一个对象:

type Effect = {
  tag: HookEffectTag,
  create: () => (() => void) | void,
  destroy: (() => void) | void,
  deps: Array<mixed> | null,
  next: Effect,
};

useEffct的调用分了三个阶段:

function mountEffect(
  create: () => (() => void) | void,
  deps: Array<mixed> | void | null,
): void {
  if (__DEV__) {
    // $FlowExpectedError - jest isn't a global, and isn't recognized outside of tests
    if ('undefined' !== typeof jest) {
      warnIfNotCurrentlyActingEffectsInDEV(
        ((currentlyRenderingFiber: any): Fiber),
      );
    }
  }
  return mountEffectImpl(
    UpdateEffect | PassiveEffect,
    UnmountPassive | MountPassive,
    create,
    deps,
  );
}

在这个阶段给useEffect打上了一个effectTag用来标记这个Effect的类型

function mountEffectImpl(fiberEffectTag, hookEffectTag, create, deps): void {
  const hook = mountWorkInProgressHook();
  const nextDeps = deps === undefined ? null : deps;
  sideEffectTag |= fiberEffectTag;
  hook.memoizedState = pushEffect(hookEffectTag, create, undefined, nextDeps);
}

这个阶段把当前Effect的tag更新到了整个component

function pushEffect(tag, create, destroy, deps) {
  const effect: Effect = {
    tag,
    create,
    destroy,
    deps,
    // Circular
    next: (null: any),
  };
  if (componentUpdateQueue === null) {
    componentUpdateQueue = createFunctionComponentUpdateQueue();
    componentUpdateQueue.lastEffect = effect.next = effect;
  } else {
    const lastEffect = componentUpdateQueue.lastEffect;
    if (lastEffect === null) {
      componentUpdateQueue.lastEffect = effect.next = effect;
    } else {
      const firstEffect = lastEffect.next;
      lastEffect.next = effect;
      effect.next = firstEffect;
      componentUpdateQueue.lastEffect = effect;
    }
  }
  return effect;
}

最后一个阶段,把当前effect添加到componentUpdateQueue的尾部。
值得注意的一点是,componentUpdateQueue最终形成了一个环,因此需要一个lastEffect标记实际上的最后一个Effect是谁。

在拿到所有的effect后,ReactcomponentUpdateQueue更新给了currentlyRenderingFiberupdateQueue,最终由workInProgress tree去收集并执行所有fiber node上的effect

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

推荐阅读更多精彩内容