React16源码: Hooks源码实现

Hooks

1 ）概述

• Hooks 在 React16.7版本出现的新功能
• Hooks 改变了整体应用开发的模式，同时开发体验会和以前会变得不一样
• Hooks 让函数组件具有类组件的能力
  • 在 function component 里面没有this无法保存 state
  • 通过 Hooks可以让 function component 代替 class component
  • Hooks 让函数组件变得更强大

2 ）用例演示

import React, { useState, useEffect } from 'react'

export default () => {
  const [name, setName] = useState('Wang')

  useEffect(() => {
    console.log('component update')

    return () => {
      console.log('unbind')
    }
  }, [])

  return (
    <>
      <p>My Name is: {name}</p>
      <input type="text" value={name} onChange={e => setName(e.target.value)} />
    </>
  )
}

• 这里声明了一个functional component ，在以前对比class component它缺少的是什么
  • 缺少就是this对象，它没有this对象，那么它就不能有 this.state
  • 它就具有包含自己本身的状态的这么一个功能
  • 它没有生命周期方法
• 在这里面我们使用了hooks来给我们的组件去存储了 state
  • 使用 useState 传入了一个默认值是 Wang
  • 然后它返回一个数组，这是一个数组的解构
  • 这个数组第一项是state的对应的这个变量
  • 第二项是让我们去改变这个state的方法
  • 这就是我们通过useState返回给我们的唯一的两个东西
• 然后我们就可以在我们渲染的过程当中去使用这个state了，同样可以去修改这个state
  • 比如说我们绑定了这个input的 onchange 事件
  • 就是去修改这个state，就是我们的name
  • 输入的内容之后它的state就自动更新
  • 在下一次渲染的时候能够拿到 state
• 这就是hooks，它给 function component 提供了class component 所具有的能力
  • 它的意义不仅仅是为了替代 class compoment
  • 是想要去帮助我们去拆分一些在组件内部的逻辑
  • 把他们提取出来，能够给更多的组件进行一个复用
  • 以前在class compoment 里面是很难去拆分这部分逻辑的
• 还有一个是跟 class component 的最大区别，就是生命周期方法
  • 在function component 里面，使用hooks可以通过一个叫做useEffect的这个API
  • 这个东西呢我们就可以传入一个方法，这个方法里面
  • 比如说,随便写一句 component updated
  • 在hooks里面，他没有着重的去区分 mounted 和 updated
  • 它的理念是 mounted和updated 都是 updated
  • 每一次有组件内容更新的时候都会去调用我们传入的这个 effect 回调函数
• 如果需要事件绑定什么的，之前在unmount的时候，去解除事件绑定，那这时怎么办
  • 很简单，看到上面 return一个方法，这个方法就是解除我们的绑定，这边叫做 unbind
  • 在目前这种情况下，每一次有更新的时候都会先执行unbind，然后再重新bind
  • 这是比较符合react更新的一个逻辑的，就是在它有任何更新的时候
  • 都会把之前的状态全部消除，然后返回新的状态
• 当然这对于一些事件监听的绑定不是特别友好, 解决方案如下
  • 把它改造成行为，类似于 componentDidMount 和 componentWillUnmount
  • 直接在这个 useEffect 传入第二个参数，然后传一个空数组
  • 比如说传入一个props来进行一个它是否有变化的一个区分
  • 如果传一个空数组，它就没有东西区分，就代表着只要执行一次
  • 这样，在输入改变state时，就没有了，只再刷新后，才会执行一次
• 同样，在跳出的时候打印输出了 unbind
• 这就是我们使用 hooks 模拟生命周期方法的一个用法

3 ) 源码分析

这个要在 React 16.7 版本上来找，在 React.js 中可看到

import {
  useCallback,
  // ... 很多 khoos
} from './ReactHooks';

现在定位到 ReactHooks.js

/**
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 *
 * @flow
 */

import type {
  MutableSource,
  MutableSourceGetSnapshotFn,
  MutableSourceSubscribeFn,
  ReactContext,
} from 'shared/ReactTypes';
import type {OpaqueIDType} from 'react-reconciler/src/ReactFiberHostConfig';

