前言
本次React源码参考版本为17.0.3
。这是React源码系列第二篇,建议初看源码的同学从第一篇开始看起,这样更有连贯性,下面有源码系列链接。
热身准备
useContext
可以帮助我们跨越组件层级直接传递变量,避免了在每一个层级手动的传递 props 属性
,实现共享,要配合createContext
使用。
createContext
createContext
主要功能是创建一个context
,提供Provider
和Consumer
。Provider
主要将context
内容暴露出来,Consumer
可以拿到对应context
的Provider
暴露的内容使用。
示例代码:
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);
它会将Provider
的prop
上的value
字段存到context._currentValue
中。
Consumer
<Context.Consumer>
在渲染时,beginWork
阶段,会执行
prepareToReadContext(workInProgress, renderLanes);
var newValue = readContext(context, newProps.unstable_observedBits);
通过上面代码可以拿到Provider
的prop
上的value
。
值得注意的是, Consumer
标签下包裹的必须是一个函数,如果不是函数会报错。
Consumer
会将拿到的value
作为函数的参数传入函数中去使用。如同上面示例代码中获取到的v
。
useContext
useContext
需要将createContext
创建的Context
作为参数进行调用。
值得一提的是,前面讲的hook
在初始化和更新时会有两套不同函数执行。但是在useContext
只有一个,也就是useContext
在初始化和更新时执行的是一套代码。
初始化 mount & 更新 update
useContext
在mount
时主要会调用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
上,最后将Provider
的prop
上的value
返回。
总结
useContext
的原理类似于观察者模式。Provider
是被观察者, Consumer
和useContext
是观察者。当Provider
上的值发生变化, 观察者是可以观察到的,从而同步信息给到组件。
主要使用场景就是多层级组件值的传递,如果值较多可以考虑配合useReducer
使用。
看完这篇文章, 我们可以弄明白下面这几个问题:
useContext
的原理是什么?
系列文章安排:
- React源码系列之一:Fiber;
- React源码系列之二:React的渲染机制;
- React源码系列之三:hooks之useState,useReducer;
- React源码系列之四:hooks之useEffect;
- React源码系列之五:hooks之useCallback,useMemo;
- React源码系列之六:hooks之useContext;
- React源码系列之七:React的合成事件;
- React源码系列之八:React的diff算法;
- React源码系列之九:React的更新机制;
- React源码系列之十:Concurrent Mode;
参考: