React Hooks

1、前言

React Hooks 是 React 16.8 引入的新特性,允许在不使用 Class 的前提下使用 state 和其他特性。React Hooks 要解决的问题是状态共享,是继 render-props 和 higher-order components 之后的第三种状态逻辑复用方案,不会产生 JSX 嵌套地狱问题。

2、状态逻辑复用

一般来说,组件是 UI 和逻辑,但是逻辑这一层面却很难复用。对用户而言,组件就像一个黑盒,应该拿来即用。但当组件的样式或者结构不满足需求的时候,只能去重新实现这个组件。
React Hooks 原理与最佳实践 - 图1
在开发 React 应用的时候,经常会遇到类似下面这种场景,可能会有两个疑问:

  1. Loading 是否可以复用?
  2. Loading 该怎么复用?

React Hooks 原理与最佳实践 - 图2
这几个例子都指向了同一个问题,那就是如何实现组件的逻辑复用?

2.1 render props

将函数作为 props 传给父组件,父组件中的状态共享,通过参数传给函数,实现渲染,这就是 render props。使用 render prop 的库有 React Router、Downshift 以及 Formik。以下面这个 Toggle 组件为例子,一般可以这样用:
React Hooks 原理与最佳实践 - 图3
可以看到,控制 Modal 组件是否展示的状态被提取到了 Toggle 组件中,这个 Toggle 组件还可以拿来多次复用到其他组件里面。那么这个 Toggle 是怎么实现的呢?看到实现后就会理解 render props 的原理
React Hooks 原理与最佳实践 - 图4
关于 render props 的更多内容可以参考 React 中文网的相关章节:Render Props。

2.2 higher-order components

higher-order components 一般简称 hoc,中文翻译为高阶组件。从名字上就可以看出来,高阶组件肯定和高阶函数有什么千丝万缕的关系。高阶组件的本质是一个高阶函数,它接收一个组件,返回一个新的组件。在这个新的组件中的状态共享,通过 props 传给原来的组件。以刚刚那个 Toggle 组件为例子,高阶组件同样可以被多次复用,常常可以配合装饰器一起使用。
React Hooks 原理与最佳实践 - 图5
高阶组件的实现和 render props 也不太一样,主要是一个高阶函数。
React Hooks 原理与最佳实践 - 图6

2.3 render props 和高阶组件的弊端

不管是 render props 还是高阶组件,他们要做的都是实现状态逻辑的复用,可这俩是完美的解决方案吗?考虑一下,如果依赖了多个需要复用的状态逻辑的时候,该怎么写呢?以 render props 为例:
React Hooks 原理与最佳实践 - 图7
看看这个代码,有没有一种似曾相识的感觉?这一天,终于想起被“回调地狱”支配的恐惧。不得不再次祭出这张图了。
React Hooks 原理与最佳实践 - 图8
同样地,高阶组件也会有这个问题,但由于装饰器的简洁性,没有 render props 看起来那么可怕。除此之外,他们俩还有另一个问题,那就是组件嵌套过深之后,会给调试带来很大的麻烦。这个是 render props 中组件嵌套在 React 开发者工具中的表现。
React Hooks 原理与最佳实践 - 图9
对于高阶组件来说,如果没有对组件手动设置 name/displayName,就会遇到更严重的问题,那就是一个个匿名组件嵌套。毕竟上面 render props 的嵌套至少能知道组件名。
React Hooks 原理与最佳实践 - 图10
社区里面也已经有很多解决 render props 嵌套的方案,其中 Epitath 提供了一种以 generator 的方法来解决嵌套问题,利用 generator 实现了伪同步代码。
React Hooks 原理与最佳实践 - 图11

2.4 React Hooks

React Hooks 则可以完美解决上面的嵌套问题,它拥有下面这几个特性。

  1. 多个状态不会产生嵌套,写法还是平铺的
  2. 允许函数组件使用 state 和部分生命周期
  3. 更容易将组件的 UI 与状态分离

React Hooks 原理与最佳实践 - 图12
上面是一个结合了 useState 和 useEffect 两个 hook 方法的例子,主要是在 resize 事件触发时获取到当前的 window.innerWidth。这个 useWindowWidth 方法可以拿来在多个地方使用。常用的 Hook 方法如下:
React Hooks 原理与最佳实践 - 图13