import invariant from 'shared/invariant';

import ReactCurrentDispatcher from './ReactCurrentDispatcher';

type BasicStateAction<S> = (S => S) | S;
type Dispatch<A> = A => void;

function resolveDispatcher() {
  const dispatcher = ReactCurrentDispatcher.current;
  invariant(
    dispatcher !== null,
    'Invalid hook call. Hooks can only be called inside of the body of a function component. This could happen for' +
      ' one of the following reasons:\n' +
      '1. You might have mismatching versions of React and the renderer (such as React DOM)\n' +
      '2. You might be breaking the Rules of Hooks\n' +
      '3. You might have more than one copy of React in the same app\n' +
      'See https://reactjs.org/link/invalid-hook-call for tips about how to debug and fix this problem.',
  );
  return dispatcher;
}

export function useContext<T>(
  Context: ReactContext<T>,
  unstable_observedBits: number | boolean | void,
): T {
  const dispatcher = resolveDispatcher();
  if (__DEV__) {
    if (unstable_observedBits !== undefined) {
      console.error(
        'useContext() second argument is reserved for future ' +
          'use in React. Passing it is not supported. ' +
          'You passed: %s.%s',
        unstable_observedBits,
        typeof unstable_observedBits === 'number' && Array.isArray(arguments[2])
          ? '\n\nDid you call array.map(useContext)? ' +
              'Calling Hooks inside a loop is not supported. ' +
              'Learn more at https://reactjs.org/link/rules-of-hooks'
          : '',
      );
    }

    // TODO: add a more generic warning for invalid values.
    if ((Context: any)._context !== undefined) {
      const realContext = (Context: any)._context;
      // Don't deduplicate because this legitimately causes bugs
      // and nobody should be using this in existing code.
      if (realContext.Consumer === Context) {
        console.error(
          'Calling useContext(Context.Consumer) is not supported, may cause bugs, and will be ' +
            'removed in a future major release. Did you mean to call useContext(Context) instead?',
        );
      } else if (realContext.Provider === Context) {
        console.error(
          'Calling useContext(Context.Provider) is not supported. ' +
            'Did you mean to call useContext(Context) instead?',
        );
      }
    }
  }
  return dispatcher.useContext(Context, unstable_observedBits);
}

export function useState<S>(
  initialState: (() => S) | S,
): [S, Dispatch<BasicStateAction<S>>] {
  const dispatcher = resolveDispatcher();
  return dispatcher.useState(initialState);
}

export function useReducer<S, I, A>(
  reducer: (S, A) => S,
  initialArg: I,
  init?: I => S,
): [S, Dispatch<A>] {
  const dispatcher = resolveDispatcher();
  return dispatcher.useReducer(reducer, initialArg, init);
}

export function useRef<T>(initialValue: T): {|current: T|} {
  const dispatcher = resolveDispatcher();
  return dispatcher.useRef(initialValue);
}

export function useEffect(
  create: () => (() => void) | void,
  deps: Array<mixed> | void | null,
): void {
  const dispatcher = resolveDispatcher();
  return dispatcher.useEffect(create, deps);
}

export function useLayoutEffect(
  create: () => (() => void) | void,
  deps: Array<mixed> | void | null,
): void {
  const dispatcher = resolveDispatcher();
  return dispatcher.useLayoutEffect(create, deps);
}

export function useCallback<T>(
  callback: T,
  deps: Array<mixed> | void | null,
): T {
  const dispatcher = resolveDispatcher();
  return dispatcher.useCallback(callback, deps);
}

export function useMemo<T>(
  create: () => T,
  deps: Array<mixed> | void | null,
): T {
  const dispatcher = resolveDispatcher();
  return dispatcher.useMemo(create, deps);
}

export function useImperativeHandle<T>(
  ref: {|current: T | null|} | ((inst: T | null) => mixed) | null | void,
  create: () => T,
  deps: Array<mixed> | void | null,
): void {
  const dispatcher = resolveDispatcher();
  return dispatcher.useImperativeHandle(ref, create, deps);
}

