QuestDB JIT编译器对数值列IN操作符的支持优化

背景介绍

QuestDB作为一款高性能的时间序列数据库，其查询性能优化一直是开发重点。在最新开发中，团队关注到了JIT(即时编译)编译器对SQL查询中IN操作符的处理优化问题。

问题分析

在QuestDB的SQL查询处理中，JIT编译器前端CompiledFilterIRSerializer负责将SQL过滤条件AST转换为中间表示(IR)二进制代码。当前存在一个性能优化机会：对于数值列使用IN操作符的查询，编译器生成的IR代码与使用多个OR条件连接的等效查询不同。

具体来说，以下两种查询形式：

WHERE long_col = 42 AND int_col IN (2, 3, 5) AND short_col = 0;
WHERE long_col = 42 AND (int_col =2 OR int_col = 3 OR int_col = 5) AND short_col = 0;

理想情况下，这两种语法形式应该生成相同的IR代码，从而获得相同的执行性能。但当前实现中，IN操作符的处理路径与显式OR条件不同，可能导致性能差异。

技术实现

QuestDB的JIT编译流程包含两个主要部分：

1. 前端(CompiledFilterIRSerializer)：将SQL过滤条件AST转换为IR
2. 后端：基于C++和asmjit库实现的IR编译器

测试套件CompiledFilterIRSerializerTest负责验证IR生成的正确性。

优化方案

开发团队提出的解决方案是修改JIT编译器前端，使IN操作符在数值列上的处理与等效的OR条件表达式生成完全相同的IR代码。这种优化可以带来以下好处：

1. 消除语法形式差异导致的性能差异
2. 保持查询计划的一致性
3. 为后续优化奠定基础

扩展优化

在基础优化完成后，团队还计划将这一优化扩展到时间戳列的区间查询场景。例如对atimestamp IN <time_range_expression>的支持，这将进一步提升时间序列查询场景下的性能。

总结

通过对QuestDB JIT编译器IN操作符处理的优化，可以统一不同语法形式的查询执行路径，消除潜在的性能差异，并为更复杂的时间序列查询优化奠定基础。这一优化体现了QuestDB对查询性能极致追求的研发理念。