1.没写过复杂 React 组件？来实现下 AntD 的码解 Space 组件吧

没写过复杂 React 组件？来实现下 AntD 的 Space 组件吧

       React 开发者在日常工作中经常编写组件，但这些大多为业务组件，码解复杂度并不高。码解

       组件通常通过传入 props 并使用 hooks 组织逻辑来渲染视图，码解偶尔会用到 context 跨层传递数据。码解

       相对复杂的码解源码编程的教材有哪些组件是怎样的呢？antd 组件库中就有许多。

       今天，码解我们将实现antd组件库中的码解一个组件——Space组件。

       首先，码解我们来了解一下Space组件的码解使用方法：

       Space是一个布局组件，用于设置组件的码解间距，还可以设置多个组件的码解对齐方式。

       例如，码解我们可以使用Space组件来包裹三个盒子，码解设置方向为水平，码解渲染结果如下：

       当然，我们也可以设置为垂直：

       水平和垂直的间距可以通过size属性设置，如large、middle、small或任意数值。

       多个子节点可以设置对齐方式，runtimeexception源码如start、end、center或baseline。

       此外，当子节点过多时，可以设置换行。

       Space组件还可以单独设置行列的间距。

       最后，它还可以设置split分割线部分。

       此外，你也可以不直接设置size，而是通过ConfigProvider修改context中的默认值。

       Space组件会读取context中的size值，这样如果有多个Space组件，就不需要每个都设置，只需要添加一个ConfigProvider即可。

       这就是Space组件的全部用法，简单回顾一下几个参数和用法：

       Space组件的使用方法很简单，但功能非常强大。

       接下来，aidl源码我们来探讨一下这样的布局组件是如何实现的。

       首先，我们来看一下它最终的DOM结构：

       每个box都包裹了一层div，并设置了ant-space-item类。

       split部分包裹了一层span，并设置了ant-space-item-split类。

       最外层包裹了一层div，并设置了ant-space类。

       这些看起来很简单，但实现起来却有很多细节。

       下面我们来写一下Space组件的实现代码：

       首先，我们声明组件props的类型。

       需要注意的是，style是React.CSSProperties类型，即可以设置各种CSS样式。

       split是React.ReactNode类型，即可以传入jsx。

       其余参数的类型根据其取值而定。

       Space组件会对所有子组件包裹一层div，因此需要遍历传入的源码365children并做出修改。

       props传入的children需要转换为数组，可以使用React.Children.toArray方法。

       虽然children已经是数组了，但为什么还要使用React.Children.toArray转换一下呢？

       因为toArray可以对children进行扁平化处理。

       更重要的是，直接调用children.sort()会报错，而toArray之后就不会了。

       因此，我们会使用React.Children.forEach、React.Children.map等方法操作children，而不是直接操作。

       但这里我们有一些特殊的需求，比如空节点不过滤掉，依然保留。

       因此，我们使用React.Children.forEach自己实现toArray：

       这部分比较容易理解，就是使用React.Children.forEach遍历jsx节点，对每个节点进行判断，如果是数组或fragment就递归处理，否则push到数组中。稳赢源码

       保不保留空节点可以根据keepEmpty的option来控制。

       这样，children就可以遍历渲染item了，这部分是这样的：

       我们单独封装了一个Item组件。

       然后，我们遍历childNodes并渲染这个Item组件。

       最后，我们将所有的Item组件放在最外层的div中：

       这样就可以分别控制整体布局和Item布局了。

       具体的布局还是通过className和样式来实现的：

       className通过props计算而来，使用了classnames包，这是react生态中常用的包，根据props动态生成className基本都会使用这个包。

       这个前缀是动态获取的，最终就是ant-space的前缀。

       这些class的样式都定义好了：

       整个容器使用inline-flex，然后根据不同的参数设置align-items和flex-direction的值。

       最后一个direction的css可能大家没用过，是设置文本方向的。

       这样，就通过props动态给最外层div添加了相应的className，设置了对应的样式。

       但还有一部分样式没有设置，也就是间距。

       其实这部分可以使用gap设置，当然，也可以使用margin，但处理起来比较麻烦。

       不过，antd这种组件自然要做得兼容性好一点，所以两种都支持，支持gap就使用gap，否则使用margin。

       问题来了，antd是如何检测浏览器是否支持gap样式的呢？

       antd创建一个div，设置样式，并添加到body下，然后查看scrollHeight的值，最后删除这个元素。

       这样就可以判断是否支持gap、column等样式，因为不支持的话高度会是0。

       然后antd提供了一个这样的hook：

       第一次会检测并设置state的值，之后直接返回这个检测结果。

       这样组件里就可以使用这个hook来判断是否支持gap，从而设置不同的样式了。

       最后，这个组件还会从ConfigProvider中取值，我们之前见过：

       所以，我们再处理一下这部分：

       使用useContext读取context中的值，并设置为props的解构默认值，这样如果传入了props.size就使用传入的值，否则使用context中的值。

       这里给Item子组件传递数据也是通过context，因为Item组件不一定会在哪一层。

       使用createContext创建context对象：

       把计算出的size和其他一些值通过Provider设置到spaceContext中：

       这样子组件就能拿到spaceContext中的值了。

       这里使用了useMemo，很多同学不会用，其实很容易理解：

       props变化会触发组件重新渲染，但有时候props并不需要变化却每次都变，这样就可以通过useMemo来避免它不必要的更新。

       useCallback也是同样的道理。

       计算size时封装了一个getNumberSize方法，为字符串枚举值设置了一些固定的数值：

       至此，这个组件我们就完成了，当然，Item组件还没展开讲。

       先来欣赏一下这个Space组件的全部源码：

       回顾一下要点：

       思路理得差不多了，再来看一下Item的实现：

       这部分比较简单，直接上全部代码了：

       通过useContext从SpaceContext中取出Space组件里设置的值。

       根据是否支持gap来分别使用gap或margin、padding的样式来设置间距。

       每个元素都用div包裹一下，设置className。

       如果不是最后一个元素并且有split部分，就渲染split部分，用span包裹。

       这块还是比较清晰的。

       最后，还有ConfigProvider的部分没有看：

       这部分就是创建一个context，并初始化一些值：

       有没有感觉antd里用context简直太多了！

       确实。

       为什么？

       因为你不能保证组件和子组件隔着几层。

       比如Form和FormItem：

       比如ConfigProvider和各种组件（这里是Space）：

       还有刚讲过的Space和Item。

       它们能用props传数据吗？

       不能，因为不知道隔几层。

       所以antd里基本都是用context传数据的。

       你会你在antd里会见到大量的用createContext创建context，通过Provider修改context值，通过Consumer或useContext读取context值的这类逻辑。

       最后，我们来测试一下自己实现的这个Space组件吧：

       测试代码如下：

       这部分不用解释了。就是ConfigProvider包裹了两个Space组件，这两个Space组件没有设置size值。

       设置了direction、align、split、wrap等参数。

       渲染结果是正确的：

       就这样，我们自己实现了antd的Space组件！

       完整代码在github：github.com/QuarkGluonPl...

       总结：

       一直写业务代码，可能很少写一些复杂的组件，而antd里就有很多复杂组件，我们挑Space组件来写了下。

       这是一个布局组件，可以通过参数设置水平、垂直间距、对齐方式、分割线部分等。

       实现这个组件的时候，我们用到了很多东西：

       很多同学不会封装布局组件，其实就是对整体和每个item都包裹一层，分别设置不同的class，实现不同的间距等的设置。

       想一下，这些东西以后写业务组件是不是也可以用上呢？