Java AggregateExpandDistinctAggregatesRule类代码示例

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
@Override public void register(RelOptPlanner planner) {
  planner.addRule(PigToEnumerableConverterRule.INSTANCE);
  for (RelOptRule rule : PigRules.ALL_PIG_OPT_RULES) {
    planner.addRule(rule);
  }
  // Don't move Aggregates around, otherwise PigAggregate.implement() won't
  // know how to correctly procuce Pig Latin
  planner.removeRule(AggregateExpandDistinctAggregatesRule.INSTANCE);
  // Make sure planner picks PigJoin over EnumerableJoin. Should there be
  // a rule for this instead for removing ENUMERABLE_JOIN_RULE here?
  planner.removeRule(EnumerableRules.ENUMERABLE_JOIN_RULE);
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
@Test public void testDistinctCountWithoutGroupBy() {
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.INSTANCE)
      .addRuleInstance(AggregateProjectMergeRule.INSTANCE)
      .build();
  final String sql = "select max(deptno), count(distinct ename)\n"
      + "from sales.emp";
  sql(sql).with(program).check();
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
@Test public void testDistinctCount1() {
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.INSTANCE)
      .addRuleInstance(AggregateProjectMergeRule.INSTANCE)
      .build();
  checkPlanning(program,
      "select deptno, count(distinct ename)"
          + " from sales.emp group by deptno");
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
@Test public void testDistinctCount2() {
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.INSTANCE)
      .addRuleInstance(AggregateProjectMergeRule.INSTANCE)
      .build();
  checkPlanning(program,
      "select deptno, count(distinct ename), sum(sal)"
          + " from sales.emp group by deptno");
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
/** Test case for
 * <a href="https://issues.apache.org/jira/browse/CALCITE-1293">[CALCITE-1293]
 * Bad code generated when argument to COUNT(DISTINCT) is a # GROUP BY
 * column</a>. */
@Test public void testDistinctCount3() {
  final String sql = "select count(distinct deptno), sum(sal)"
      + " from sales.emp group by deptno";
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.INSTANCE)
      .build();
  sql(sql).with(program).check();
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
/** Tests implementing multiple distinct count the old way, using a join. */
@Test public void testDistinctCountMultipleViaJoin() {
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.JOIN)
      .addRuleInstance(AggregateProjectMergeRule.INSTANCE)
      .build();
  checkPlanning(program,
      "select deptno, count(distinct ename), count(distinct job, ename),\n"
          + "  count(distinct deptno, job), sum(sal)\n"
          + " from sales.emp group by deptno");
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
/** Tests implementing multiple distinct count the new way, using GROUPING
 *  SETS. */
@Test public void testDistinctCountMultiple() {
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.INSTANCE)
      .addRuleInstance(AggregateProjectMergeRule.INSTANCE)
      .build();
  checkPlanning(program,
      "select deptno, count(distinct ename), count(distinct job)\n"
          + " from sales.emp group by deptno");
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
@Test public void testDistinctCountMultipleNoGroup() {
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.INSTANCE)
      .addRuleInstance(AggregateProjectMergeRule.INSTANCE)
      .build();
  checkPlanning(program,
      "select count(distinct ename), count(distinct job)\n"
          + " from sales.emp");
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
@Test public void testDistinctCountMixedJoin() {
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.JOIN)
      .addRuleInstance(AggregateProjectMergeRule.INSTANCE)
      .build();
  checkPlanning(program,
      "select deptno, count(distinct ename), count(distinct job, ename),\n"
          + "  count(distinct deptno, job), sum(sal)\n"
          + " from sales.emp group by deptno");
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
@Test public void testDistinctCountMixed() {
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.INSTANCE)
      .addRuleInstance(ProjectMergeRule.INSTANCE)
      .build();
  checkPlanning(program,
      "select deptno, count(distinct deptno, job) as cddj, sum(sal) as s\n"
          + " from sales.emp group by deptno");
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
@Test public void testDistinctCountMixed2() {
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.INSTANCE)
      .addRuleInstance(AggregateProjectMergeRule.INSTANCE)
      .addRuleInstance(ProjectMergeRule.INSTANCE)
      .build();
  checkPlanning(program,
      "select deptno, count(distinct ename) as cde,\n"
          + "  count(distinct job, ename) as cdje,\n"
          + "  count(distinct deptno, job) as cddj,\n"
          + "  sum(sal) as s\n"
          + " from sales.emp group by deptno");
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
@Test public void testDistinctCountGroupingSets1() {
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.INSTANCE)
      .addRuleInstance(ProjectMergeRule.INSTANCE)
      .build();
  checkPlanning(program,
      "select deptno, job, count(distinct ename)"
          + " from sales.emp group by rollup(deptno,job)");
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
@Test public void testDistinctCountGroupingSets2() {
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.INSTANCE)
      .addRuleInstance(ProjectMergeRule.INSTANCE)
      .build();
  checkPlanning(program,
      "select deptno, job, count(distinct ename), sum(sal)"
          + " from sales.emp group by rollup(deptno,job)");
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
@Test public void testDistinctNonDistinctAggregates() {
  final String sql = "select emp.empno, count(*), avg(distinct dept.deptno)\n"
      + "from sales.emp emp inner join sales.dept dept\n"
      + "on emp.deptno = dept.deptno\n"
      + "group by emp.empno";
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.JOIN)
      .build();
  sql(sql).with(program).check();
}
 

import org.apache.calcite.rel.rules.AggregateExpandDistinctAggregatesRule; //导入依赖的package包/类
/** Test case for
 * <a href="https://issues.apache.org/jira/browse/CALCITE-1621">[CALCITE-1621]
 * Adding a cast around the null literal in aggregate rules</a>. */
@Test public void testCastInAggregateExpandDistinctAggregatesRule() {
  final String sql = "select name, sum(distinct cn), sum(distinct sm)\n"
      + "from (\n"
      + "  select name, count(dept.deptno) as cn,sum(dept.deptno) as sm\n"
      + "  from sales.dept group by name)\n"
      + "group by name";
  final HepProgram program = HepProgram.builder()
      .addRuleInstance(AggregateExpandDistinctAggregatesRule.INSTANCE)
      .build();
  sql(sql).with(program).check();
}