1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.shardingsphere.sqlfederation.optimizer.statement;
19
20 import lombok.RequiredArgsConstructor;
21 import org.apache.calcite.adapter.enumerable.EnumerableConvention;
22 import org.apache.calcite.plan.RelOptPlanner;
23 import org.apache.calcite.rel.RelNode;
24 import org.apache.calcite.rel.metadata.JaninoRelMetadataProvider;
25 import org.apache.calcite.rel.metadata.RelMetadataQueryBase;
26 import org.apache.calcite.rel.type.RelDataType;
27 import org.apache.calcite.sql.SqlNode;
28 import org.apache.calcite.sql2rel.SqlToRelConverter;
29 import org.apache.shardingsphere.sql.parser.sql.common.statement.SQLStatement;
30 import org.apache.shardingsphere.sqlfederation.optimizer.SQLFederationExecutionPlan;
31 import org.apache.shardingsphere.sqlfederation.optimizer.converter.SQLNodeConverterEngine;
32 import org.apache.shardingsphere.sqlfederation.optimizer.operator.util.LogicalScanRelShuttle;
33 import org.apache.shardingsphere.sqlfederation.optimizer.planner.util.SQLFederationPlannerUtils;
34
35 import java.util.Objects;
36
37
38
39
40 @RequiredArgsConstructor
41 public final class SQLStatementCompiler {
42
43 private final SqlToRelConverter converter;
44
45
46
47
48
49
50
51
52 public SQLFederationExecutionPlan compile(final SQLStatement sqlStatement, final String databaseType) {
53 RelMetadataQueryBase.THREAD_PROVIDERS.set(JaninoRelMetadataProvider.DEFAULT);
54 SqlNode sqlNode = SQLNodeConverterEngine.convert(sqlStatement);
55 RelNode logicalPlan = converter.convertQuery(sqlNode, true, true).rel;
56 RelDataType resultColumnType = Objects.requireNonNull(converter.validator).getValidatedNodeType(sqlNode);
57 RelNode replacePlan = LogicalScanRelShuttle.replace(logicalPlan, databaseType);
58 RelNode rewritePlan = rewrite(replacePlan, SQLFederationPlannerUtils.createHepPlanner());
59 RelNode physicalPlan = optimize(rewritePlan, converter);
60 RelMetadataQueryBase.THREAD_PROVIDERS.remove();
61 return new SQLFederationExecutionPlan(physicalPlan, resultColumnType);
62 }
63
64 private RelNode rewrite(final RelNode logicalPlan, final RelOptPlanner hepPlanner) {
65 hepPlanner.setRoot(logicalPlan);
66 return hepPlanner.findBestExp();
67 }
68
69 private RelNode optimize(final RelNode rewritePlan, final SqlToRelConverter converter) {
70 RelOptPlanner planner = converter.getCluster().getPlanner();
71 if (rewritePlan.getTraitSet().equals(converter.getCluster().traitSet().replace(EnumerableConvention.INSTANCE))) {
72 planner.setRoot(rewritePlan);
73 } else {
74 planner.setRoot(planner.changeTraits(rewritePlan, converter.getCluster().traitSet().replace(EnumerableConvention.INSTANCE)));
75 }
76 return planner.findBestExp();
77 }
78 }