我是如何一步步封装一个React Context Composer

动机

React的状态管理方案有很多，比如Redux、Mobx、Recoil等，目前我只体验过Redux，觉得还是比较笨重一点。因为平时写Hooks比较多，所以我比较倾向于使用Context Provider配合useContext这个hook来做，这样也易于状态的拆分与组合。这里，我们不讨论各家状态管理方案的优劣，将目光聚焦于在使用Context时遇到的一个多层嵌套的问题。

下图，是我最近在写的一个taro + react hooks + ts项目抽离出来的一些代码。我对一些全局状态进行了拆分（拆分的目的是为了减少不必要的重新渲染），然后再把它们嵌套起来。这种写法让我回想起了曾经被回调地狱支配的感觉，很难受。因此，我想到了自己去封一个高阶组件，从写法上把结构“扁平化”。

<LoadingContext.Provider value={{ loading, setLoading }}>
  <UserDataContext.Provider value={{ name: "ascodelife", age: 25 }}>
    <ThemeContext.Provider value={"light"}>
    {/* ....more Providers as long as you want */}
    </ThemeContext.Provider>
  </UserDataContext.Provider>
</LoadingContext.Provider>

最易得的方案

这里，我很快的就写出了第一种方案，借助reduce去完成Provider的嵌套。

// ContextComposer.tsx
import React from 'react';

type IContextComposerProps = {
  contexts: { context: React.Context<any>; value: any }[];
};

const ContextComposer: React.FC<IContextComposerProps> = ({ contexts, children }) => {
  return (
    <>
      {contexts.reduce((child, parent) => {
        const { context, value } = parent;
        return <context.Provider value={value}>{child}</context.Provider>;
      }, children)}
    </>
  );
};

export default ContextComposer;


// App.tsx

<ContextComposer
  contexts={[
    { context: ThemeContext, value: "light" },
    { context: UserDataContext, value: { name: "ascodelife", age: 25 } },
    { context: LoadingContext, value: { loading, setLoading } },
  ]}>
    { children }
</ContextComposer>

实际体验后发现，虽然说能用是能用，但是开发体验差那么一点。它的问题在于，组件入参时传的value是any类型，这就意味着放弃了ts的静态类型检查。在传参时，由于不会对value做静态类型检查，敲起代码来不仅不会有任何代码提示，也有可能造成一些比较低级的运行时错误。差评！?

基于React.cloneElement()的改造方案

为了改造上面的这种方案，我翻到了一个比较冷门但好用的函数—— React.cloneElement()。这个函数没有很多需要值得注意的点，主要看一眼它的三个入参，第一个是parent element，第二个是parent props，第三个是剩余参数...children，除第一个参数外，其他都是可选值。

举个例子?：

<!-- 调用函数 -->
React.cloneElement(<div/>,{},<span/>);

<!-- 相当于创建了这样一个结构 -->
<div> 
    <span></span>
</div>

那么下面开始改造，reduce的架子不动，改一下入参的类型和reduce的回调。

// ContextComposer.tsx
import React from 'react';

type IContextComposerProps = {
  contexts: React.ReactElement[];
};

const ContextComposer: React.FC<IContextComposerProps> = ({ contexts, children }) => {
  return (
    <>
      {contexts.reduce((child, parent) => {
        return React.cloneElement(parent,{},child);
      }, children)}
    </>
  );
};

export default ContextComposer;



// App.tsx
<ContextComposer
  contexts={[
      <ThemeContext.Provider value={"light"} />,
      <UserDataContext.Provider value={{ name: "ascodelife", age: 25 }} />,
      <LoadingContext.Provider value={{ loading, setLoading }} />,
  ]}>
    { children }
</ContextComposer>

经过改造后，我们在传参时就好像是真的在创建一个组件（当然实际上也创建了组件，只是这个组件本身没有被渲染到虚拟Dom上，实际渲染上去的是被克隆后的副本）。同时，我们刚才关注的value的静态类型检查问题也得到了解决。
?tips: React.cloneElement(parent,{},child)等价于React.cloneElement(parent,{children:child})，你知道为什么吗？