在 React 中使用 Context 的两点注意事项

  [[419686]]

Context是个好东西，先不论代数效应之类纯理论的概念，能在组件树上无视深度地透传状态确实能给开发带来很大的便利。

但如果Context的使用上不注意一些细节，使用不当，对应用的性能是有可能造成灾难性影响的。近期在做一个产品的性能优化的时候，总结出来微不足道的两点“常识”。

关于顺序

先来看一张图，我愿称之为世界名画： 

这就一演示的界面，没什么东西，我的逻辑也很简单：

import React, {memo, useReducer} from 'react'; 
import {ConfigProvider, Tooltip} from 'antd'; 
import ZH_CN from 'antd/lib/locale-provider/zh_CN'; 
import 'antd/dist/antd.min.css'; 
 
const Text = ({text}) => { 
    return ( 
        <Tooltip title="Tooltip" placement="top"> 
            <div>{text}</div> 
        </Tooltip> 
    ); 
}; 
 
const MemoedText = memo(Text); 
 
const ConfigProviderInside = () => { 
    const [counter, inc] = useReducer((v) => v + 1, 1); 
 
    return ( 
        <ConfigProvider locale={ZH_CN}> 
            <div> 
                <MemoedText text="This is some text." /> 
                <div>App Counter: {counter}</div> 
                <button type="button" onClick={inc}> 
                    FORCE UPDATE 
                </button> 
            </div> 
        </ConfigProvider> 
    ); 
}; 

点一下按钮，整个界面都更新了一遍，是不是也还算正常？那如果做一下“简单”地优化呢： 

是不是很酸爽？我干了什么呢：

const ConfigProviderOutside = () => { 
    const [counter, inc] = useReducer((v) => v + 1, 1); 
 
    return ( 
        <div> 
            <MemoedText text="This is some text." /> 
            <div>App Counter: {counter}</div> 
            <button type="button" onClick={inc}> 
                FORCE UPDATE 
            </button> 
        </div> 
    ); 
}; 
 
render( 
    <ConfigProvider> 
        <ConfigProviderOutside /> 
    </ConfigProvider> 
); 

我把antd的ConfigProvider放到了外面，就这一行代码。

原因也很简单，antd的ConfigProvider并没有做什么优化，它每一次给Context的value都是一个全新的对象（虽然内容并没有变化），这就会导致所有关联的组件都触发更新（虽然毫无意义）。这在你的系统中的下场就是你拼合地用memo、PureComponent之类的方法优化自己写的组件，但那里面的antd组件们却欢快地渲染到停不下来。

所以第一条经验是：好 好梳理Context的上下关系，把那些理论上不会变的放到最外面，把频繁会变的往里放。

什么是不会变的呢，比如Locale、Constant，以及一些系统级的做依赖注入的，这些往往整个生命周期都不会变。

然后是类似CurrentUser这样系统启动的时候请求一次数据的，会从null变成固定的值，随后就不会变了，这一类也尽量往外放。

最后像是Router这样的，会频繁变化的，要放到最里面，免得因为它的更新把其它不怎么变的Context也带进去。

关于粒度

来看一个非常经典的Context：

const DEFAULT_VALUE = { 
    props: null, 
    openGlobalModal: () => undefined, 
    closeGlobalModal: () => undefined, 
}; 
 
const GlobalModalContext = createContext(DEFAULT_VALUE); 
 
const GlobalModalContextProvider = ({children}) => { 
    const [props, setProps] = useState(null); 
    const closeGlobalModal = useCallback( 
        () => setProps(null), 
        [] 
    ); 
    const contextValue = useMemo( 
        () => { 
            return { 
                props, 
                closeGlobalModal, 
                openGlobalModal: setProps, 
            }; 
        }, 
        [props, closeGlobalModal, setProps] 
    ); 
     
    return ( 
        <GlobalModalContext.Provider value={contextValue}> 
            {children} 
        </GlobalModalContext.Provider> 
    ); 
}; 

用一个Context来统一地管理全局单例的对话框，也是一种比较常见的玩法。如果你这么用：

const EditUserLabel = ({id}) => { 
    const {openGlobalModal} = useContext(GlobalMoadlContext); 
    const edit = useCallback( 
        () => openGlobalModal({title: '编辑用户', children: <UserForm id={id} />}), 
        [openGlobalModal, id] 
    ); 
 
    return <span onClick={edit}>编辑</span>; 
}; 
 
const columns = [ 
    // ... 
    { 
        title: '操作', 
        key: 'action', 
        dataIndex: 'id' 
        render: id => <EditUserLabel id={id} />, 
    } 
] 
 
const UserList = ({dataSource}) => ( 
    <Table rowKey="id" dataSource={dataSource} columns={columns} /> 
); 

在一个表格里每一行放一个“编辑”标签，然后在全局放一个对话框：

const GlobalModal = () => { 
    const {props} = useContext(GlobalMoadlContext); 
     
    return !!props && <Modal visible {...props} />; 
}; 

你就会惊讶地发现， 每当你编辑一个用户（或在其它地方触发对话框），表格中每一行的编辑标签都会更新。

原因很容易分析， 因为当你打开对话框的时候，props是变化的，因而contextValue也变化了，所以虽然编辑标签只用了openGlobalModal这个永远不会变的东西，却也硬生生被带着渲染了起来。

如果想追求更少地渲染，就要关注第二条经验： 一个Context中的东西往往并不一起被使用，将它们按使用场景分开，特别是要将多变的和不变的分开。

像上面的代码，就可以优化成这样：

const GlobalModalPropsContext = createContext(null); 
 
const DEFAULT_ACTION = { 
    openGlobalModal: () => undefined, 
    closeGlobalModal: () => undefined, 
}; 
 
const GlobalModalActionContext = createContext(DEFAULT_ACTION); 
 
const GlobalModalContextProvider = ({children}) => { 
    const [props, setProps] = useState(null); 
    const closeGlobalModal = useCallback( 
        () => setProps(null), 
        [] 
    ); 
    const actionValue = useMemo( 
        () => { 
            return { 
                closeGlobalModal, 
                openGlobalModal: setProps, 
            }; 
        }, 
        [closeGlobalModal, setProps] 
    ); 
     
    return ( 
        // 注意第一条经验，变得少的在外面 
        <GlobalModalActionContext.Provider value={actionValue}> 
            <GlobalModalPropsContext.Provider value={props}> 
                {children} 
            </GlobalModalPropsContext.Provider> 
        </GlobalModalActionContext.Provider> 
    ); 
}; 

只要根据实际的需要，去访问2个不同的Context，就可以做到最优化的属性粒度和最少的渲染。

当然我也建议不要直接暴露Context本身，而是将它按照使用场景暴露成若干个hook，这样你可以在一开始不做特别的优化，当性能出现瓶颈的时候再拆Context，只需要修改hook的实现就能做到对外的兼容。

总结

1. 关注在应用中使用的Context的顺序，让不变的在外层，多变的在内层。
2. Context中的内容可以按使用场景和变与不变来拆分成多个更细粒度匠，以减少渲染。