3、useState & useRef

useState 是 React Hooks 中很基本的一个 API,它的用法主要有这几种:

  1. useState 接收一个初始值,返回一个数组,数组里面分别是当前值和修改这个值的方法(类似 state 和 setState)。
  2. useState 接收一个函数,返回一个数组。
  3. setCount 可以接收新值,也可以接收一个返回新值的函数。

    1. const [ count1, setCount1 ] = useState(0);
    2. const [ count2, setCount2 ] = useState(() => 0);
    3. setCount1(1); // 修改 state

    3.1 和 class state 的区别

    虽然函数组件也有了 state,但是 function state 和 class state 还是有一些差异:

  4. function state 的粒度更细,class state 过于无脑。

  5. function state 保存的是快照,class state 保存的是最新值。
  6. 引用类型的情况下,class state 不需要传入新的引用,而 function state 必须保证是个新的引用。

    3.2 快照(闭包) vs 最新值(引用)

    在开始前,先抛出这么一个问题。在 1s 内频繁点击10次按钮,下面代码的执行表现是什么?
    React Hooks 原理与最佳实践 - 图14
    如果是这段代码呢?它又会是什么表现?
    React Hooks 原理与最佳实践 - 图15
    如果能成功答对,已经掌握了 useState 的用法。在第一个例子中,连续点击十次,页面上的数字会从0增长到10。而第二个例子中,连续点击十次,页面上的数字只会从0增长到1。
    这个是为什么呢?其实这主要是引用和闭包的区别。
    class 组件里面可以通过 this.state 引用到 count,所以每次 setTimeout 的时候都能通过引用拿到上一次的最新 count,所以点击多少次最后就加了多少。
    在 function component 里面每次更新都是重新执行当前函数,也就是说 setTimeout 里面读取到的 count 是通过闭包获取的,而这个 count 实际上只是初始值,并不是上次执行完成后的最新值,所以最后只加了1次。

    3.3 快照和引用的转换

    如果想让函数组件也是从0加到10,那么该怎么来解决呢?如果模仿类组件里面的 this.state,用一个引用来保存 count 不就好了吗?没错,这样是可以解决,只是这个引用该怎么写呢?在 state 里面设置一个对象好不好?就像下面这样:

    1. const [state, setState] = useState({ count: 0 })

    答案是不行,因为即使 state 是个对象,但每次更新的时候,要传一个新的引用进去,这样的引用依然是没有意义。

    1. setState({
    2. count: state.count + 1
    3. })

    3.3 useRef

    想要解决这个问题,那就涉及到另一个新的 Hook 方法 —— useRef。useRef 是一个对象,它拥有一个 current 属性,并且不管函数组件执行多少次,而 useRef 返回的对象永远都是原来那一个。
    React Hooks 原理与最佳实践 - 图16
    useRef 有下面这几个特点:

  7. useRef 是一个只能用于函数组件的方法。

  8. useRef 是除字符串 ref、函数 refcreateRef 之外的第四种获取 ref 的方法。
  9. useRef 在渲染周期内永远不会变,因此可以用来引用某些数据。
  10. 修改 ref.current 不会引发组件重新渲染。