export function useDebugValue<T>(
  value: T,
  formatterFn: ?(value: T) => mixed,
): void {
  if (__DEV__) {
    const dispatcher = resolveDispatcher();
    return dispatcher.useDebugValue(value, formatterFn);
  }
}

export const emptyObject = {};

export function useTransition(): [(() => void) => void, boolean] {
  const dispatcher = resolveDispatcher();
  return dispatcher.useTransition();
}

export function useDeferredValue<T>(value: T): T {
  const dispatcher = resolveDispatcher();
  return dispatcher.useDeferredValue(value);
}

export function useOpaqueIdentifier(): OpaqueIDType | void {
  const dispatcher = resolveDispatcher();
  return dispatcher.useOpaqueIdentifier();
}

export function useMutableSource<Source, Snapshot>(
  source: MutableSource<Source>,
  getSnapshot: MutableSourceGetSnapshotFn<Source, Snapshot>,
  subscribe: MutableSourceSubscribeFn<Source, Snapshot>,
): Snapshot {
  const dispatcher = resolveDispatcher();
  return dispatcher.useMutableSource(source, getSnapshot, subscribe);
}

• 定位到 useState

  export function useState<S>(
    initialState: (() => S) | S,
  ): [S, Dispatch<BasicStateAction<S>>] {
    const dispatcher = resolveDispatcher();
    return dispatcher.useState(initialState);
  }
  

  • 这边调用了一个 dispatcher.useState(initialState);
  • 这个dispatch等于 const dispatcher = resolveDispatcher();

• 对应的再去找到这个 resolveDispatcher ，它是个function

  function resolveDispatcher() {
    const dispatcher = ReactCurrentOwner.currentDispatcher;
    invariant(
      dispatcher !== null,
      'Hooks can only be called inside the body of a function component.',
    );
    return dispatcher;
  }
  

  • 可看到是通过 ReactCurrentOwner.currentDispatcher 去获取的
  • 这个就要涉及到后续react-dom渲染的过程
  • 在我们使用阶段，是没有拿到任何真正节点的实例的。
  • 比如在我们创建了react element，我们传进去的是这个class component的类
  • 而不是它的一个new的一个实例，拿不到对应的东西
  • 只有在 react-dom 进行真正的渲染的过程当中，才会去为我们创建这个实例
  • 它这边提供了这个方法，在实际调用是要在我们的 react-dom进行渲染的时候
  • 它才会为我们这个 ReactCurrentOwner 去设置属性

• ReactCurrentOwner 它是一个全局的类，定位一下

  /**
   * Copyright (c) Facebook, Inc. and its affiliates.
   *
   * This source code is licensed under the MIT license found in the
   * LICENSE file in the root directory of this source tree.
   *
   * @flow
   */
  
  import type {Fiber} from 'react-reconciler/src/ReactFiber';
  import typeof {Dispatcher} from 'react-reconciler/src/ReactFiberDispatcher';
  
  /**
   * Keeps track of the current owner.
   *
   * The current owner is the component who should own any components that are
   * currently being constructed.
   */
  const ReactCurrentOwner = {
    /**
     * @internal
     * @type {ReactComponent}
     */
    current: (null: null | Fiber),
    currentDispatcher: (null: null | Dispatcher),
  };
  
  export default ReactCurrentOwner;
  

• 我们可以看到 ReactCurrentOwner ，它就是一个对象，里面有两个属性

• 一个是 current , 就是这个current就对应正在目前正在渲染的哪一个节点的一个实例

• currentDispatcher 是实例对应的 dispatcher

• 它一开始初始化的时候，是两个 null

• 然后到后期每一个组件进行渲染的时候，它才会进行一个渲染

• 就跟我们在class component里面看到的，它的 setState 调用的是 this.updateSetState

• 在这里的 dispatch 也是从不同平台上面，它传入进来的一个东西，不是我们在react中定义的

• 所以在react中我们可以看到的源码就非常的简单，就是这么一些相关的东西

• 同理，useEffect和其他内容也基本是类似的，只不过它最终调用的方法会不一样

• 比如说useReducer，那么它调用的是 dispatcher.useReducer

• 它们最终都是调用 dispatcher 上面的方法