背景
最近询问了一下社区CSE相关的工作,社区反馈可以使用Calc。目前对这个不太了解,于是学习一下。
代码
- RelNode
Calc是一个接口,有一个实现是LogicalCalc。注释中提到了相关的优化规则。
- Rule
Calc相关的规则都收集在CALC_RULES中。
- Program
calc有一个专门的HepProgram
该program也被封装在standard的program中
standard会在Prepare被调用,这意味着Calcite执行query的单元测试可以使用了standard的program,也就是说有可能触发Calc相关的优化。
测试
在JdbcTest中编写下面的测试用例可以发现Calcite通过RexProgram支持了CSE
@Test void testCommonSubExpression() throws SQLException {
CalciteAssert.hr()
.query("select \"salary\","
+ "12 * \"salary\", "
+ "3 * \"salary\", "
+ "\"salary\" * 12 + \"salary\" * 3,"
+ "\"salary\" * 3 + \"salary\" * 12 from \"hr\".\"emps\"")
.returns("");
}
如下图所示:salary * 12 + salary * 3
和salary * 3 + salary * 12
都指向$t13
,可见Calcite提取CSE的时候支持简单的交换律
注意这里的conditionExpr为null,结合LogicalCalc的注视不难猜出Filter也有可能触发这种优化,而一起构建新的program。
另外,转完的SQL并无CSE的优化,就是说无法直接给其他引擎复用。