useRef vs createRef:

  1. 两者都是获取 ref 的方式,都有一个 current 属性。
  2. useRef 只能用于函数组件,createRef 可以用在类组件中。
  3. useRef 在每次重新渲染后都保持不变,而 createRef 每次都会发生变化。

    3.4 写需求遇到的坑

    React Hooks 原理与最佳实践 - 图17
    之前在写需求的时候遇到过这样的一个坑。bankIdref 都是从接口获取到的,这里很自然就想到在 useCallback 里面指定依赖。
    React Hooks 原理与最佳实践 - 图18
    但是呢,这个 handlerReappear 方法需要在第一次进入页面的时候,向 JS Bridge 注册的事件,这就导致了一个问题,不管后来 handlerReappear 如何变化,registerHandler 里面依赖的 callback 都是第一次的,这也是闭包导致的问题。当然,在 useEffect 里面也指定了依赖不好吗?但要注意这是个注册事件,意味着每次都要清除上一次的事件,需要调用到 JS Bridge,在性能上肯定不是个好办法。
    最终,选择使用 useRef 来保存 bankIdref,这样就可以通过引用来获取到最新的值。
    React Hooks 原理与最佳实践 - 图19

    3.5 Vue3 Composition API

    React Hooks 原理与最佳实践 - 图20
    在 vue3 里面提供了新的 Composition API,之前知乎有个问题是 React Hooks 是否可以改为用类似 Vue 3 Composition API 的方式实现?

    4、useEffect

    useEffect 是一个 Effect Hook,常用于一些副作用的操作,在一定程度上可以充当 componentDidMountcomponentDidUpdatecomponentWillUnmount 这三个生命周期。useEffect 是非常重要的一个方法,可以说是 React Hooks 的灵魂,它用法主要有这么几种:

  4. useEffect 接收两个参数,分别是要执行的回调函数、依赖数组。

  5. 如果依赖数组为空数组,那么回调函数会在第一次渲染结束后(componentDidMount)执行,返回的函数会在组件卸载时(componentWillUnmount)执行。
  6. 如果不传依赖数组,那么回调函数会在每一次渲染结束后(componentDidMountcomponentDidUpdate)执行。
  7. 如果依赖数组不为空数组,那么回调函数会在依赖值每次更新渲染结束后(componentDidUpdate)执行,这个依赖值一般是 state 或者 props。

React Hooks 原理与最佳实践 - 图21
useEffect 比较重要,它主要有这几个作用:

  1. 代替部分生命周期,如 componentDidMount、componentDidUpdate、componentWillUnmount。
  2. 更加 reactive,类似 mobx 的 reaction 和 vue 的 watch。
  3. 从命令式变成声明式,不需要再关注应该在哪一步做某些操作,只需要关注依赖数据。
  4. 通过 useEffect 和 useState 可以编写一系列自定义的 Hook。

    4.1 useEffect vs useLayoutEffect

    useLayoutEffect 也是一个 Hook 方法,从名字上看和 useEffect 差不多,他俩用法也比较像。在90%的场景下都会用 useEffect,然而在某些场景下却不得不用 useLayoutEffect。useEffect 和 useLayoutEffect 的区别是:

  5. useEffect 不会 block 浏览器渲染,而 useLayoutEffect 会。

  6. useEffect 会在浏览器渲染结束后执行,useLayoutEffect 则是在 DOM 更新完成后,浏览器绘制之前执行。

这两句话该怎么来理解呢?以一个移动的方块为例子:
React Hooks 原理与最佳实践 - 图22
React Hooks 原理与最佳实践 - 图23
在 useEffect 里面会让这个方块往后移动 600px 距离,可以看到这个方块在移动过程中会闪一下。但如果换成了 useLayoutEffect 呢?会发现方块不会再闪动,而是直接出现在了 600px 的位置。
React Hooks 原理与最佳实践 - 图24
React Hooks 原理与最佳实践 - 图25
原因是 useEffect 是在浏览器绘制之后执行的,所以方块一开始就在最左边,于是看到了方块移动的动画。然而 useLayoutEffect 是在绘制之前执行的,会阻塞页面的绘制,所以页面会在 useLayoutEffect 里面的代码执行结束后才去继续绘制,于是方块就直接出现在了右边。那么这里的代码是怎么实现的呢?以 preact 为例,useEffect 在 options.commit 阶段执行,而 useLayoutEffect 在 options.diffed 阶段执行。然而在实现 useEffect 的时候使用了 requestAnimationFramerequestAnimationFrame 可以控制 useEffect 里面的函数在浏览器重绘结束,下次绘制之前执行。
React Hooks 原理与最佳实践 - 图26

5、useMemo

useMemo 的用法类似 useEffect,常常用于缓存一些复杂计算的结果。useMemo 接收一个函数和依赖数组,当数组中依赖项变化的时候,这个函数就会执行,返回新的值。

  1. const sum = useMemo(() => {
  2. // 一系列计算
  3. }, [count])

