React源码系列之六:hooks之useContext

3,426 阅读2分钟

前言

本次React源码参考版本为17.0.3。这是React源码系列第二篇,建议初看源码的同学从第一篇开始看起,这样更有连贯性,下面有源码系列链接。

热身准备

useContext可以帮助我们跨越组件层级直接传递变量,避免了在每一个层级手动的传递 props 属性,实现共享,要配合createContext使用。

createContext

createContext主要功能是创建一个context,提供ProviderConsumerProvider主要将context内容暴露出来,Consumer可以拿到对应contextProvider暴露的内容使用。

示例代码:

export const Context = createContext(null)

<Context.Provider value='initialValue'>
  <Context.Consumer>
    {(v) => {
      return <h2>{v}</h2>
    }}
  </Context.Consumer>
</Context.Provider>

Provider

<Context.Provider>在渲染时,beginWork阶段,会执行

pushProvider(workInProgress, newValue);

它会将Providerprop上的value字段存到context._currentValue中。

Consumer

<Context.Consumer>在渲染时,beginWork阶段,会执行

prepareToReadContext(workInProgress, renderLanes);
var newValue = readContext(context, newProps.unstable_observedBits);

通过上面代码可以拿到Providerprop上的value

值得注意的是, Consumer标签下包裹的必须是一个函数,如果不是函数会报错。 Consumer会将拿到的value作为函数的参数传入函数中去使用。如同上面示例代码中获取到的v

useContext

useContext需要将createContext创建的Context作为参数进行调用。

值得一提的是,前面讲的hook在初始化和更新时会有两套不同函数执行。但是在useContext只有一个,也就是useContext在初始化和更新时执行的是一套代码。

初始化 mount & 更新 update

useContextmount时主要会调用readContext函数:

function readContext(context, observedBits) {

  var contextItem = {
    context: context,  // 传入的context
    observedBits: resolvedObservedBits,  // 观察范围(默认全部update)
    next: null
  };

  lastContextDependency = contextItem;
  currentlyRenderingFiber.dependencies = {
    lanes: NoLanes,
    firstContext: contextItem,
    responders: null
  };
  } else {
    // Append a new context item.
    lastContextDependency = lastContextDependency.next = contextItem;
  }

  return  context._currentValue ;
}

精简了下代码,可以看到,readContext会创建一个contextItem并以链表的结构记录在对应fiber.dependencies上,最后将Providerprop上的value返回。

总结

useContext的原理类似于观察者模式。Provider是被观察者, ConsumeruseContext是观察者。当Provider上的值发生变化, 观察者是可以观察到的,从而同步信息给到组件。

主要使用场景就是多层级组件值的传递,如果值较多可以考虑配合useReducer使用。

看完这篇文章, 我们可以弄明白下面这几个问题:

  1. useContext的原理是什么?

系列文章安排:

  1. React源码系列之一:Fiber
  2. React源码系列之二:React的渲染机制
  3. React源码系列之三:hooks之useState,useReducer
  4. React源码系列之四:hooks之useEffect
  5. React源码系列之五:hooks之useCallback,useMemo
  6. React源码系列之六:hooks之useContext;
  7. React源码系列之七:React的合成事件;
  8. React源码系列之八:React的diff算法;
  9. React源码系列之九:React的更新机制;
  10. React源码系列之十:Concurrent Mode;

参考:

React官方文档

github