image.png

Calcite 主要功能

总结来说Calcite有以下主要功能:

  • SQL 解析
  • SQL 校验
  • 查询优化
  • SQL 生成器
  • 数据连接

    Calcite 解析SQl的步骤

    image.png
    如上图中所述,一般来说Calcite解析SQL有以下几步:

  • Parser. 此步中Calcite通过Java CC将SQL解析成未经校验的AST

  • Validate. 该步骤主要作用是校证Parser步骤中的AST是否合法,如验证SQL scheme、字段、函数等是否存在; SQL语句是否合法等. 此步完成之后就生成了RelNode树(关于RelNode树, 请参考下文)
  • Optimize. 该步骤主要的作用优化RelNode树, 并将其转化成物理执行计划。主要涉及SQL规则优化如:基于规则优化(RBO)及基于代价(CBO)优化; Optimze 这一步原则上来说是可选的, 通过Validate后的RelNode树已经可以直接转化物理执行计划,但现代的SQL解析器基本上都包括有这一步,目的是优化SQL执行计划。此步得到的结果为物理执行计划。
  • Execute,即执行阶段。此阶段主要做的是:将物理执行计划转化成可在特定的平台执行的程序。如Hive与Flink都在在此阶段将物理执行计划CodeGen生成相应的可执行代码。

    Calcite相关组件

    Calcite主要有以下概念:

  • Catelog: 主要定义SQL语义相关的元数据与命名空间。

  • SQL parser: 主要是把SQL转化成AST.
  • SQL validator: 通过Catalog来校证AST.
  • Query optimizer: 将AST转化成物理执行计划、优化物理执行计划.
  • SQL generator: 反向将物理执行计划转化成SQL语句.

    Catalog

    Catalog:主要定义被SQL访问的命名空间,主要包括以下几点:
  1. schema: 主要定义schema与表的集合,schame 并不是强制一定需要的,比如说有两张同名的表T1, T2,就需要schema要区分这两张表,如A.T1, B.T1
  2. 表:对应关系数据库的表,代表一类数据,在calcite中由RelDataType定义
  3. RelDataType 代表表的数据定义,如表的数据列名称、类型等。

    Schema

    1. public interface Schema {
    2. Table getTable(String name);
    3. Set<String> getTableNames();
    4. Set<String> getFunctionNames();
    5. Schema getSubSchema(String name);
    6. Set<String> getSubSchemaNames();
    7. Expression getExpression(SchemaPlus parentSchema, String name);
    8. boolean isMutable();

    Table

    public interface Table {
    
    RelDataType getRowType(RelDataTypeFactory typeFactory);
    Statistic getStatistic();
    
    Schema.TableType getJdbcTableType();
    }
    

    其中RelDataType代表Row的数据类型, Statistic 用于统计表的相关数据、特别是在CBO用于计算表的代价。

    SQL Parser

    由Java CC编写,将SQL转化成AST.

  • Java CC 指的是Java Compiler Compiler, 可以将一种特定域相关的语言转化成Java语言
  • 在Calcite中将标记(Token)表示为 SqlNode, 并且Sqlnode可以通过unparse方法反向转化成SQL
    cast(id as float)
    
    Java CC 可表示为
    <CAST>
    <LPAREN>
    e = Expression(ExprContext.ACCEPT_SUBQUERY)
    <AS>
    dt = DataType() {agrs.add(dt);}
    <RPAREN>
    ....
    

    Query Optimizer

    INSERT INTO tmp_node
    SELECT s1.id1, s1.id2, s2.val1
    FROM source1 as s1 INNER JOIN source2 AS s2
    ON s1.id1 = s2.id1 and s1.id2 = s2.id2 where s1.val1 > 5 and s2.val2 = 3;
    
    通过Calcite转化为:
    LogicalTableModify(table=[[TMP_NODE]], operation=[INSERT], flattened=[false])
    LogicalProject(ID1=[$0], ID2=[$1], VAL1=[$7])
      LogicalFilter(condition=[AND(>($2, 5), =($8, 3))])
        LogicalJoin(condition=[AND(=($0, $5), =($1, $6))], joinType=[INNER])
          LogicalTableScan(table=[[SOURCE1]])
          LogicalTableScan(table=[[SOURCE2]])
    
    是未经优化的RelNode树,可以发现最底层是TableScan,也是读取表的原始数据,紧接着是LogicalJoin,Joiner的类型为INNER JOIN, LogicalJoin之后接下做LogicalFilter 操作,对应SQL中的WHERE条件,最后做Project也就是投影操作。
    但是我们可以观察到对于INNER JOIN而言, WHERE 条件是可以下推,如
    LogicalTableModify(table=[[TMP_NODE]], operation=[INSERT], flattened=[false])
    LogicalProject(ID1=[$0], ID2=[$1], VAL1=[$7])
        LogicalJoin(condition=[AND(=($0, $5), =($1, $6))], joinType=[inner])
          LogicalFilter(condition=[=($4, 3)])  
            LogicalProject(ID1=[$0], ID2=[$1],      ID3=[$2], VAL1=[$3], VAL2=[$4],VAL3=[$5])
              LogicalTableScan(table=[[SOURCE1]])
          LogicalFilter(condition=[>($3,5)])    
            LogicalProject(ID1=[$0], ID2=[$1], ID3=[$2], VAL1=[$3], VAL2=[$4],VAL3=[$5])
              LogicalTableScan(table=[[SOURCE2]])
    
    这样可以减少JOIN的数据量,提高SQL效率,实际过程中可以将JOIN 的中条件下推以较少Join的数据量
    image.png
    T1 JOIN T2 JOIN T3
    
    类似于此种情况JOIN的顺序是上图的前者还是后者?这就涉及到Optimizer所使用的方法,Optimizer主要目的就是减小SQL所处理的数据量、减少所消耗的资源并最大程度提高SQL执行效率如:剪掉无用的列、合并投影、子查询转化成JOIN、JOIN重排序、下推投影、下推过滤等。目前主要有两类优化方法:基于语法(RBO)与基于代价(CBO)的优化

    RBO(Rule Based Optimization)

    通俗一点的话就是事先定义一系列的规则,然后根据这些规则来优化执行计划。

    ProjectFilterRule

    此Rule的使用场景为Filter在Project之上,可以将Filter下推。假如某一个RelNode树
      LogicalFilter
        LogicalProject
          LogicalTableScan
    
    则可优化成
    LogicalProject
        LogicalFilter
          LogicalTableScan
    

    FilterJoinRule

    此Rule的使用场景为Filter在Join之上,可以先做Filter然后再做Join, 以减少Join的数量

等等,还有很多类似的规则。但RBO一定程度上是经验试的优化方法,无法有一个公式上的判断哪种优化更优。 在Calcite中实现方法为 HepPlanner

CBO(Cost Based Optimization)

通俗一点的说法是:通过某种算法计算SQL所有可能的执行计划的“代价”,选择某一个代价较低的执行计划,如上文中三张表作JOIN, 一般来说RBO无法判断哪种执行计划优化更好,只有分别计算每一种JOIN方法的代价。
Calcite会将每一种操作(如LogicaJoin、LocialFilter、 LogicalProject、LogicalScan) 结合实际的Schema转化成具体的代价数,比较不同的执行计划所具有的代价,然后选择相对小计划作为最终的结果,之所以说相对小,这是因为如果要完全遍历计算所有可能的代价可能得不偿失,花费更多的人力与资源,因此只是说选择相对最优的执行计划。CBO目的是“避免使用最差的执行计划,而不是找到最好的”
目前Calcite中就是采用CBO进行优化,实现方法为VolcanoPlanner,有关此算法的具体内容可以参考原码

如何使用Calcite

由于Calcite是Java语言编写,因此只需要在工程或项目中引入相应的Jar包即可,下面为一个可以运行的例子:


public class TestOne {
    public static class TestSchema {
        public final Triple[] rdf = {new Triple("s", "p", "o")};
    }

    public static void main(String[] args) {
        SchemaPlus schemaPlus = Frameworks.createRootSchema(true);

        //给schema T中添加表
        schemaPlus.add("T", new ReflectiveSchema(new TestSchema()));
        Frameworks.ConfigBuilder configBuilder = Frameworks.newConfigBuilder();
        //设置默认schema
        configBuilder.defaultSchema(schemaPlus);

        FrameworkConfig frameworkConfig = configBuilder.build();

        SqlParser.ConfigBuilder paresrConfig = SqlParser.configBuilder(frameworkConfig.getParserConfig());

        //SQL 大小写不敏感
        paresrConfig.setCaseSensitive(false).setConfig(paresrConfig.build());

        Planner planner = Frameworks.getPlanner(frameworkConfig);

        SqlNode sqlNode;
        RelRoot relRoot = null;
        try {
            //parser阶段
            sqlNode = planner.parse("select \"a\".\"s\", count(\"a\".\"s\") from \"T\".\"rdf\" \"a\" group by \"a\".\"s\"");
            //validate阶段
            planner.validate(sqlNode);
            //获取RelNode树的根
            relRoot = planner.rel(sqlNode);
        } catch (Exception e) {
            e.printStackTrace();
        }

        RelNode relNode = relRoot.project();
        System.out.print(RelOptUtil.toString(relNode));
    }
}

类Triple 对应的表定义:

public class Triple {
    public String s;
    public String p;
    public String o;

    public Triple(String s, String p, String o) {
        super();
        this.s = s;
        this.p = p;
        this.o = o;
    }

}