举个例子会更加清楚 useMemo 的使用场景,就以下面这个 DatePicker 组件的计算为例:
React Hooks 原理与最佳实践 - 图27
DatePicker 组件每次打开或者切换月份的时候,都需要大量的计算来算出当前需要展示哪些日期。然后再将计算后的结果渲染到单元格里面,这里可以使用 useMemo 来缓存,只有当传入的日期变化时才去计算。

6、useCallback

和 useMemo 类似,只不过 useCallback 是用来缓存函数。

6.1 匿名函数导致不必要的渲染

在编写 React 组件的时候,经常会用到事件处理函数,很多人都会简单粗暴的传一个箭头函数。

  1. class App extends Component { render() { return <h1 onClick={() => {}}></h1> }}class App extends Component {
  2. render() {
  3. return <h1 onClick={() => {}}></h1>
  4. }
  5. }

这种箭头函数有个问题,那就是在每一次组件重新渲染的时候都会生成一个重复的匿名箭头函数,导致传给组件的参数发生了变化,对性能造成一定的损耗。
在函数组件里面,同样会有这个传递新的匿名函数的问题。从下面这个例子来看,每次点击 div,就会引起 Counter 组件重新渲染。这次更新明显和 Input 组件无关,但每次重新渲染之后,都会创建新的 onChange 方法。这样相当于传给 Input 的 onChange 参数变化,即使 Input 内部做过 shadowEqual 也没有意义了,都会跟着重新渲染。原本只想更新 count 值的,可 Input 组件 却做了不必要的渲染。
React Hooks 原理与最佳实践 - 图28
这就是体现 useCallback 价值的地方了,可以用 useCallback 指定依赖项。在无关更新之后,通过 useCallback 取的还是上一次缓存起来的函数。因此,useCallback 常常配合 React.memo 来一起使用,用于进行性能优化。
React Hooks 原理与最佳实践 - 图29

7、useReducer && useContext

7.1 useReducer

useReducer 和 useState 的用法很相似,甚至在 preact 中,两者实现都是一样的。useReducer 接收一个 reducer 函数和初始 state,返回了 state 和 dispatch 函数,常常用于管理一些复杂的状态,适合 action 比较多的场景。
React Hooks 原理与最佳实践 - 图30

7.2 useContext

新版 Context 常常有一个提供数据的生产者(Provider),和一个消费数据的消费者(Consumer),需要通过 Consumer 来以 render props 的形式获取到数据。如果从祖先组件传来了多个 Provider,那最终就又陷入了 render props 嵌套地狱。
React Hooks 原理与最佳实践 - 图31
useContext 允许以扁平化的形式获取到 Context 数据。即使有多个祖先组件使用多个 Context.Provider 传值,也可以扁平化获取到每一个 Context 数据。
React Hooks 原理与最佳实践 - 图32

7.3 实现一个简单的 Redux

通过 useReducer 和 useContext,完全可以实现一个小型的 Redux。
reducer.js
React Hooks 原理与最佳实践 - 图33
Context.js

  1. export const Context = createContext(null);

App.js
React Hooks 原理与最佳实践 - 图34

8、Custom Hooks

对于 react 来说,在函数组件中使用 state 固然有一些价值,但最有价值的还是可以编写通用 custom hooks 的能力。想像一下,一个单纯不依赖 UI 的业务逻辑 hook,开箱即用。不仅可以在不同的项目中复用,甚至还可以跨平台使用,react、react native、react vr 等等。编写自定义 hook 也需要以 use 开头,这样保证可以配合 eslint 插件使用。在 custom hooks 中也可以调用其他 hook,当前的 hook 也可以被其他 hook 或者组件调用。以官网上这个获取好友状态的自定义 Hook 为例:
React Hooks 原理与最佳实践 - 图35
这个自定义 Hook 里面对好友的状态进行了监听,每次状态更新的时候都会去更新 isOnline,当组件卸载的时候会清除掉这个监听。这就是 React Hooks 最有用的地方,它允许编写自定义 Hook,然后这个自定义 Hook 可以复用给多个组件,并且不会和 UI 耦合到一起。

