React Hooks 原理

React Hooks 原理

目前,Hooks 应该是 React 中最火的概念了,在阅读这篇文章之前,希望你已经了解了基本的 Hooks 用法。

在使用 Hooks 的时候,我们可能会有很多疑惑

为什么只能在函数最外层调用 Hook,不要在循环、条件判断或者子函数中调用?

为什么 useEffect 第二个参数是空数组,就相当于 ComponentDidMount ,只会执行一次?

自定义的 Hook 是如何影响使用它的函数组件的?

Capture Value 特性是如何产生的?

......

这篇文章我们不会讲解 Hooks 的概念和用法,而是会带你从零实现一个 tiny hooks,知其然知其所以然。

useState

最简单的 useState 用法是这样的:

demo1: https://codesandbox.io/s/v0nqm309q3

functionCounter(){var[count,setCount]=useState(0);return(<div><div>{count}</div><ButtononClick={()=>{setCount(count+1);}}>点击</Button></div>);}

基于 useState 的用法,我们尝试着自己实现一个 useState:

demo2:https://codesandbox.io/s/myy5qvoxpp

functionuseState(initialValue){varstate=initialValue;functionsetState(newState){state=newState;render();}return[state,setState];}

这时我们发现,点击 Button 的时候,count 并不会变化,为什么呢?我们没有存储 state,每次渲染 Counter 组件的时候,state 都是新重置的。

自然我们就能想到,把 state 提取出来,存在 useState 外面。

demo3:https://codesandbox.io/s/q9wq6w5k3w

var_state;// 把 state 存储在外面functionuseState(initialValue){_state=_state||initialValue;// 如果没有 _state,说明是第一次执行,把 initialValue 复制给它functionsetState(newState){_state=newState;render();}return[_state,setState];}

到目前为止,我们实现了一个可以工作的 useState,至少现在来看没啥问题。

接下来,让我们看看 useEffect 是怎么实现的。

useEffect

useEffect 是另外一个基础的 Hook,用来处理副作用,最简单的用法是这样的:

demo4:https://codesandbox.io/s/93jp55qyp4

useEffect(()=>{console.log(count);},[count]);

我们知道 useEffect 有几个特点:

有两个参数 callback 和 dependencies 数组

如果 dependencies 不存在,那么 callback 每次 render 都会执行

如果 dependencies 存在,只有当它发生了变化, callback 才会执行

我们来实现一个 useEffect

demo5:https://codesandbox.io/s/3kv3zlvzl1

let_deps;// _deps 记录 useEffect 上一次的 依赖functionuseEffect(callback,depArray){consthasNoDeps=!depArray;// 如果 dependencies 不存在consthasChangedDeps=_deps? !depArray.every((el,i)=>el===_deps[i])// 两次的 dependencies 是否完全相等:true;/* 如果 dependencies 不存在,或者 dependencies 有变化*/if(hasNoDeps||hasChangedDeps){callback();_deps=depArray;}}

到这里,我们又实现了一个可以工作的 useEffect,似乎没有那么难。

此时我们应该可以解答一个问题:

Q:为什么第二个参数是空数组,相当于 componentDidMount ?

A:因为依赖一直不变化,callback 不会二次执行。

Not Magic, just Arrays

到现在为止,我们已经实现了可以工作的 useState 和 useEffect。但是有一个很大的问题:它俩都只能使用一次,因为只有一个 _state 和 一个 _deps。比如

const[count,setCount]=useState(0);const[username,setUsername]=useState('fan');

count 和 username 永远是相等的,因为他们共用了一个 _state,并没有地方能分别存储两个值。我们需要可以存储多个 _state 和 _deps。

如 《React hooks: not magic, just arrays》所写,我们可以使用数组,来解决 Hooks 的复用问题。

demo6:https://codesandbox.io/s/50ww35vkzl

代码关键在于:

初次渲染的时候,按照 useState,useEffect 的顺序,把 state,deps 等按顺序塞到 memoizedState 数组中。

更新的时候,按照顺序,从 memoizedState 中把上次记录的值拿出来。

如果还是不清楚,可以看下面的图。

letmemoizedState=[];// hooks 存放在这个数组letcursor=0;// 当前 memoizedState 下标functionuseState(initialValue){memoizedState[cursor]=memoizedState[cursor]||initialValue;constcurrentCursor=cursor;functionsetState(newState){memoizedState[currentCursor]=newState;render();}return[memoizedState[cursor++],setState];// 返回当前 state,并把 cursor 加 1}functionuseEffect(callback,depArray){consthasNoDeps=!depArray;constdeps=memoizedState[cursor];consthasChangedDeps=deps? !depArray.every((el,i)=>el===deps[i]):true;if(hasNoDeps||hasChangedDeps){callback();memoizedState[cursor]=depArray;}cursor++;}

我们用图来描述 memoizedState 及 cursor 变化的过程。

1. 初始化

2. 初次渲染

3. 事件触发

4. Re Render

到这里,我们实现了一个可以任意复用的 useState 和 useEffect。

同时,也可以解答几个问题:

Q:为什么只能在函数最外层调用 Hook?为什么不要在循环、条件判断或者子函数中调用。

A:memoizedState 数组是按 hook定义的顺序来放置数据的,如果 hook 顺序变化,memoizedState 并不会感知到。

Q:自定义的 Hook 是如何影响使用它的函数组件的?

A:共享同一个 memoizedState,共享同一个顺序。

Q:“Capture Value” 特性是如何产生的?

A:每一次 ReRender 的时候,都是重新去执行函数组件了,对于之前已经执行过的函数组件,并不会做任何操作。

真正的 React 实现

虽然我们用数组基本实现了一个可用的 Hooks,了解了 Hooks 的原理,但在 React 中,实现方式却有一些差异的。

React 中是通过类似单链表的形式来代替数组的。通过 next 按顺序串联所有的 hook。

typeHooks={memoizedState:any,// 指向当前渲染节点 FiberbaseState:any,// 初始化 initialState, 已经每次 dispatch 之后 newStatebaseUpdate:Update<any>| null,// 当前需要更新的 Update ,每次更新完之后,会赋值上一个 update,方便 react 在渲染错误的边缘,数据回溯queue:UpdateQueue<any>| null,// UpdateQueue 通过next:Hook| null,// link 到下一个 hooks,通过 next 串联每一 hooks}typeEffect={tag:HookEffectTag,// effectTag 标记当前 hook 作用在 life-cycles 的哪一个阶段create:()=>mixed,// 初始化 callbackdestroy:(()=>mixed)| null,// 卸载 callbackdeps:Array<mixed>| null,next:Effect,// 同上 };

memoizedState,cursor 是存在哪里的?如何和每个函数组件一一对应的?

我们知道,react 会生成一棵组件树(或Fiber 单链表),树中每个节点对应了一个组件,hooks 的数据就作为组件的一个信息,存储在这些节点上,伴随组件一起出生,一起死亡。

参考文章

Deep dive: How do React hooks really work?

React hooks: not magic, just arrays

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

推荐阅读更多精彩内容