Search in sources :

Example 1 with PlannerImpl

use of org.apache.calcite.prepare.PlannerImpl in project herddb by diennea.

the class CalcitePlanner method runPlanner.

private PlannerResult runPlanner(String defaultTableSpace, String query) throws RelConversionException, SqlParseException, ValidationException, MetadataStorageManagerException, StatementExecutionException {
    try {
        SchemaPlus subSchema = getSchemaForTableSpace(defaultTableSpace);
        if (subSchema == null) {
            clearCache();
            throw new StatementExecutionException("tablespace " + defaultTableSpace + " is not available");
        }
        Properties props = new Properties();
        props.put(CalciteConnectionProperty.TIME_ZONE.camelName(), TimeZone.getDefault().getID());
        props.put(CalciteConnectionProperty.LOCALE.camelName(), Locale.ROOT.toString());
        props.put(CalciteConnectionProperty.DEFAULT_NULL_COLLATION.camelName(), NullCollation.LAST.toString());
        final CalciteConnectionConfigImpl calciteRuntimeContextConfig = new CalciteConnectionConfigImpl(props);
        final FrameworkConfig config = Frameworks.newConfigBuilder().parserConfig(SQL_PARSER_CONFIG).defaultSchema(subSchema).traitDefs(TRAITS).context(new Context() {

            @Override
            public <C> C unwrap(Class<C> aClass) {
                if (aClass == CalciteConnectionConfigImpl.class || aClass == CalciteConnectionConfig.class) {
                    return (C) calciteRuntimeContextConfig;
                }
                return null;
            }
        }).programs(Programs.ofRules(Programs.RULE_SET)).build();
        Planner planner = new PlannerImpl(config);
        if (LOG.isLoggable(Level.FINER)) {
            LOG.log(Level.FINER, "Query: {0}", query);
        }
        try {
            SqlNode n = planner.parse(query);
            n = planner.validate(n);
            RelNode logicalPlan = planner.rel(n).project();
            if (LOG.isLoggable(DUMP_QUERY_LEVEL)) {
                LOG.log(DUMP_QUERY_LEVEL, "Query: {0} {1}", new Object[] { query, RelOptUtil.dumpPlan("-- Logical Plan", logicalPlan, SqlExplainFormat.TEXT, SqlExplainLevel.ALL_ATTRIBUTES) });
            }
            RelDataType originalRowType = logicalPlan.getRowType();
            RelOptCluster cluster = logicalPlan.getCluster();
            final RelOptPlanner optPlanner = cluster.getPlanner();
            optPlanner.addRule(CoreRules.FILTER_REDUCE_EXPRESSIONS);
            RelTraitSet desiredTraits = cluster.traitSet().replace(EnumerableConvention.INSTANCE);
            final RelCollation collation = logicalPlan instanceof Sort ? ((Sort) logicalPlan).collation : null;
            if (collation != null) {
                desiredTraits = desiredTraits.replace(collation);
            }
            final RelNode newRoot = optPlanner.changeTraits(logicalPlan, desiredTraits);
            optPlanner.setRoot(newRoot);
            RelNode bestExp = optPlanner.findBestExp();
            if (LOG.isLoggable(DUMP_QUERY_LEVEL)) {
                LOG.log(DUMP_QUERY_LEVEL, "Query: {0} {1}", new Object[] { query, RelOptUtil.dumpPlan("-- Best  Plan", bestExp, SqlExplainFormat.TEXT, SqlExplainLevel.ALL_ATTRIBUTES) });
            }
            return new PlannerResult(bestExp, originalRowType, logicalPlan, n);
        } catch (AssertionError err) {
            throw new StatementExecutionException("Internal Calcite error " + err, err);
        }
    } catch (java.lang.LinkageError err) {
        LOG.log(Level.SEVERE, "Error on Java Classpath", err);
        throw new StatementExecutionException("Internal Calcite error " + err, err);
    }
}
Also used : InitializerContext(org.apache.calcite.sql2rel.InitializerContext) Context(org.apache.calcite.plan.Context) DataContext(org.apache.calcite.DataContext) RelOptCluster(org.apache.calcite.plan.RelOptCluster) PlannerImpl(org.apache.calcite.prepare.PlannerImpl) CalciteConnectionConfigImpl(org.apache.calcite.config.CalciteConnectionConfigImpl) SchemaPlus(org.apache.calcite.schema.SchemaPlus) RelDataType(org.apache.calcite.rel.type.RelDataType) RelTraitSet(org.apache.calcite.plan.RelTraitSet) Properties(java.util.Properties) RelOptPlanner(org.apache.calcite.plan.RelOptPlanner) StatementExecutionException(herddb.model.StatementExecutionException) RelCollation(org.apache.calcite.rel.RelCollation) RelNode(org.apache.calcite.rel.RelNode) EnumerableSort(org.apache.calcite.adapter.enumerable.EnumerableSort) Sort(org.apache.calcite.rel.core.Sort) Planner(org.apache.calcite.tools.Planner) RelOptPlanner(org.apache.calcite.plan.RelOptPlanner) FrameworkConfig(org.apache.calcite.tools.FrameworkConfig) SqlNode(org.apache.calcite.sql.SqlNode)

Aggregations

StatementExecutionException (herddb.model.StatementExecutionException)1 Properties (java.util.Properties)1 DataContext (org.apache.calcite.DataContext)1 EnumerableSort (org.apache.calcite.adapter.enumerable.EnumerableSort)1 CalciteConnectionConfigImpl (org.apache.calcite.config.CalciteConnectionConfigImpl)1 Context (org.apache.calcite.plan.Context)1 RelOptCluster (org.apache.calcite.plan.RelOptCluster)1 RelOptPlanner (org.apache.calcite.plan.RelOptPlanner)1 RelTraitSet (org.apache.calcite.plan.RelTraitSet)1 PlannerImpl (org.apache.calcite.prepare.PlannerImpl)1 RelCollation (org.apache.calcite.rel.RelCollation)1 RelNode (org.apache.calcite.rel.RelNode)1 Sort (org.apache.calcite.rel.core.Sort)1 RelDataType (org.apache.calcite.rel.type.RelDataType)1 SchemaPlus (org.apache.calcite.schema.SchemaPlus)1 SqlNode (org.apache.calcite.sql.SqlNode)1 InitializerContext (org.apache.calcite.sql2rel.InitializerContext)1 FrameworkConfig (org.apache.calcite.tools.FrameworkConfig)1 Planner (org.apache.calcite.tools.Planner)1