9、React Hooks 原理

由于 preact hooks 的代码和原有的逻辑耦合度很小,这里为了更加浅显易懂,选用了 preact hooks 的源码来解读。

9.1 Hooks 执行流程

在 React 中,组件返回的 JSX 元素也会被转换为虚拟 DOM,就是下方的 vnode,每个 vnode 上面挂载了一个 _component 属性,这个属性指向了组件实例。而在组件实例上面又挂载了一个 _hooks 属性,这个 _hooks 属性里面保存了执行一个组件的时候,里面所有 Hook 方法相关的信息。
React Hooks 原理与最佳实践 - 图36
首先,有一个全局的 currentIndex 变量,当组件第一次渲染或者更新的时候,它会在每次进入一个函数组件的时候都重置为0,每次遇到一个 Hook 方法就会增加1,同时将这个 Hook 方法的信息放到 _list 里面。
当下次进来或者进入下一个组件的时候, currentIndex 又会被置为0。

组件渲染 => currentIndex 重置 0 => 遇到 Hooks 方法,放进 _list => currentIndex++ => 渲染结束 组件更新 => currentIndex 重置 0 => 遇到 Hooks 方法,获取 _list[currentIndex]=> currentIndex++ => 重复上面步骤 => 更新结束

这个时候就会从刚才的 _list 里面根据 currentIndex 来取出对应项,所以每次进来执行 useState,它依然能拿到上一次更新后的值,因为这里是缓存了起来。
React Hooks 原理与最佳实践 - 图37
通过上面的分析就不难发现,为什么 hooks 方法不能放在条件语句里面了。因为每次进入这个函数的时候,都是要和 currentIndex 一一匹配的,如果更新前后少了一个 Hook 方法,那么就完全对不上了,导致出现大问题。

9.2 useState 和 useReducer

这样再来看下面 useState 和 useReducer 的源码就会更容易理解一些。
React Hooks 原理与最佳实践 - 图38
很明显,getHookState 是根据 currentIndex 来从 _list 里面取和当前 Hook 相关的一些信息。如果是初始化状态(即没有 hookState._component)这个属性的时候,就会去初始化 useState 的两个返回值,否则就会直接返回上一次缓存的结果。

9.3 useEffect

useEffect 和 useState 差不多,区别就在 useEffect 接收的函数会放到一个 _pendingEffects 里面,而非 _list 里面。
React Hooks 原理与最佳实践 - 图39
在 diff 结束之后会从 _pendingEffects 里面取出来函数一个个执行。afterPaint 里面使用了 requestAnimateFrame 这个方法,所以传给 useEffect 里面的方法是在浏览器绘制结束之后才会执行的。
React Hooks 原理与最佳实践 - 图40

9.4 总结

最后,这里对 React Hooks 的整个运行流程来进行一下总结和梳理。

  1. 每个组件实例上挂载一个 _hooks 属性,保证了组件之间不会影响。
  2. 每当遇到一个 hooks 方法,就将其 push 到 currentComponent._hooks._list 中,且 currentIndex 加一。
  3. 每次渲染进入一个组件的时候,都会从将 currentIndex 重置为 0 。遇到 hooks 方法时,currentIndex 重复第二步。这样可以把 currentIndex 和 currentComponent._hooks._list 中的对应项匹配起来,直接取上次缓存的值。
  4. 函数组件每次重新执行后,useState 中还能保持上一次的值,就是来自于步骤3中的缓存。
  5. 由于依赖了 currentComponent 实例,所以 hooks 不能用于普通函数中。

    10、React Hooks 实践

    得益于 react hooks 将业务逻辑从 ui 中抽离出来,目前社区里面关于 react hooks 的实践,大都是从功能点出发。
    从最简单的 api 封装,例如 useDebounce、useThrottle、useImmerState 等等,再到业务层面功能封装,比较出名的库有 react-use、umijs/hooks 等等。
    举个栗子:umijs/hooks 的表格:
    React Hooks 原理与最佳实践 - 图41
    在后台管理系统开发中,表格是非常常见的场景,将分页、查询、loading、排序等等功能打包封装成通用 Hook,就能发挥很大的潜力。