向量化执行使表达式性能提升10倍成为可能

查询执行引擎对数据库系统性能非常重要。TIDB是一个开源兼容MySQL的HTAP数据库，部署广泛使用的火山模型来执行查询。不幸的是，当查询一个大库时，向量化模型会造成较高的解释开销以及较低的CPU CACHE命中率。

受MonetDB／X100： Hyper－Pipelining Query Execution论文启发，在TIDB中部署向量化执行引擎以提高查询性能。（建议学习 Andy Pavlo 的Query Execution课程，其中详细介绍了有关执行模型和表达式计算的原理）。

2017年年末，我们对TIDB的SQL执行引擎完成了3项优化：

1） 在执行引擎中完成列式存储。类似Apache的Arrow。

2） 将一次一个tuple的迭代模式（火山模型）改成一次一批（1024个tuple）

3） 基于向量化执行原则，优化各种查询操作符

得益于这些优化，与TIDB1．0相比，TIDB2．0显著提升了查询分析的性能。可以查看TPC－H性能

最近我们release了TiDB2．1和3．0，向量化执行引擎更加稳定。现在正在开发TiDB4．0，包括向量化表达式。

本文，深入分析了为什么使用向量化引擎，如何实现它以及如何与社区贡献者合作完成多于360个函数的向量化，还有对未来的看法。

为什么使用向量化

之前TiDB实现了火山模型的执行引擎。这个迭代模型使用标准数据访问接口。在各个算子之间执行open（）－next（）－close（），一行一行处理数据。火山模型简单且可扩展。

但是，当执行大型查询时，火山模型会带来较高的解析代价。另外，也不能重复你利用现代CPU硬件的特性，如CPU CACHE、分支预测、指令流水线。

向量化执行使用单指令在内存中执行一组连续的相似的数据项。与火山模型相比，向量化模型大大降低了解释开销。因此我们选择了向量化执行引擎。

本节中，使用表达式colA＊0．8 ＋ colB来展示基于行的执行和基于向量化执行之间的开销差距。

TIDB根据算术运算符和运算符优先级，将此表达式解析为表达式求值树。在这个树中，每个非叶子节点代表一个算术运算符，叶节点代表数据源。每个非叶节点要么是一个常量如0．8，要么是表中的一个字段如colA。节点之间的父子关系表示计算上的依赖关系：子节点的结算结果是父节点的输入数据。

表达式求值树

每个节点的计算逻辑可以抽象成下面接口：

type Node interface

｛    evalReal（row Row） （val float64， isNull bool）

｝

以＊，0．8，col节点为例，3个节点都实现了上面的接口，伪代码如下：

基于行执行的解释开销

和火山模型类似，上面讨论的表达式实现是对行进行迭代。

逐行迭代的解释开销是多少？下面我们来看一个函数在TiDB中的实现。

builtinArithmeticMultiplyRealSig函数实现2个浮点数相乘。下面的代码块描述了这个函数的实现。右边的数组表示对应行的汇编指令数，是代码汇编后得到的。要注意，此块仅包含在正常条件下迭代的行，并忽略错误处理的逻辑：

下面列出了builtinArithmeticMultiplyRealSig每个功能任务及执行它的汇编指令数量。

每次这个函数执行乘法时，82条指令中仅有8条在执行“真正的”乘法，这仅占总指令的10％左右，其他90％被视为解释开销。一旦将这个函数向量化，它的性能提高了仅9倍。

向量化减少解释开销

和SQL算子的向量化优化一样，我们可以一次处理并返回一批数据。这减少了表达式计算过程中的解释开销。

假设一批数据有1024行，优化后每次调用一个函数处理这1024行数据，然后返回。函数调用的解释开销变成原来的1／1024．

因此，我们可以添加以下接口来向量化表达式计算：

type Node interface ｛

batchEvalReal（rows ［］Row） （vals ［］float64， isNull ［］bool）

｝

我们的向量化处理接口长什么样？为什么？

真正的源代码和上面显示的模型并不完全一样。它看起来像这样：

type VecNode interface ｛

vecEvalReal（input ＊Chunk， result ＊Column）

｝

想知道为什么？

为了说明原因，先介绍下TiDB的chunk结构。也就是查询执行阶段数据在内存中的表示。

2017年底，在进行向量化优化，引入了chunk概念。一个chunk由多列组成。列由两种类型：

1） 固定长度的列，数据时固定长度，不能改变

2） 变长列，数据长度可变

不管数据长度是固定的还是变长的，列中数据在Column．data字段（即数组）中是连续存储在内存中的。如果数据长度变化，column．offset记录数据偏移量。如果数据是固定长度，则不需要记录偏移量。

下图说明了我们最近为chunk引入的新向量访问接口：

1） 对于定长数据，如int64，Golang的unsafe包直接转换column．data为［］int64（［］int64中的Int64s（）），返回结果。读取或更改column．data的用户可以直接操作这个数组。这是访问固定长度数据的最有效方式。

2） 对于变长数据，例如字符串，可以使用GetString（rowIdx int） string来获取对应行的数据，仅能以追加的方式更新。随机更改变长数据列中一个元素，需要移动后续所有数据。这会带来巨大开销。为了提升性能，此操作未在column中。

再返回来看下接口设计。

基于chunk的实现以及Golang的特性，我们优化了表达式计算以下几个方面：

1） 直接通过＊chunk而不是［］Row，避免创建大量Row对象。降低了Golang垃圾回收的压力并提高了性能

2） column．data通过column而不是Row，减少函数调用次数。帮助减少解释开销和加速访问数据。

