org.apache.calcite.plan.hep.HepProgramBuilder Java Examples
The following examples show how to use
org.apache.calcite.plan.hep.HepProgramBuilder.
You can vote up the ones you like or vote down the ones you don't like,
and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example #1
Source File: HepPlannerTest.java From calcite with Apache License 2.0 | 6 votes |
@Test void testRuleClass() throws Exception { // Verify that an entire class of rules can be applied. HepProgramBuilder programBuilder = HepProgram.builder(); programBuilder.addRuleClass(CoerceInputsRule.class); HepPlanner planner = new HepPlanner( programBuilder.build()); planner.addRule( new CoerceInputsRule(LogicalUnion.class, false, RelFactories.LOGICAL_BUILDER)); planner.addRule( new CoerceInputsRule(LogicalIntersect.class, false, RelFactories.LOGICAL_BUILDER)); final String sql = "(select name from dept union select ename from emp)\n" + "intersect (select fname from customer.contact)"; sql(sql).with(planner).check(); }
Example #2
Source File: SqlHintsConverterTest.java From calcite with Apache License 2.0 | 6 votes |
@Test void testHintsPropagateWithDifferentKindOfRels() { final String sql = "select /*+ AGG_STRATEGY(TWO_PHASE) */\n" + "ename, avg(sal)\n" + "from emp group by ename"; final RelNode rel = tester.convertSqlToRel(sql).rel; final RelHint hint = RelHint.builder("AGG_STRATEGY") .inheritPath(0) .hintOption("TWO_PHASE") .build(); // AggregateReduceFunctionsRule does the transformation: // AGG -> PROJECT + AGG HepProgram program = new HepProgramBuilder() .addRuleInstance(AggregateReduceFunctionsRule.INSTANCE) .build(); HepPlanner planner = new HepPlanner(program); planner.setRoot(rel); RelNode newRel = planner.findBestExp(); new ValidateHintVisitor(hint, Aggregate.class).go(newRel); }
Example #3
Source File: SqlHintsConverterTest.java From calcite with Apache License 2.0 | 6 votes |
@Test void testHintsPropagationInHepPlannerRules() { final String sql = "select /*+ use_hash_join(r, s), use_hash_join(emp, dept) */\n" + "ename, job, sal, dept.name\n" + "from emp join dept on emp.deptno = dept.deptno"; final RelNode rel = tester.convertSqlToRel(sql).rel; final RelHint hint = RelHint.builder("USE_HASH_JOIN") .inheritPath(0) .hintOption("EMP") .hintOption("DEPT") .build(); // Validate Hep planner. HepProgram program = new HepProgramBuilder() .addRuleInstance(MockJoinRule.INSTANCE) .build(); HepPlanner planner = new HepPlanner(program); planner.setRoot(rel); RelNode newRel = planner.findBestExp(); new ValidateHintVisitor(hint, Join.class).go(newRel); }
Example #4
Source File: MaterializedViewSubstitutionVisitorTest.java From calcite with Apache License 2.0 | 6 votes |
private RelNode canonicalize(RelNode rel) { HepProgram program = new HepProgramBuilder() .addRuleInstance(FilterProjectTransposeRule.INSTANCE) .addRuleInstance(FilterMergeRule.INSTANCE) .addRuleInstance(FilterJoinRule.FILTER_ON_JOIN) .addRuleInstance(FilterJoinRule.JOIN) .addRuleInstance(FilterAggregateTransposeRule.INSTANCE) .addRuleInstance(ProjectMergeRule.INSTANCE) .addRuleInstance(ProjectRemoveRule.INSTANCE) .addRuleInstance(ProjectJoinTransposeRule.INSTANCE) .addRuleInstance(ProjectSetOpTransposeRule.INSTANCE) .addRuleInstance(FilterToCalcRule.INSTANCE) .addRuleInstance(ProjectToCalcRule.INSTANCE) .addRuleInstance(FilterCalcMergeRule.INSTANCE) .addRuleInstance(ProjectCalcMergeRule.INSTANCE) .addRuleInstance(CalcMergeRule.INSTANCE) .build(); final HepPlanner hepPlanner = new HepPlanner(program); hepPlanner.setRoot(rel); return hepPlanner.findBestExp(); }
Example #5
Source File: HepPlannerTest.java From calcite with Apache License 2.0 | 6 votes |
@Test void testGC() throws Exception { HepProgramBuilder programBuilder = HepProgram.builder(); programBuilder.addMatchOrder(HepMatchOrder.TOP_DOWN); programBuilder.addRuleInstance(CalcMergeRule.INSTANCE); programBuilder.addRuleInstance(ProjectToCalcRule.INSTANCE); programBuilder.addRuleInstance(FilterToCalcRule.INSTANCE); HepPlanner planner = new HepPlanner(programBuilder.build()); planner.setRoot( tester.convertSqlToRel("select upper(name) from dept where deptno=20").rel); planner.findBestExp(); // Reuse of HepPlanner (should trigger GC). planner.setRoot( tester.convertSqlToRel("select upper(name) from dept where deptno=20").rel); planner.findBestExp(); }
Example #6
Source File: HepPlannerTest.java From calcite with Apache License 2.0 | 6 votes |
@Test void testSubprogram() throws Exception { // Verify that subprogram gets re-executed until fixpoint. // In this case, the first time through we limit it to generate // only one calc; the second time through it will generate // a second calc, and then merge them. HepProgramBuilder subprogramBuilder = HepProgram.builder(); subprogramBuilder.addMatchOrder(HepMatchOrder.TOP_DOWN); subprogramBuilder.addMatchLimit(1); subprogramBuilder.addRuleInstance(ProjectToCalcRule.INSTANCE); subprogramBuilder.addRuleInstance(FilterToCalcRule.INSTANCE); subprogramBuilder.addRuleInstance(CalcMergeRule.INSTANCE); HepProgramBuilder programBuilder = HepProgram.builder(); programBuilder.addSubprogram(subprogramBuilder.build()); final String sql = "select upper(ename) from\n" + "(select lower(ename) as ename from emp where empno = 100)"; sql(sql).with(programBuilder.build()).check(); }
Example #7
Source File: HepPlannerTest.java From calcite with Apache License 2.0 | 6 votes |
/** Tests that if two relational expressions are equivalent, the planner * notices, and only applies the rule once. */ @Test void testCommonSubExpression() { // In the following, // (select 1 from dept where abs(-1)=20) // occurs twice, but it's a common sub-expression, so the rule should only // apply once. HepProgramBuilder programBuilder = HepProgram.builder(); programBuilder.addRuleInstance(FilterToCalcRule.INSTANCE); final HepTestListener listener = new HepTestListener(0); HepPlanner planner = new HepPlanner(programBuilder.build()); planner.addListener(listener); final String sql = "(select 1 from dept where abs(-1)=20)\n" + "union all\n" + "(select 1 from dept where abs(-1)=20)"; planner.setRoot(tester.convertSqlToRel(sql).rel); RelNode bestRel = planner.findBestExp(); assertThat(bestRel.getInput(0).equals(bestRel.getInput(1)), is(true)); assertThat(listener.getApplyTimes() == 1, is(true)); }
Example #8
Source File: BatsOptimizerTest.java From Bats with Apache License 2.0 | 6 votes |
static HepPlanner createHepPlanner() { HepProgramBuilder builder = new HepProgramBuilder(); // builder.addRuleInstance(FilterJoinRule.FilterIntoJoinRule.FILTER_ON_JOIN); // builder.addRuleInstance(FilterJoinRule.JOIN); builder.addRuleCollection(Programs.CALC_RULES); // builder.addRuleCollection(Programs.RULE_SET); // builder.addRuleInstance(ReduceExpressionsRule.PROJECT_INSTANCE); // 加上这个可以把100+100变成200,常量折叠 // builder.addRuleInstance(ReduceExpressionsRule.FILTER_INSTANCE); // builder.addRuleInstance(FilterProjectTransposeRule.INSTANCE); // HepMatchOrder order = HepMatchOrder.TOP_DOWN; // builder.addMatchOrder(order); // builder.addConverters(true); HepPlanner hepPlanner = new HepPlanner(builder.build()); hepPlanner.addRelTraitDef(ConventionTraitDef.INSTANCE); hepPlanner.addRelTraitDef(RelCollationTraitDef.INSTANCE); return hepPlanner; }
Example #9
Source File: RelFieldTrimmerTest.java From calcite with Apache License 2.0 | 6 votes |
@Test void testCalcFieldTrimmer0() { final RelBuilder builder = RelBuilder.create(config().build()); final RelNode root = builder.scan("EMP") .project(builder.field("EMPNO"), builder.field("ENAME"), builder.field("DEPTNO")) .exchange(RelDistributions.SINGLETON) .project(builder.field("EMPNO"), builder.field("ENAME")) .build(); final HepProgram hepProgram = new HepProgramBuilder(). addRuleInstance(ProjectToCalcRule.INSTANCE).build(); final HepPlanner hepPlanner = new HepPlanner(hepProgram); hepPlanner.setRoot(root); final RelNode relNode = hepPlanner.findBestExp(); final RelFieldTrimmer fieldTrimmer = new RelFieldTrimmer(null, builder); final RelNode trimmed = fieldTrimmer.trim(relNode); final String expected = "" + "LogicalCalc(expr#0..1=[{inputs}], proj#0..1=[{exprs}])\n" + " LogicalExchange(distribution=[single])\n" + " LogicalCalc(expr#0..1=[{inputs}], proj#0..1=[{exprs}])\n" + " LogicalProject(EMPNO=[$0], ENAME=[$1])\n" + " LogicalTableScan(table=[[scott, EMP]])\n"; assertThat(trimmed, hasTree(expected)); }
Example #10
Source File: ResultProcessor.java From quark with Apache License 2.0 | 6 votes |
public static String getParsedSql(RelNode relNode, SqlDialect dialect) throws SQLException { if (dialect.getDatabaseProduct() == SqlDialect.DatabaseProduct.HIVE) { final HepProgram program = new HepProgramBuilder() .addRuleInstance(JoinCalcTransposeRule.LEFT_CALC) .addRuleInstance(JoinCalcTransposeRule.RIGHT_CALC) .addRuleInstance(CalcMergeRule.INSTANCE) .build(); final RelOptPlanner planner = relNode.getCluster().getPlanner(); final HepPlanner hepPlanner = new HepPlanner(program, planner.getContext()); hepPlanner.setRoot(relNode); relNode = hepPlanner.findBestExp(); } RelToSqlConverter relToSqlConverter = new RelToSqlConverter(dialect); RelToSqlConverter.Result res = relToSqlConverter.visitChild(0, relNode); SqlNode sqlNode = res.asQuery(); String result = sqlNode.toSqlString(dialect, false).getSql(); return result.replace("\n", " "); }
Example #11
Source File: HepPlannerTest.java From calcite with Apache License 2.0 | 5 votes |
@Test void testMatchUntilFixpoint() throws Exception { // Verify that both unions get rewritten. HepProgramBuilder programBuilder = HepProgram.builder(); programBuilder.addMatchLimit(HepProgram.MATCH_UNTIL_FIXPOINT); programBuilder.addRuleInstance(UnionToDistinctRule.INSTANCE); sql(UNION_TREE).with(programBuilder.build()).check(); }
Example #12
Source File: AbstractMaterializedViewRule.java From Bats with Apache License 2.0 | 5 votes |
@Override public Pair<RelNode, RelNode> pushFilterToOriginalViewPlan(RelBuilder builder, RelNode topViewProject, RelNode viewNode, RexNode cond) { // We add (and push) the filter to the view plan before triggering the rewriting. // This is useful in case some of the columns can be folded to same value after // filter is added. HepProgramBuilder pushFiltersProgram = new HepProgramBuilder(); if (topViewProject != null) { pushFiltersProgram.addRuleInstance(filterProjectTransposeRule); } pushFiltersProgram.addRuleInstance(this.filterAggregateTransposeRule) .addRuleInstance(this.aggregateProjectPullUpConstantsRule).addRuleInstance(this.projectMergeRule); final HepPlanner tmpPlanner = new HepPlanner(pushFiltersProgram.build()); // Now that the planner is created, push the node RelNode topNode = builder.push(topViewProject != null ? topViewProject : viewNode).filter(cond).build(); tmpPlanner.setRoot(topNode); topNode = tmpPlanner.findBestExp(); RelNode resultTopViewProject = null; RelNode resultViewNode = null; while (topNode != null) { if (topNode instanceof Project) { if (resultTopViewProject != null) { // Both projects could not be merged, we will bail out return Pair.of(topViewProject, viewNode); } resultTopViewProject = topNode; topNode = topNode.getInput(0); } else if (topNode instanceof Aggregate) { resultViewNode = topNode; topNode = null; } else { // We move to the child topNode = topNode.getInput(0); } } return Pair.of(resultTopViewProject, resultViewNode); }
Example #13
Source File: Programs.java From Bats with Apache License 2.0 | 5 votes |
/** Creates a program that executes a list of rules in a HEP planner. */ public static Program hep(Iterable<? extends RelOptRule> rules, boolean noDag, RelMetadataProvider metadataProvider) { final HepProgramBuilder builder = HepProgram.builder(); for (RelOptRule rule : rules) { builder.addRuleInstance(rule); } return of(builder.build(), noDag, metadataProvider); }
Example #14
Source File: SqlHintsConverterTest.java From calcite with Apache License 2.0 | 5 votes |
@Test void testHintsForCalc() { final String sql = "select /*+ resource(mem='1024MB')*/ ename, sal, deptno from emp"; final RelNode rel = tester.convertSqlToRel(sql).rel; final RelHint hint = RelHint.builder("RESOURCE") .hintOption("MEM", "1024MB") .build(); // planner rule to convert Project to Calc. HepProgram program = new HepProgramBuilder() .addRuleInstance(ProjectToCalcRule.INSTANCE) .build(); HepPlanner planner = new HepPlanner(program); planner.setRoot(rel); RelNode newRel = planner.findBestExp(); new ValidateHintVisitor(hint, Calc.class).go(newRel); }
Example #15
Source File: Programs.java From Bats with Apache License 2.0 | 5 votes |
/** Creates a program that invokes heuristic join-order optimization * (via {@link org.apache.calcite.rel.rules.JoinToMultiJoinRule}, * {@link org.apache.calcite.rel.rules.MultiJoin} and * {@link org.apache.calcite.rel.rules.LoptOptimizeJoinRule}) * if there are 6 or more joins (7 or more relations). */ public static Program heuristicJoinOrder( final Iterable<? extends RelOptRule> rules, final boolean bushy, final int minJoinCount) { return (planner, rel, requiredOutputTraits, materializations, lattices) -> { final int joinCount = RelOptUtil.countJoins(rel); final Program program; if (joinCount < minJoinCount) { program = ofRules(rules); } else { // Create a program that gathers together joins as a MultiJoin. final HepProgram hep = new HepProgramBuilder() .addRuleInstance(FilterJoinRule.FILTER_ON_JOIN) .addMatchOrder(HepMatchOrder.BOTTOM_UP) .addRuleInstance(JoinToMultiJoinRule.INSTANCE) .build(); final Program program1 = of(hep, false, DefaultRelMetadataProvider.INSTANCE); // Create a program that contains a rule to expand a MultiJoin // into heuristically ordered joins. // We use the rule set passed in, but remove JoinCommuteRule and // JoinPushThroughJoinRule, because they cause exhaustive search. final List<RelOptRule> list = Lists.newArrayList(rules); list.removeAll( ImmutableList.of(JoinCommuteRule.INSTANCE, JoinAssociateRule.INSTANCE, JoinPushThroughJoinRule.LEFT, JoinPushThroughJoinRule.RIGHT)); list.add(bushy ? MultiJoinOptimizeBushyRule.INSTANCE : LoptOptimizeJoinRule.INSTANCE); final Program program2 = ofRules(list); program = sequence(program1, program2); } return program.run( planner, rel, requiredOutputTraits, materializations, lattices); }; }
Example #16
Source File: RelOptTestBase.java From calcite with Apache License 2.0 | 5 votes |
public Sql withRule(RelOptRule... rules) { final HepProgramBuilder builder = HepProgram.builder(); for (RelOptRule rule : rules) { builder.addRuleInstance(rule); } return with(builder.build()); }
Example #17
Source File: InterpreterTest.java From calcite with Apache License 2.0 | 5 votes |
@Test void testInterpretSemiJoin() throws Exception { final String sql = "select x, y from (values (1, 'a'), (2, 'b'), (3, 'c')) as t(x, y)\n" + "where x in\n" + "(select x from (values (1, 'd'), (3, 'g')) as t2(x, y))"; SqlNode validate = planner.validate(planner.parse(sql)); RelNode convert = planner.rel(validate).rel; final HepProgram program = new HepProgramBuilder() .addRuleInstance(SemiJoinRule.PROJECT) .build(); final HepPlanner hepPlanner = new HepPlanner(program); hepPlanner.setRoot(convert); final RelNode relNode = hepPlanner.findBestExp(); final Interpreter interpreter = new Interpreter(dataContext, relNode); assertRows(interpreter, true, "[1, a]", "[3, c]"); }
Example #18
Source File: RelMetadataTest.java From calcite with Apache License 2.0 | 5 votes |
@Test void testCalcColumnOriginsTable() { final String sql = "select name,deptno from dept where deptno > 10"; final RelNode relNode = convertSql(sql); final HepProgram program = new HepProgramBuilder(). addRuleInstance(ProjectToCalcRule.INSTANCE).build(); final HepPlanner planner = new HepPlanner(program); planner.setRoot(relNode); final RelNode calc = planner.findBestExp(); final RelMetadataQuery mq = calc.getCluster().getMetadataQuery(); final RelColumnOrigin nameColumn = mq.getColumnOrigin(calc, 0); assertThat(nameColumn.getOriginColumnOrdinal(), is(1)); final RelColumnOrigin deptnoColumn = mq.getColumnOrigin(calc, 1); assertThat(deptnoColumn.getOriginColumnOrdinal(), is(0)); }
Example #19
Source File: HepPlannerTest.java From calcite with Apache License 2.0 | 5 votes |
private long checkRuleApplyCount(HepMatchOrder matchOrder) { final HepProgramBuilder programBuilder = HepProgram.builder(); programBuilder.addMatchOrder(matchOrder); programBuilder.addRuleInstance(ReduceExpressionsRule.FILTER_INSTANCE); programBuilder.addRuleInstance(ReduceExpressionsRule.PROJECT_INSTANCE); final HepTestListener listener = new HepTestListener(0); HepPlanner planner = new HepPlanner(programBuilder.build()); planner.addListener(listener); planner.setRoot(tester.convertSqlToRel(COMPLEX_UNION_TREE).rel); planner.findBestExp(); return listener.getApplyTimes(); }
Example #20
Source File: HepPlannerTest.java From calcite with Apache License 2.0 | 5 votes |
@Test void testRelNodeCacheWithDigest() { HepProgramBuilder programBuilder = HepProgram.builder(); HepPlanner planner = new HepPlanner( programBuilder.build()); String query = "(select n_nationkey from SALES.CUSTOMER) union all\n" + "(select n_name from CUSTOMER_MODIFIABLEVIEW)"; sql(query).withTester(t -> createDynamicTester()) .withDecorrelation(true) .with(programBuilder.build()) .with(planner) .checkUnchanged(); }
Example #21
Source File: Programs.java From Bats with Apache License 2.0 | 5 votes |
public static Program subQuery(RelMetadataProvider metadataProvider) { final HepProgramBuilder builder = HepProgram.builder(); builder.addRuleCollection(ImmutableList.of((RelOptRule) SubQueryRemoveRule.FILTER, SubQueryRemoveRule.PROJECT, SubQueryRemoveRule.JOIN)); return of(builder.build(), true, metadataProvider); }
Example #22
Source File: HepPlannerTest.java From calcite with Apache License 2.0 | 5 votes |
@Test void testGroup() throws Exception { // Verify simultaneous application of a group of rules. // Intentionally add them in the wrong order to make sure // that order doesn't matter within the group. HepProgramBuilder programBuilder = HepProgram.builder(); programBuilder.addGroupBegin(); programBuilder.addRuleInstance(CalcMergeRule.INSTANCE); programBuilder.addRuleInstance(ProjectToCalcRule.INSTANCE); programBuilder.addRuleInstance(FilterToCalcRule.INSTANCE); programBuilder.addGroupEnd(); final String sql = "select upper(name) from dept where deptno=20"; sql(sql).with(programBuilder.build()).check(); }
Example #23
Source File: RelOptMaterializations.java From Bats with Apache License 2.0 | 5 votes |
private static List<RelNode> substitute( RelNode root, RelOptMaterialization materialization) { // First, if the materialization is in terms of a star table, rewrite // the query in terms of the star table. if (materialization.starTable != null) { RelNode newRoot = RelOptMaterialization.tryUseStar(root, materialization.starRelOptTable); if (newRoot != null) { root = newRoot; } } // Push filters to the bottom, and combine projects on top. RelNode target = materialization.queryRel; HepProgram program = new HepProgramBuilder() .addRuleInstance(FilterProjectTransposeRule.INSTANCE) .addRuleInstance(ProjectMergeRule.INSTANCE) .addRuleInstance(ProjectRemoveRule.INSTANCE) .build(); final HepPlanner hepPlanner = new HepPlanner(program); hepPlanner.setRoot(target); target = hepPlanner.findBestExp(); hepPlanner.setRoot(root); root = hepPlanner.findBestExp(); return new MaterializedViewSubstitutionVisitor(target, root) .go(materialization.tableRel); }
Example #24
Source File: MycatCalcitePlanner.java From Mycat2 with GNU General Public License v3.0 | 5 votes |
public RelNode pullUpUnion(RelNode relNode1) { HepProgramBuilder hepProgramBuilder = new HepProgramBuilder(); hepProgramBuilder.addMatchLimit(PULL_RULES.size()); hepProgramBuilder.addRuleCollection(PULL_RULES); final HepPlanner planner2 = new HepPlanner(hepProgramBuilder.build()); planner2.setRoot(relNode1); return planner2.findBestExp(); }
Example #25
Source File: MaterializationExpander.java From dremio-oss with Apache License 2.0 | 5 votes |
private static RelNode applyRule(RelNode node, RelOptRule rule) { final HepProgramBuilder builder = HepProgram.builder(); builder.addMatchOrder(HepMatchOrder.ARBITRARY); builder.addRuleCollection(ImmutableList.of(rule)); final HepProgram program = builder.build(); final HepPlanner planner = new HepPlanner(program); planner.setRoot(node); return planner.findBestExp(); }
Example #26
Source File: HepPlannerTest.java From calcite with Apache License 2.0 | 5 votes |
@Test void testMatchLimitOneBottomUp() throws Exception { // Verify that only the bottom union gets rewritten. HepProgramBuilder programBuilder = HepProgram.builder(); programBuilder.addMatchLimit(1); programBuilder.addMatchOrder(HepMatchOrder.BOTTOM_UP); programBuilder.addRuleInstance(UnionToDistinctRule.INSTANCE); sql(UNION_TREE).with(programBuilder.build()).check(); }
Example #27
Source File: HepPlannerTest.java From calcite with Apache License 2.0 | 5 votes |
@Test void testMatchLimitOneTopDown() throws Exception { // Verify that only the top union gets rewritten. HepProgramBuilder programBuilder = HepProgram.builder(); programBuilder.addMatchOrder(HepMatchOrder.TOP_DOWN); programBuilder.addMatchLimit(1); programBuilder.addRuleInstance(UnionToDistinctRule.INSTANCE); sql(UNION_TREE).with(programBuilder.build()).check(); }
Example #28
Source File: HepPlannerTest.java From calcite with Apache License 2.0 | 5 votes |
/** * Ensures {@link org.apache.calcite.rel.AbstractRelNode} digest does not include * full digest tree. */ @Test void relDigestLength() { HepProgramBuilder programBuilder = HepProgram.builder(); HepPlanner planner = new HepPlanner( programBuilder.build()); StringBuilder sb = new StringBuilder(); final int n = 10; sb.append("select * from ("); sb.append("select name from sales.dept"); for (int i = 0; i < n; i++) { sb.append(" union all select name from sales.dept"); } sb.append(")"); RelRoot root = tester.convertSqlToRel(sb.toString()); planner.setRoot(root.rel); RelNode best = planner.findBestExp(); // Good digest should look like rel#66:LogicalProject(input=rel#64:LogicalUnion) // Bad digest includes full tree like rel#66:LogicalProject(input=rel#64:LogicalUnion(...)) // So the assertion is to ensure digest includes LogicalUnion exactly once assertIncludesExactlyOnce("best.getDescription()", best.toString(), "LogicalUnion"); assertIncludesExactlyOnce("best.getDigest()", best.getDigest(), "LogicalUnion"); }
Example #29
Source File: HepPlannerTest.java From calcite with Apache License 2.0 | 5 votes |
@Test void testRuleDescription() throws Exception { // Verify that a rule can be applied via its description. HepProgramBuilder programBuilder = HepProgram.builder(); programBuilder.addRuleByDescription("FilterToCalcRule"); HepPlanner planner = new HepPlanner( programBuilder.build()); planner.addRule(FilterToCalcRule.INSTANCE); final String sql = "select name from sales.dept where deptno=12"; sql(sql).with(planner).check(); }
Example #30
Source File: RelFieldTrimmerTest.java From calcite with Apache License 2.0 | 5 votes |
@Test void testCalcFieldTrimmer2() { final RelBuilder builder = RelBuilder.create(config().build()); final RelNode root = builder.scan("EMP") .project(builder.field("EMPNO"), builder.field("ENAME"), builder.field("DEPTNO")) .exchange(RelDistributions.SINGLETON) .filter( builder.call(SqlStdOperatorTable.GREATER_THAN, builder.field("EMPNO"), builder.literal(100))) .project(builder.field("EMPNO"), builder.field("ENAME")) .build(); final HepProgram hepProgram = new HepProgramBuilder() .addRuleInstance(ProjectToCalcRule.INSTANCE) .addRuleInstance(FilterToCalcRule.INSTANCE) .addRuleInstance(CalcMergeRule.INSTANCE).build(); final HepPlanner hepPlanner = new HepPlanner(hepProgram); hepPlanner.setRoot(root); final RelNode relNode = hepPlanner.findBestExp(); final RelFieldTrimmer fieldTrimmer = new RelFieldTrimmer(null, builder); final RelNode trimmed = fieldTrimmer.trim(relNode); final String expected = "" + "LogicalCalc(expr#0..1=[{inputs}], expr#2=[100], expr#3=[>($t0, $t2)], proj#0." + ".1=[{exprs}], $condition=[$t3])\n" + " LogicalExchange(distribution=[single])\n" + " LogicalCalc(expr#0..1=[{inputs}], proj#0..1=[{exprs}])\n" + " LogicalProject(EMPNO=[$0], ENAME=[$1])\n" + " LogicalTableScan(table=[[scott, EMP]])\n"; assertThat(trimmed, hasTree(expected)); }