3） 将用于存储数据的列放到参数中并传递，而不是直接返回［］float64和［］bool数组。提高了内存利用率，减少了Golang GC开销。

基于这些优化，我们的向量化表达式计算接口变成了今天这个样子。

怎么实现向量化执行引擎？

本节介绍如何实现向量化。

以multiplyRealNode为例：

对于这个功能：

1） 前2行：cache pool中的ColumnBuffer用于缓冲右孩子（第二个参数）的数据。左孩子的数据存储再result指针指向的内存中。

2） Columns．MergeNulls（cols．．．）合并多列的NULL标签。这个方法类似result．nulls［i］ ＝ result．nulls［i］ ｜｜ buf．nulls［i］。Column内部使用一个bitmap来维护NULL标签。当调用这个函数时，一个列来做一个按位操作来核并NULLs。

3） 一个循环直接将左右字节的的数据相乘。

4） 再乘法过程中中，该函数调用左右子接口来获取他们的数据。

这种实现方式通过向量化减少了解释开销，对现代CPU更有利：

1） 顺序访问一个向量数据，减少了CPU CACHE的miss

2） 大多数计算工作在一个简单循环中，有助于CPU分支预测与指令流水线。

向量化执行和基于行的执行引擎性能比较

本节使用TiDB源码进行基准测试，并比较向量化前后的性能。使用相同数据（2列浮点数组成的1024行）分别col0＊0．8 ＋ col1计算：

上面的结果表明向量化执行比基于行的执行引擎快4倍。下图对比了LT向量化前后各种小于（LT）函数的性能。横轴表示LT用于测试的函数，纵轴表示完成操作持续的时间（单位纳秒）

下面比较了向量化前后算术函数的性能：

我们测试了300多个向量化函数后，发现这些函数中有超过50％的函数性能提高了1倍以上，18．7％的函数实现了10的性能提升。

如何将360＋个内置函数向量化？

迈向4．0的过程中，表达式向量化是一个巨大的工程。因为涉及500多个内置函数。开发人员相当少－完成相当困难。位更有效开发代码，使用Golang将尽可能多的内置函数向量化test／template。此模板生成向量化函数的源码。位让更多人参与这个项目，我们在开发者社区中成立了向量化表达式工作组。通过这种方法，社区贡献者承担了向量化过程中大部分工作。

在我们的努力下，重构了仅三分之二的内置函数，大部分都有显著的性能提升。有些甚至能有一两个数量级的性能提升。

使用一个模板进行向量化

当我们对内置函数进行向量化时，我们发现很多函数都有相似之处。例如，大多数LT（ ＜）、GT（ ＞） 和LE（ ＜＝） 函数具有相似的逻辑。它们仅使用的运算符不同。因此，可以使用模板来生成这些函数的代码。目前，Golang 不支持泛型类型和宏定义，所以我们使用text／template包来生成代码。基于Golang模板的语法，我们将要生成的函数抽象成模板。例如，这里是比较函数的模板，如LTand GT：

针对不同类型数据和算子，模板生成相应代码。这个模板在expression／generator包里。

社区帮助完成向量化

除了使用模板向量化内置函数外，还在社区开始了向量化活动，让更多人参与进来。编写了一个脚本，为所有内置函数生成向量化接口：

func （b ＊builtinCastStringAsIntSig） vecEvalInt（input ＊chunk．Chunk， result ＊chunk．Column） error ｛

return errors．Errorf（＂not implemented＂）

｝

func （b ＊builtinCastStringAsDurationSig） vectorized（） bool ｛

return false

｝

保留函数接口可以避免合并代码时不同pull request同时修改同一个文档引起的冲突。我们还编写了一个测试框架。贡献者将函数向量化后，他们可以使用框架来测试内置函数的正确性和性能，只需编写几行简单的配置即可。

如下代码所示，贡献者只需要在vecBuiltinArithemeticCases

接下来，它们可以按照我们编写的两个函数运行。然后，他们可以执行正确性和性能测试。

正确性和性能测试都直接生成随机数据，并比较向量化执行和基于行执行的性能。上述两个操作可以帮助贡献者轻松向量化我们的内置函数。在社区的帮助下，我们在短短两个月内对 360 多个函数进行了向量化处理。在此期间，我们的社区贡献者提交了 256 份 PR。该活动还产生了九名活跃贡献者（他们在一年内合并了至少 8 个 PR）和两个审阅者（他们在一年内合并了至少 30 个 PR）。

下一步做什么

通过引入向量化执行，我们显着提高了表达式执行的性能。目前，默认情况下，我们的主分支上启用了向量化执行。一旦表达式的所有内置函数都支持向量化执行，该表达式将使用向量化执行。未来我们也会使用向量化执行来提升 TiDB 的 HTAP 能力。

请注意，上面的所有性能测试数据都在内存中。如果数据存储在磁盘上，读取它可能会导致高开销。在这种情况下，向量化执行的好处可能不太明显。但总的来说，向量化执行明显提升了 TiDB 表达式评估的性能。

此外，当我们对表达式进行向量化时，我们发现向量化执行可以应用于许多其他情况以提高性能。例如：

在哈希连接中，我们为内部数据（参见PR ＃12076）和外部数据（参见PR ＃12669）向量计算哈希键。在某些场景下，性能分别提升7％和18％。我们要感谢一位名叫sduzh的社区贡献者，他独自完成了这些改进。

在哈希聚合中，我们对数据进行了向量化编码。在本地测试中，性能比以前快了20％到60％。有关详细信息，请参阅PR ＃12729。

在StreamAggregation运算符中，我们将数据向量划分为组。在本地测试中，性能提升了约 60％。有关详细信息，请参阅PR ＃12903。

原文